java ArrayList을 동일한 속성에 따라 그룹화

java ArrayList을 동일한 속성에 따라 그룹화

서론：

SQL을 사용하여 데이터를 쿼리할 때 일반적으로 GROUP BY 문을 사용하여 데이터를 분할할 수 있습니다. 그러나 성능을 고려할 때에는 GROUP BY를 사용하지 않고, 데이터를 먼저 추출한 후 메모리에서 특정 속성에 따라 분할하는 코드를 사용합니다.

코드

public class SkuVo {
  private Long skuId;
  private String productName;
  private Long brandStoreSn;
  public SkuVo(Long skuId, String productName, Long brandStoreSn) {
    super();
    this.skuId = skuId;
    this.productName = productName;
    this.brandStoreSn = brandStoreSn;
  }
  public Long getSkuId() {
    return skuId;
  }
  public void setSkuId(Long skuId) {
    this.skuId = skuId;
  }
  public String getProductName() {
    return productName;
  }
  public void setProductName(String productName) {
    this.productName = productName;
  }
  public Long getBrandStoreSn() {
    return brandStoreSn;
  }
  public void setBrandStoreSn(Long brandStoreSn) {
    this.brandStoreSn = brandStoreSn;
  }
  @Override
  public String toString() {
    return "SkuVo [skuId=" + skuId + ", productName=" + productName + ", brandStoreSn=" + brandStoreSn + "]";
  }
}

데이터를 쿼리에서 추출한 후 List<SkuVo>에 저장된 경우를 가정합니다. skuId에 따라 List<SkuVo>를 분할하는 알고리즘을 사용하여 skuId가 같은 것을 하나의 그룹으로 묶습니다.

분할 알고리즘

public class TestArrayListGroupByKey {
  public static void main(String[] args) {
    /*1、데이터 준비**/
    SkuVo sku1 = new SkuVo(1L,"p1"100L);
    SkuVo sku2 = new SkuVo(2L,"p2"101L);
    SkuVo sku3 = new SkuVo(3L,"p3"102L);
    SkuVo sku4 = new SkuVo(3L,"p4"103L);
    SkuVo sku5 = new SkuVo(2L,"p5"100L);
    SkuVo sku6 = new SkuVo(5L,"p6"100L);
    List<SkuVo> skuVoList = Arrays.asList(new SkuVo [] {sku1,sku2,sku3,sku4,sku5,sku6});
    /*2、그룹화 알고리즘**/
    Map<Long, List<SkuVo>> skuIdMap = new HashMap<>();
    for (SkuVo skuVo : skuVoList) {
      List<SkuVo> tempList = skuIdMap.get(skuVo.getSkuId());
      /*데이터를 가져올 수 없으면, 빈 ArrayList를 새로 생성합니다**/
      if (tempList == null) {
        tempList = new ArrayList<>();
        tempList.add(skuVo);
        skuIdMap.put(skuVo.getSkuId(), tempList);
      }
      else {
        /*어떤 sku가 이미 저장되었으면, 기존 List에 데이터를 추가합니다**/
        tempList.add(skuVo);
      }
    }
    /*3、map을 순회하며 결과를 확인**/
    for(Long skuId : skuIdMap.keySet()){
      System.out.println(skuIdMap.get(skuId));
    }
  }
}

결과는 다음과 같습니다

[SkuVo [skuId=1, productName=p1, brandStoreSn=100]]
[SkuVo [skuId=2, productName=p2, brandStoreSn=101], SkuVo [skuId=2, productName=p5, brandStoreSn=100]]
[SkuVo [skuId=3, productName=p3, brandStoreSn=102], SkuVo [skuId=3, productName=p4, brandStoreSn=103]]
[SkuVo [skuId=5, productName=p6, brandStoreSn=100]]

출력 결과를 보면, 데이터가 skuId에 따라 그룹화되었습니다.

읽어주셔서 감사합니다. 많은 도움이 되길 바랍니다. 많은 분들의 사이트 지원에 감사합니다!