Получение идентификатора сохраняемого дочернего объекта в отношении один-ко-многим

Question

У меня есть два класса сущностей A и B, которые выглядят следующим образом.

public class A{

    @Id
    @GeneratedValue(strategy = GenerationType.AUTO)
    private Long id;    

    @OneToMany(mappedBy = "a", fetch = FetchType.LAZY, cascade = {CascadeType.ALL})
    private List<B> blist = new ArrayList<B>();

    //Other class members;

}

Класс B:

public class B{

    @Id
    @GeneratedValue(strategy = GenerationType.AUTO)
    private Long id;

    @ManyToOne
    private A a;

    //Other class members;

}

У меня есть метод, который добавляет объект B к объекту A. Я хочу вернуть идентификатор вновь добавленного объекта B.

например:

public Long addBtoA(long aID){

            EntityTransaction tx = myDAO.getEntityManagerTransaction();

            tx.begin();
            A aObject = myDAO.load(aID);
            tx.commit();

            B bObject = new B();

            bObject.addB(bObject);

            tx.begin();
            myDAO.save(aObject);
            tx.commit();

            //Here I want to return the ID of the saved bObject.
            // After saving  aObject it's list of B objects has the newly added bObject with it's id. 
            // What is the best way to get its id?


}

Answer 1

Я не думаю, что принятый ответ правильный. Смотри https://coderanch.com/t/628230/framework/Spring-Data-obtain-id-added

tldr; Вам просто нужно создать репозиторий для дочернего элемента B, чтобы вы могли полностью сохранить дочерний элемент независимо от его родительского элемента. Как только у вас есть сохраненный файл B entity, свяжите его с его родителем A.

Вот пример кода: Todo является родителем, а Comment - дочерним.

@Entity
public class Todo {

    @OneToMany(mappedBy = "todo", cascade = CascadeType.ALL, orphanRemoval = true, fetch = FetchType.LAZY)
    private Set<Comment> comments = new HashSet<>();

    // getters/setters omitted.
}

@Entity
public class Comment {
    @Id
    @GeneratedValue(strategy = GenerationType.AUTO)
    private Long id;

    @ManyToOne
    @JoinColumn(name = "todo_id")
    private Todo todo;

    // getters/setters omitted.
}

Если это было смоделировано в весенних данных, вы создаете 2 хранилища. TodoRepository и CommentRepository, которые Autowired in.

Учитывая конечную точку отдыха, способную получить POST /api/todos/1/comments, чтобы связать новый комментарий с данным идентификатором todo.

    @PostMapping(value = "/api/todos/{todoId}/comments")
    public ResponseEntity<Resource<Comment>> comments(@PathVariable("todoId") Long todoId,
                                                      @RequestBody Comment comment) {

        Todo todo = todoRepository.findOne(todoId);

        // SAVE the comment first so its filled with the id from the DB.
        Comment savedComment = commentRepository.save(comment);

        // Associate the saved comment to the parent Todo.
        todo.addComment(savedComment);

        // Will update the comment with todo id FK.
        todoRepository.save(todo);

        // return payload...
    }

Если вместо этого вы сделали ниже и сохранили предоставленный параметр comment. Единственный способ получить новый комментарий - это перебрать todo.getComments() и найти предоставленный comment, что раздражает и нецелесообразно, если коллекция является Set.

  @PostMapping(value = "/api/todos/{todoId}/comments")
    public ResponseEntity<Resource<Comment>> comments(@PathVariable("todoId") Long todoId,
                                                      @RequestBody Comment comment) {

        Todo todo = todoRepository.findOne(todoId);

        // Associate the supplied comment to the parent Todo.
        todo.addComment(comment);

        // Save the todo which will cascade the save into the child 
        // Comment table providing cascade on the parent is set 
        // to persist or all etc.
        Todo savedTodo = todoRepository.save(todo);

        // You cant do comment.getId
        // Hibernate creates a copy of comment and persists it or something.
        // The only way to get the new id is iterate through 
        // todo.getComments() and find the matching comment which is 
        // impractical especially if the collection is a set. 

        // return payload...
    }

Answer 2

У меня есть метод, который добавляет объект B к объекту A.Я хочу вернуть идентификатор вновь добавленного объекта B.

Тогда просто сделайте это!После того, как новый экземпляр B был сохранен (и измененный сброшен в базу данных), его id был назначен, просто верните его.Вот метод тестирования, который иллюстрирует это поведение:

@Test
public void test_Add_B_To_A() {
    EntityManagerFactory emf = Persistence.createEntityManagerFactory("MyPu");
    EntityManager em = emf.createEntityManager();
    em.getTransaction().begin();

    A a = em.find(A.class, 1L);

    B b = new B();
    A.addToBs(b); // convenient method that manages the bidirectional association

    em.getTransaction().commit(); // pending changes are flushed

    em.close();
    emf.close();

    assertNotNull(b.getId());
}

Кстати, ваш код немного запутан, вам не нужно commit после каждого взаимодействия с EM.

Answer 3

Сначала вы должны сохранить вновь созданный объект, а затем добавить его в свой контейнер.Кроме того, метод save для org.hibernate.Session возвращает идентификатор вновь сохраненного объекта.Так что вам просто нужно обновить свой код и / или DAO, чтобы он вел себя так:

newObject.setContainer(container); // facultative (only if the underlying SGBD forbids null references to the container)
Long id = (Long) hibernateSession.save(newObject); // assuming your identifier is a Long
container.add(newObject);
// now, id contains the id of your new object

В любом случае, для всех объектов с сгенерированными идентификаторами вы всегда можете сделать что-то вроде этого:

hibernateSession.persist(object); // persist returns void...
return object.getId(); // ... but you should have a getId method anyway

Answer 4

В случае, если кто-то не найдет решение по предыдущим комментариям, другой вариант - добавить

@GeneratedValue(strategy = yourChosenStrategy)

поверх идентификатора сущности, которую вы сохраняете (или через ее получатель).В этом случае при вызове persist идентификатор будет автоматически установлен в постоянном объекте.

Надеюсь, это поможет!