使用实体类作为视图模型类可能有点棘手,特别是在这种情况下,当实体类之间存在双向一对多关系时Checkin
and Guest
.
让我们首先验证实体类和存储库是否按所示工作。为了进行测试运行,我必须添加@GeneratedValue
对于类中的 id 字段Checkin
.
其他变化:
- Use
Set
代替List
. See https://stackoverflow.com/a/6563037/14072498
- Use
@Getter
and @Setter
代替@Data
对于实体类
关系
我添加了一个测试类CheckinRepositoryTest
下面为了验证代码。
如前所述,使用实体类作为视图模型类可能很棘手,因此下一步将引入两个新的视图模型类:CheckinVM
and GuestVM
,以及一个新的服务等级GuestService
这将负责保存Checkin
with Guest
实例。代码如下所示。
请注意,我只添加了骨架GuestService
班级,我把它留给你来实现。里面的代码CheckinRepositoryTest
表明如何实现它。
有了视图模型类,@JsonProperty(access = JsonProperty.Access.WRITE_ONLY)
应该被删除。
所有部分都启动并运行后,您应该向新视图模型类添加验证,以及一些验证行为的集成测试。如果您遵循 TDD,您将按自己的方式添加测试。
您还应该查看您的控制器方法,并针对成功和失败情况使用适当的状态代码。
嘉宾班
package no.mycompany.myapp.misc;
import com.fasterxml.jackson.annotation.JsonProperty;
import lombok.Getter;
import lombok.Setter;
import javax.persistence.*;
@Getter
@Setter
@Entity
public class Guest {
@Id
@GeneratedValue(strategy = GenerationType.IDENTITY)
private Long guest_id;
private String name;
private String mobile_no;
private String address;
@ManyToOne(fetch = FetchType.LAZY)
@JoinColumn(name = "guest_checkin_id", nullable = false)
@JsonProperty(access = JsonProperty.Access.WRITE_ONLY)
private Checkin checkin;
}
GuestVM 类
package no.mycompany.myapp.misc;
import lombok.Data;
import lombok.NoArgsConstructor;
@Data
@NoArgsConstructor
public class GuestVM {
private long id;
private String name;
private String mobile_no;
private String address;
public GuestVM(Guest guest) {
this.id = guest.getGuest_id();
this.name = guest.getName();
this.mobile_no = guest.getMobile_no();
this.address = guest.getAddress();
}
}
报到类
package no.mycompany.myapp.misc;
import lombok.Getter;
import lombok.Setter;
import javax.persistence.*;
import java.util.HashSet;
import java.util.Set;
@Getter
@Setter
@Entity
public class Checkin {
@Id
@GeneratedValue(strategy = GenerationType.IDENTITY)
private Long id;
private Integer no_of_guests;
@OneToMany(mappedBy = "checkin", cascade = CascadeType.ALL)
private Set<Guest> guests = new HashSet<>();
}
签入VM类
package no.mycompany.myapp.misc;
import lombok.Getter;
import lombok.Setter;
import lombok.NoArgsConstructor;
import java.util.List;
import java.util.stream.Collectors;
@Getter
@Setter
@NoArgsConstructor
public class CheckinVM {
private long id;
private List<GuestVM> guests;
public CheckinVM(Checkin checkin) {
this.id = checkin.getId();
this.guests = checkin.getGuests().stream().map(GuestVM::new).collect(Collectors.toList());
}
}
签入存储库
package no.mycompany.myapp.misc;
import org.springframework.data.jpa.repository.JpaRepository;
public interface CheckinRepository extends JpaRepository<Checkin, Long> { }
签入存储库测试
package no.mycompany.myapp.misc;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.autoconfigure.orm.jpa.DataJpaTest;
import org.springframework.boot.test.autoconfigure.orm.jpa.TestEntityManager;
import static org.assertj.core.api.AssertionsForClassTypes.assertThat;
@DataJpaTest
public class CheckinRepositoryTest {
@Autowired
TestEntityManager testEntityManager;
@Autowired
CheckinRepository checkinRepository;
@Test
public void test() {
// create instances
var checkinInDb = new Checkin();
var guestInDb = new Guest();
// add relations
guestInDb.setCheckin(checkinInDb);
checkinInDb.getGuests().add(guestInDb);
// save check-in
checkinRepository.save(checkinInDb);
// verify that check-in has one guest
var checkin = testEntityManager.find(Checkin.class, checkinInDb.getId());
assertThat(checkin.getGuests().size()).isEqualTo(1);
// verify that guest is connected to a check-in
var guest = testEntityManager.find(Guest.class, guestInDb.getGuest_id());
assertThat(guest.getCheckin()).isNotNull();
}
}
CheckinService 类:我把这个留给你来实现
package no.mycompany.myapp.misc;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
@Service
@RequiredArgsConstructor
public class CheckinService {
private final CheckinRepository checkinRepository;
public CheckinVM saveCheckin(CheckinVM checkin) {
return null; // TODO: implement this
}
}
控制器类
package no.mycompany.myapp.misc;
import lombok.RequiredArgsConstructor;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
@RestController
@RequestMapping("/checkin")
@RequiredArgsConstructor
public class CheckinController {
private final CheckinService checkinService;
@PostMapping("/add")
ResponseEntity<Object> roomCheckin(@RequestBody CheckinVM checkin) {
if (checkinService.saveCheckin(checkin) != null) {
return ResponseEntity.accepted().body("Checkin Successful");
}
return ResponseEntity.unprocessableEntity().body("Failed to Checkin");
}
}