Raggruppamento di ArrayList in Java per un attributo comune

Raggruppamento di ArrayList in Java per un attributo comune

Introduzione:

Di solito, quando si esegue una query SQL per ottenere un elenco di dati, si può utilizzare la clausola GROUP BY di SQL per raggruppare i dati, ma a volte, per motivi di prestazioni, non si utilizza GROUP BY, ma si prelevano i dati prima e si raggruppano in memoria in base a un attributo specifico utilizzando il codice.

Codice

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 + "]";
  }
}

Supponiamo di aver estratto un lotto di dati da una query di dati, esistenti in List<SkuVo>. Utilizzare un algoritmo per raggruppare List<SkuVo> in base a skuId, i quali con lo stesso skuId vengono raggruppati insieme.

Algoritmo di raggruppamento

public class TestArrayListGroupByKey {
  public static void main(String[] args) {
    /*1、Preparazione dei dati**/
    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、Algoritmo di raggruppamento**/
    Map<Long, List<SkuVo>> skuIdMap = new HashMap<>();
    for (SkuVo skuVo : skuVoList) {
      List<SkuVo> tempList = skuIdMap.get(skuVo.getSkuId());
      /*Se non è possibile ottenere i dati, crea direttamente un nuovo ArrayList vuoto**/
      if (tempList == null) {
        tempList = new ArrayList<>();
        tempList.add(skuVo);
        skuIdMap.put(skuVo.getSkuId(), tempList);
      }
      else {
        /*Se un某个sku prima è stato memorizzato, aggiungi direttamente i dati alla lista originale**/
        tempList.add(skuVo);
      }
    }
    /*3、Esplora la mappa, verifica i risultati**/
    for(Long skuId : skuIdMap.keySet()){
      System.out.println(skuIdMap.get(skuId));
    }
  }
}

I risultati sono i seguenti

[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]]

Dalla visualizzazione dei risultati, i dati sono stati raggruppati secondo skuId.

Grazie per aver letto, spero che possa aiutare tutti, grazie per il supporto al nostro sito!