DATAJDBC-96 - Database based ID generation.

No longer using batch inserts, which won't (easily) hold up for long anyway, since we need to decide between update and insert anyway. If one wants to support batch insert one also has to check if any autoid columns are present, because those seem not to work with batch inserts with many or at least some drivers.

Related ticket: SPR-1836.
Original pull request: #5.

src/main/java/org/springframework/data/jdbc/mapping/model/JdbcPersistentEntity.java

@@ -50,4 +50,8 @@ public class JdbcPersistentEntity<T> extends BasicPersistentEntity<T, JdbcPersis
 	public Object getIdValue(T instance) {
 		return getPropertyAccessor(instance).getProperty(getIdProperty());
 	}
+
+	public void setId(T instance, Object value) {
+		getPropertyAccessor(instance).setProperty(getIdProperty(),value);
+	}
 }

src/main/java/org/springframework/data/jdbc/repository/SimpleJdbcRepository.java

@@ -28,6 +28,8 @@ import org.springframework.data.repository.CrudRepository;
 import org.springframework.jdbc.core.namedparam.MapSqlParameterSource;
 import org.springframework.jdbc.core.namedparam.NamedParameterJdbcOperations;
 import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate;
+import org.springframework.jdbc.support.GeneratedKeyHolder;
+import org.springframework.jdbc.support.KeyHolder;
 
 /**
  * @author Jens Schauder
@@ -50,22 +52,24 @@ public class SimpleJdbcRepository<T, ID extends Serializable> implements CrudRep
 	}
 
 	@Override
-	public <S extends T> S save(S entity) {
+	public <S extends T> S save(S instance) {
 
-		template.update(sql.getInsert(), getPropertyMap(entity));
+		KeyHolder holder = new GeneratedKeyHolder();
 
-		return entity;
+		template.update(
+				sql.getInsert(),
+				new MapSqlParameterSource(getPropertyMap(instance)),
+				holder);
+
+		entity.setId(instance, holder.getKey());
+
+		return instance;
 	}
 
 	@Override
 	public <S extends T> Iterable<S> save(Iterable<S> entities) {
 
-		Map<String, ?>[] batchValues = StreamSupport
-				.stream(entities.spliterator(), false)
-				.map(i -> getPropertyMap(i))
-				.toArray(size -> new Map[size]);
-
-		template.batchUpdate(sql.getInsert(), batchValues);
+		entities.forEach(this::save);
 
 		return entities;
 	}