Skip to content

DATAJDBC-557 - Save entity only have id and collections throws exception. #224

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 2 commits into from
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -115,7 +115,7 @@ public <T> Object insert(T instance, Class<T> domainType, Identifier identifier)
identifier.forEach((name, value, type) -> addConvertedPropertyValue(parameterSource, name, value, type));

Object idValue = getIdValueOrNull(instance, persistentEntity);
if (idValue != null) {
if (idValue != null || parameterSource.getIdentifiers().isEmpty()) {

RelationalPersistentProperty idProperty = persistentEntity.getRequiredIdProperty();
addConvertedPropertyValue(parameterSource, idProperty, idValue, idProperty.getColumnName());
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -610,6 +610,10 @@ private UpdateBuilder.UpdateWhereAndOr createBaseUpdate() {
getBindMarker(columnName))) //
.collect(Collectors.toList());

if (assignments.isEmpty()) {
assignments.add(Assignments.value(getIdColumn(), getBindMarker(entity.getIdColumn())));
}

return Update.builder() //
.table(table) //
.set(assignments) //
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,229 @@
/*
* Copyright 2017-2020 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.springframework.data.jdbc.repository;

import static org.assertj.core.api.Assertions.*;

import junit.framework.AssertionFailedError;
import lombok.Data;
import lombok.RequiredArgsConstructor;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Set;

import org.junit.ClassRule;
import org.junit.Rule;
import org.junit.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Import;
import org.springframework.data.annotation.Id;
import org.springframework.data.jdbc.repository.support.JdbcRepositoryFactory;
import org.springframework.data.jdbc.testing.DatabaseProfileValueSource;
import org.springframework.data.jdbc.testing.TestConfiguration;
import org.springframework.data.repository.CrudRepository;
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate;
import org.springframework.test.annotation.IfProfileValue;
import org.springframework.test.annotation.ProfileValueSourceConfiguration;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.rules.SpringClassRule;
import org.springframework.test.context.junit4.rules.SpringMethodRule;
import org.springframework.transaction.annotation.Transactional;

/**
* Very simple use cases for creation and usage of JdbcRepositories.
*
* @author Jens Schauder
* @author Thomas Lang
*/
@ContextConfiguration
@ProfileValueSourceConfiguration(DatabaseProfileValueSource.class)
@Transactional
public class JdbcRepositoryWithCollectionsAndIDOnlyEntityIntegrationTests {

@Configuration
@Import(TestConfiguration.class)
static class Config {

@Autowired JdbcRepositoryFactory factory;

@Bean
Class<?> testClass() {
return JdbcRepositoryWithCollectionsAndIDOnlyEntityIntegrationTests.class;
}

@Bean
DummyEntityRepository dummyEntityRepository() {
return factory.getRepository(DummyEntityRepository.class);
}
}

@ClassRule public static final SpringClassRule classRule = new SpringClassRule();
@Rule public SpringMethodRule methodRule = new SpringMethodRule();

@Autowired NamedParameterJdbcTemplate template;
@Autowired DummyEntityRepository repository;

@Test // DATAJDBC-557
public void saveAndLoadEmptySet() {

DummyEntity entity = repository.save(createDummyEntity());

assertThat(entity.id).isNotNull();

DummyEntity reloaded = repository.findById(entity.id).orElseThrow(AssertionFailedError::new);

assertThat(reloaded.content) //
.isNotNull() //
.isEmpty();
}

@Test // DATAJDBC-557
public void saveAndLoadNonEmptySet() {

Element element1 = new Element();
Element element2 = new Element();

DummyEntity entity = createDummyEntity();
entity.content.add(element1);
entity.content.add(element2);

entity = repository.save(entity);

assertThat(entity.id).isNotNull();
assertThat(entity.content).allMatch(element -> element.id != null);

DummyEntity reloaded = repository.findById(entity.id).orElseThrow(AssertionFailedError::new);

assertThat(reloaded.content) //
.isNotNull() //
.extracting(e -> e.id) //
.containsExactlyInAnyOrder(element1.id, element2.id);
}

@Test // DATAJDBC-557
public void findAllLoadsCollection() {

Element element1 = new Element();
Element element2 = new Element();

DummyEntity entity = createDummyEntity();
entity.content.add(element1);
entity.content.add(element2);

entity = repository.save(entity);

assertThat(entity.id).isNotNull();
assertThat(entity.content).allMatch(element -> element.id != null);

Iterable<DummyEntity> reloaded = repository.findAll();

assertThat(reloaded) //
.extracting(e -> e.id, e -> e.content.size()) //
.containsExactly(tuple(entity.id, entity.content.size()));
}

@Test // DATAJDBC-557
@IfProfileValue(name = "current.database.is.not.mssql", value = "true") // DATAJDBC-278
public void updateSet() {

Element element1 = createElement("one");
Element element2 = createElement("two");
Element element3 = createElement("three");

DummyEntity entity = createDummyEntity();
entity.content.add(element1);
entity.content.add(element2);

entity = repository.save(entity);

entity.content.remove(element1);
element2.content = "two changed";
entity.content.add(element3);

entity = repository.save(entity);

assertThat(entity.id).isNotNull();
assertThat(entity.content).allMatch(element -> element.id != null);

DummyEntity reloaded = repository.findById(entity.id).orElseThrow(AssertionFailedError::new);

// the elements got properly updated and reloaded
assertThat(reloaded.content) //
.isNotNull() //
.extracting(e -> e.id, e -> e.content) //
.containsExactlyInAnyOrder( //
tuple(element2.id, "two changed"), //
tuple(element3.id, "three") //
);

Long count = template.queryForObject("select count(1) from Element", new HashMap<>(), Long.class);
assertThat(count).isEqualTo(2);
}

@Test // DATAJDBC-557
public void deletingWithSet() {

Element element1 = createElement("one");
Element element2 = createElement("two");

DummyEntity entity = createDummyEntity();
entity.content.add(element1);
entity.content.add(element2);

entity = repository.save(entity);

repository.deleteById(entity.id);

assertThat(repository.findById(entity.id)).isEmpty();

Long count = template.queryForObject("select count(1) from Element", new HashMap<>(), Long.class);
assertThat(count).isEqualTo(0);
}

private Element createElement(String content) {

Element element = new Element();
element.content = content;
return element;
}

private static DummyEntity createDummyEntity() {

DummyEntity entity = new DummyEntity();
return entity;
}

interface DummyEntityRepository extends CrudRepository<DummyEntity, Long> {}

@Data
static class DummyEntity {

@Id private Long id;
Set<Element> content = new HashSet<>();

}

@RequiredArgsConstructor
static class Element {

@Id private Long id;
String content;
}

}
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
DROP TABLE element;
DROP TABLE dummy_entity;

CREATE TABLE dummy_entity ( id BIGINT GENERATED BY DEFAULT AS IDENTITY ( START WITH 1 ) PRIMARY KEY, NAME VARCHAR(100));
CREATE TABLE element (id BIGINT GENERATED BY DEFAULT AS IDENTITY (START WITH 1) PRIMARY KEY, content VARCHAR(100), dummy_entity BIGINT);
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
CREATE TABLE dummy_entity ( id BIGINT GENERATED BY DEFAULT AS IDENTITY ( START WITH 1 ) PRIMARY KEY, NAME VARCHAR(100));
CREATE TABLE element (id BIGINT GENERATED BY DEFAULT AS IDENTITY (START WITH 1) PRIMARY KEY, content VARCHAR(100), dummy_entity BIGINT);
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
CREATE TABLE dummy_entity ( id BIGINT GENERATED BY DEFAULT AS IDENTITY ( START WITH 1 ) PRIMARY KEY, NAME VARCHAR(100));
CREATE TABLE element (id BIGINT GENERATED BY DEFAULT AS IDENTITY (START WITH 1) PRIMARY KEY, content VARCHAR(100), dummy_entity BIGINT);
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
CREATE TABLE dummy_entity ( id BIGINT AUTO_INCREMENT PRIMARY KEY, NAME VARCHAR(100));
CREATE TABLE element (id BIGINT AUTO_INCREMENT PRIMARY KEY, content VARCHAR(100), dummy_entity BIGINT);
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
DROP TABLE IF EXISTS dummy_entity;
DROP TABLE IF EXISTS element;
CREATE TABLE dummy_entity ( id BIGINT identity PRIMARY KEY, NAME VARCHAR(100));
CREATE TABLE element (id BIGINT identity PRIMARY KEY, content VARCHAR(100), dummy_entity BIGINT);
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
CREATE TABLE dummy_entity ( id BIGINT AUTO_INCREMENT PRIMARY KEY, NAME VARCHAR(100));
CREATE TABLE element (id BIGINT AUTO_INCREMENT PRIMARY KEY, content VARCHAR(100), dummy_entity BIGINT);
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
DROP TABLE element;
DROP TABLE dummy_entity;
CREATE TABLE dummy_entity ( id SERIAL PRIMARY KEY, NAME VARCHAR(100));
CREATE TABLE element (id SERIAL PRIMARY KEY, content VARCHAR(100), dummy_entity BIGINT);