我有两个实体:
@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
答案 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)
首先插入所有唯一类别并检查数据库中是否存在相同的类别。 确保类别应该是唯一的。