Question

所以，我有一辆VehicleDto：

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

我需要在请求有效负载中使用CarDto或BikeDto。

在post请求有效负载中，将有多个字段是VehicleDto的属性，例如，someId。现在，这个someId也是CarDto和BikeDto的一部分，以及任何其他Dto，它是VehicleDto的孩子。

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

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??
}

超级车辆：

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

儿童班车：

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

我应该如何设计这些问题？

Answer 1

为什么不使用继承代替DTO的关联，就像实体一样？然后将这些DTO映射到entites并返回一些mapper（我更喜欢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类型。我的示例控制器收到VehicleDto并尝试将其作为Bike实体存储在数据库中DtoMapper和VehicleService。最后一步 - 它从数据库中再次读取它并以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观看其他类（实体，服务，存储库）。

将多个子dtos设计成单个请求弹簧启动

1 个答案: