java - 将多个子DTO设计成单个请求Spring Boot

Question

所以，我有一个 VehicleDto:

class VehicleDto {
    private String someId
    private String vType;
    private CarDto car;
    private BikeDto bike;
}

我需要在请求负载中包含 CarDto 或 BikeDto。

在post请求负载中会有多个字段，这些字段是VehicleDto的属性，例如这里的someId。现在，这个 someId 也是 CarDto 和 BikeDto 以及作为 VehicleDto 子级的任何其他 Dto 的一部分。

所以当我尝试保存到数据库时，我遇到了一些问题。

if (vehicleDto.getVType().equals("CAR")) {
    this.saveCar(vehicleDto);
}

private boolean saveCar(TicketSoldCreateDto ticketSoldCreateDto) {
    CarDto carDto = ticketSoldCreateDto.getCar();
    carDto is mapped to Car model
    // Now how do I map the rest of the fields in vehicleDto to Car model??
}

super 车辆:

@MappedSuperclass
@Data
public abstract class Vehicle extends AbstractBaseEntity {
// fields same as vehicleDto
}

child 车:

@Entity
@Data
public class Car extends Vehicle {
// Some fields
}

我应该如何设计这样的问题？

Answer 1

为什么不使用继承而不是 DTO 的关联，就像实体一样？ 然后使用一些映射器将这些 DTO 映射到实体并返回(我更喜欢 mapstruct )。

我做了一个完整的例子on github .

DTO:

@Data
@JsonTypeInfo(use = JsonTypeInfo.Id.NAME, include = JsonTypeInfo.As.PROPERTY, property = "type")
@JsonSubTypes({
        @JsonSubTypes.Type(value = CarDto.class, name = "car"),
        @JsonSubTypes.Type(value = BikeDto.class, name = "bike")
})
public class VehicleDto {
    private Long id;
    private String type;
    private Integer modelYear;
}

@Data
public class BikeDto extends VehicleDto {
    private String frameType;
}

@Data
public class CarDto extends VehicleDto {
    private Boolean isCabriolet;
}

@JsonTypeInfo和 @JsonSubTypes需要自动解析 Controller 中的 DTO 类型。我的示例 Controller 接收 VehicleDto 并尝试使用 DtoMapper 和 VehicleService 将其作为 Bike 实体存储在数据库中。最后一步 - 它再次从数据库中读取并使用 BikeDto 进行响应。

@Controller
public class SampleController {

    @Autowired
    private VehicleService vehicleService;

    @Autowired
    private DtoMapper mapper;

    @PostMapping("/testDto")
    @ResponseBody
    @Transactional
    public BikeDto testDto(@RequestBody VehicleDto vehicleDto) {

        if (vehicleDto instanceof BikeDto)
            vehicleService.saveBike(mapper.toBikeEntity((BikeDto) vehicleDto));

        return mapper.toBikeDto(vehicleService.getBike(vehicleDto.getId()));
    }
}

对于 DtoMapper，我使用了 Mapstruct，它将我的 Bike 实体转换为 BikeDto 并返回:

@Mapper(componentModel = "spring")
@Component
public interface DtoMapper {

    @Mapping(target = "type", constant = "bike")
    BikeDto toBikeDto(Bike entity);

    Bike toBikeEntity(BikeDto dto);
}

最后，这个例子的测试类。它将 BikeDto 作为 POST 正文传递，并期望它返回。

@RunWith(SpringRunner.class)
@SpringBootTest
@ActiveProfiles("scratch")
public class SampleDataJpaApplicationTests {

    @Autowired
    private WebApplicationContext context;
    private MockMvc mvc;

    @Before
    public void setUp() {
        this.mvc = MockMvcBuilders.webAppContextSetup(this.context).build();
    }

    @Test
    public void testDto() throws Exception {

        BikeDto bikeDto = new BikeDto();
        bikeDto.setId(42L);
        bikeDto.setType("bike");
        bikeDto.setModelYear(2019);
        bikeDto.setFrameType("carbon");

        Gson gson = new Gson();
        String json = gson.toJson(bikeDto);

        this.mvc.perform(post("/testDto").contentType(MediaType.APPLICATION_JSON).content(json))
                .andExpect(status().isOk())
                .andExpect(content().json(json));
    }
}

POST (BikeDto) 正文:

{
 "id":42,
 "type":"bike",
 "modelYear":2019,
 "frameType":"carbon"
}

您可以在完整示例 on github 中观看其他类(实体、服务、存储库) .