动态选择 Spring Data JPA 存储库

Question

我对 Java 中的类型转换和 Spring 概念很陌生。

我有以下实体：

@MappedSuperclass
public class Fish {}

@Entity
public class Whale extends Fish{
  // There is no other property here..
}

@Entity
public class Shark extends Fish{
  // There is no other property here..
}

我已经为这些实体创建了相应的存储库：

public interface WhaleRepository extends CrudRepository<Whale, String> {}

public interface SharkRepository extends CrudRepository<Whale, String> {}

我有一个控制器，根据端点我想保存数据..

@RestController
public class FishController {

    @ResponseBody
    @RequestMapping(value = "{fish-type}")
    public ResponseEntity<Long> create(@PathVariable("fish-type") String fishType, @RequestBody Fish fish){
        if(fishType.equals("whale"}{
           // Error: The method save(S) in the type CrudRepository<Whale,Long> is not applicable for the arguments (Fish)
           new WhaleRepository().save(fish); 
        }
        else if(fishType.equals("shark"}{
           // Error: The method save(S) in the type CrudRepository<Shark,Long> is not applicable for the arguments (Fish)
           new SharkRepository().save(fish);
        }
    }
}

有没有一种方法可以动态选择存储库并保留数据。

Answer 1

是的，你可以。

首先，您必须创建一个抽象实体和存储库。

其次，您需要在 Jackson 中使用继承来在请求中发送通用实体。

例如，类似这样的内容：

FishRepository 类：

public interface FishRepository extends JpaRepository<Fish, Long> {}

Fish 类：

@JsonTypeInfo(
        use = JsonTypeInfo.Id.NAME,
        include = JsonTypeInfo.As.PROPERTY,
        property = "fishType")
@JsonSubTypes({
        @JsonSubTypes.Type(value = Whale.class, name = "whale"),
        @JsonSubTypes.Type(value = Shark.class, name = "shark")
})
@MappedSuperclass
public abstract class Fish {

    @Id
    private Long id;

    public Fish() {
    }

    // getter/setter ..
}

Controller 类：

@PostMapping(value = "/fish")
public String create(@RequestBody Fish fish) {
    ...
    fishRepository.save(fish);
    ...
}

现在您需要在请求中发送fishType。例如：

{
    ...,
    "fishType": "shark"
}

参考文献：

Spring @RequestBody 继承

具有抽象类/继承的 Spring Data Rest Repository