如果父级没有ID

时间:2017-11-22 08:03:31

标签: java hibernate spring-data-jpa one-to-many spring-repositories

我有两个实体:

@Data
@AllArgsConstructor
@NoArgsConstructor
@ToString(exclude = "products")
@Entity
@Table(name = "categories")
public class Category {
  @Id
  @GeneratedValue(strategy = GenerationType.AUTO)
  private Long id;
  private String categoryName;
  private String categoryType;
  @OneToMany(mappedBy = "category", cascade = CascadeType.ALL)
  private List<Product> products;
}

和第二个实体:

@Data
@AllArgsConstructor
@NoArgsConstructor
@Entity
@Table(name = "products")
public class Product {
  @Id
  @GeneratedValue(strategy = GenerationType.AUTO)
  private Long id;
  private String productName;
  private String productDescription;
  @ManyToOne()
  private Category category;
}

我为每个实体都有2个存储库:

@Repository
public interface ProductRepository extends JpaRepository<Product, Long> {
}

@Repository
public interface CategoryRepository extends JpaRepository<Category, Long> {
}

然后我将这个JSON发送给cintroller,我注意保存所有产品:

{
    "products": [{
        "productName": "product1",
        "productDescription": "productDescription1",
        "category": {
            "categoryName": "catName1",
            "categoryType": "type1"
        }
    },
    {
        "productName": "product2",
        "productDescription": "productDescription2",
        "category": {
            "categoryName": "catName1",
            "categoryType": "type1"
        }
    },
    {
        "productName": "product3",
        "productDescription": "productDescription3",
        "category": {
            "categoryName": "catName2",
            "categoryType": "type2"
        }
    }]
}

然后我需要保存它:

@PostMapping(value = "test")
public Category test(@RequestBody() RequestTest products) {

  List<Product> productList = products.getProducts();
  for (Product product : productList) {
  //save all
  }
}

类别没有id,当我得到这个JSON时,我需要创建所有产品并设置类别,但我没有类别ID,如果我做下一个:

for (Product product : productList) {
      Category category = product.getCategory();
      Category save = categoryRepository.save(category);
      product.setCategory(save);
      productRepository.save(product);
}

我为每个产品创建了新的ctegore,我得到了这个:

49  catName1    type1
51  catName1    type1
53  catName2    type2
                                        cat_id
50  productDescription1 product1        49
52  productDescription2 product2        51
54  productDescription3 product3        53

但我需要:

49  catName1    type1
51  catName2    type2
                                        cat_id
50  productDescription1 product1        49  
52  productDescription2 product2        49  
54  productDescription3 product3        51  

3 个答案:

答案 0 :(得分:1)

如果类别对象已存在,您可能希望先找到。

您可以创建一个查询,搜索类别是否存在相同的名称和类型。如果它存在,请使用该对象,该对象也具有ID,否则继续使用当前代码。

答案 1 :(得分:1)

在我看来,最好的方法是通过引入地图来跟踪已保存的类别。

假设categoryName是一个唯一列,那么您可以执行以下操作:

Map<String,Category> categoryMap = new HashMap<>();

for (Product product : productList) {
      String categoryName = product.getCategory().getCategoryName(); 
      Category category = categoryMap.get(categoryName);

      if(category == null){
         category = categoryRepository.save(product.getCategory());
         categoryMap.put(categoryName,category);
      }       

      product.setCategory(save);
      productRepository.save(product);
}

答案 2 :(得分:0)

首先插入所有唯一类别并检查数据库中是否存在相同的类别。 确保类别应该是唯一的。