SPU规格选项实现多组合SKU-笛卡尔积

SPU规格选项实现多组合SKU-笛卡尔积

根据SPU生成SKU

/**
   * 新增SPU单元
   *
   * @param spSpu SPU单元
   * @return 结果
   */
  @Override
  public int insertSpSpu(SpSpu spSpu) {
      spSpu.setCreateTime(DateUtils.getNowDate());

      // 通过规格列表id查询规则参数，拼接为JSON数组字符串
      String[] strings = spSpu.getSpecItems().replace("，", ",").split(",");

      // 记录排列组合选项
      List<List<String>> list = new ArrayList<>();
      // 记录规格选项
      List<String> specNameList = new ArrayList<>();

      JSONObject jsonObject = new JSONObject();
      for (String temp : strings) {
          SpSpec spSpec = iSpSpecService.selectSpSpecById(Long.parseLong(temp));

          jsonObject.put(spSpec.getName(), spSpec.getOptions());

          List<String> optionList =
                  Arrays.asList(spSpec.getOptions().replace("，", ",").split(","));
          list.add(optionList);

          specNameList.add(spSpec.getName());
      }
      spSpu.setSpecItems(jsonObject.toJSONString());

      // 雪花id
      IdentifierGenerator identifierGenerator = new DefaultIdentifierGenerator();
      spSpu.setSpuId(identifierGenerator.nextId(new Object()).toString());

      if (spSpuMapper.insertSpSpu(spSpu) != 1) {
          log.error("SPU数据插入失败，SPU_ID：{}，SPU名称：{}", spSpu.getSpuId(), spSpu.getSpuName());
      }

      /** 排列组合生成SKU */
      /** 生成笛卡尔积数据 */
      List<String> descartesList = descartes(list);

      descartesList.forEach(
              item -> {
                  SpSku spSku = new SpSku();
                  spSku.setSpuId(spSpu.getSpuId());

                  // 雪花id
                  IdentifierGenerator identifierGeneratorSku = new DefaultIdentifierGenerator();
                  spSku.setSkuId(identifierGeneratorSku.nextId(new Object()).toString());

                  spSku.setSkuName(spSpu.getSpuName() + " " + item);

                  spSku.setImages(spSpu.getImages());
                  spSku.setMainImage(spSpu.getMainImage());
                  spSku.setPrice(new BigDecimal(0));
                  spSku.setSaleNum(0L);

                  // 生成规格字符串
                  JSONObject object = new JSONObject();
                  List<String> paramList = Arrays.asList(item.split(" "));
                  for (int i = 0; i < paramList.size(); i++) {
                      object.put(specNameList.get(i), paramList.get(i));
                  }
                  spSku.setSpec(object.toJSONString());

                  spSku.setStock(0L);
                  spSku.setWeight(new BigDecimal(0));

                  spSku.setCreateTime(DateUtils.getNowDate());

                  if (iSpSkuService.insertSpSku(spSku) != 1) {
                      log.error("SKU数据插入失败，SKU_ID：{}，规格参数：{}", spSku.getSkuId(), spSku.getSpec());
                  }
              });

      return 1;
  }

  /**
   * 笛卡尔集合实现
   *
   * @param lists
   */
  public List<String> descartes(List<List<String>> lists) {
      List<String> tempList = new ArrayList<>();
      for (List<String> list : lists) {
          if (tempList.isEmpty()) {
              tempList = list;
          } else {
              // java8新特性，stream流
              tempList =
                      tempList.stream()
                              .flatMap(item -> list.stream().map(item2 -> item + " " + item2))
                              .collect(Collectors.toList());
          }
      }
      return tempList;
  }

笛卡尔积示例

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

public class DescartesDemo {

    public static void main(String[] args) {

        List<String> colorList = Arrays.asList("红色", "黑色", "金色");
        List<String> sizeList = Arrays.asList("32G", "64G");
        List<String> placeList = Arrays.asList("国产", "进口");

        List<String> descartesList = descartes(colorList, sizeList, placeList);
        descartesList.forEach(System.out::println);

    }

    public static List<String> descartes(List<String>... lists) {
        List<String> tempList = new ArrayList<>();
        for (List<String> list : lists) {
            if (tempList.isEmpty()) {
                tempList = list;
            } else {
                // java8新特性，stream流
                tempList = tempList.stream().flatMap(item -> list.stream().map(item2 -> item + " " + item2)).collect(Collectors.toList());
            }
        }
        return tempList;
    }
}

运行结果：

红色 32G 国产
红色 32G 进口
红色 64G 国产
红色 64G 进口
黑色 32G 国产
黑色 32G 进口
黑色 64G 国产
黑色 64G 进口
金色 32G 国产
金色 32G 进口
金色 64G 国产
金色 64G 进口