Skip to content

Commit e2f8bf2

Browse files
committed
DATACASS-825 - Polishing.
Reoder methods. Remove unused code and imports. Original pull request: #181.
1 parent 418c743 commit e2f8bf2

File tree

4 files changed

+150
-131
lines changed

4 files changed

+150
-131
lines changed

spring-data-cassandra/src/main/java/org/springframework/data/cassandra/repository/support/FindByIdQuery.java

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,6 @@ static boolean hasCompositeKeys(Iterable<?> ids) {
9696
if (id instanceof MapId) {
9797

9898
MapId mapId = (MapId) id;
99-
Iterator<String> iterator = mapId.keySet().iterator();
10099

101100
if (mapId.size() > 1) {
102101
return true;

spring-data-cassandra/src/main/java/org/springframework/data/cassandra/repository/support/SimpleCassandraRepository.java

Lines changed: 63 additions & 54 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@
2222
import java.util.List;
2323
import java.util.Optional;
2424

25-
import org.jetbrains.annotations.NotNull;
2625
import org.springframework.data.cassandra.core.CassandraOperations;
2726
import org.springframework.data.cassandra.core.CassandraTemplate;
2827
import org.springframework.data.cassandra.core.InsertOptions;
@@ -36,8 +35,6 @@
3635
import org.springframework.data.mapping.context.AbstractMappingContext;
3736
import org.springframework.util.Assert;
3837

39-
import com.datastax.oss.driver.api.querybuilder.insert.Insert;
40-
4138
/**
4239
* Repository base implementation for Cassandra.
4340
*
@@ -75,6 +72,10 @@ public SimpleCassandraRepository(CassandraEntityInformation<T, ID> metadata, Cas
7572
this.mappingContext = operations.getConverter().getMappingContext();
7673
}
7774

75+
// -------------------------------------------------------------------------
76+
// Methods from CrudRepository
77+
// -------------------------------------------------------------------------
78+
7879
/* (non-Javadoc)
7980
* @see org.springframework.data.repository.CrudRepository#save(S)
8081
*/
@@ -112,34 +113,6 @@ public <S extends T> List<S> saveAll(Iterable<S> entities) {
112113
return result;
113114
}
114115

115-
/* (non-Javadoc)
116-
* @see org.springframework.data.cassandra.repository.TypedIdCassandraRepository#insert(java.lang.Object)
117-
*/
118-
@Override
119-
public <S extends T> S insert(S entity) {
120-
121-
Assert.notNull(entity, "Entity must not be null");
122-
123-
return this.operations.insert(entity);
124-
}
125-
126-
/* (non-Javadoc)
127-
* @see org.springframework.data.cassandra.repository.TypedIdCassandraRepository#insert(java.lang.Iterable)
128-
*/
129-
@Override
130-
public <S extends T> List<S> insert(Iterable<S> entities) {
131-
132-
Assert.notNull(entities, "The given Iterable of entities must not be null");
133-
134-
List<S> result = new ArrayList<>();
135-
136-
for (S entity : entities) {
137-
result.add(this.operations.insert(entity));
138-
}
139-
140-
return result;
141-
}
142-
143116
/* (non-Javadoc)
144117
* @see org.springframework.data.repository.CrudRepository#findById(java.lang.Object)
145118
*/
@@ -166,14 +139,6 @@ public boolean existsById(ID id) {
166139
return this.operations.exists(id, this.entityInformation.getJavaType());
167140
}
168141

169-
/* (non-Javadoc)
170-
* @see org.springframework.data.repository.CrudRepository#count()
171-
*/
172-
@Override
173-
public long count() {
174-
return this.operations.count(this.entityInformation.getJavaType());
175-
}
176-
177142
/* (non-Javadoc)
178143
* @see org.springframework.data.repository.CrudRepository#findAll()
179144
*/
@@ -198,14 +163,11 @@ public List<T> findAllById(Iterable<ID> ids) {
198163
}
199164

200165
/* (non-Javadoc)
201-
* @see org.springframework.data.cassandra.repository.CassandraRepository#findAll(org.springframework.data.domain.Pageable)
166+
* @see org.springframework.data.repository.CrudRepository#count()
202167
*/
203168
@Override
204-
public Slice<T> findAll(Pageable pageable) {
205-
206-
Assert.notNull(pageable, "Pageable must not be null");
207-
208-
return this.operations.slice(Query.empty().pageRequest(pageable), this.entityInformation.getJavaType());
169+
public long count() {
170+
return this.operations.count(this.entityInformation.getJavaType());
209171
}
210172

211173
/* (non-Javadoc)
@@ -231,16 +193,8 @@ public void delete(T entity) {
231193
}
232194

233195
/* (non-Javadoc)
234-
* @see org.springframework.data.repository.CrudRepository#deleteAll(java.lang.Iterable)
196+
* @see org.springframework.data.repository.CrudRepository#deleteAllById(java.lang.Iterable)
235197
*/
236-
@Override
237-
public void deleteAll(Iterable<? extends T> entities) {
238-
239-
Assert.notNull(entities, "The given Iterable of entities must not be null");
240-
241-
entities.forEach(this.operations::delete);
242-
}
243-
244198
@Override
245199
public void deleteAllById(Iterable<? extends ID> ids) {
246200

@@ -253,6 +207,17 @@ public void deleteAllById(Iterable<? extends ID> ids) {
253207
this.operations.delete(createIdsInQuery(ids), this.entityInformation.getJavaType());
254208
}
255209

210+
/* (non-Javadoc)
211+
* @see org.springframework.data.repository.CrudRepository#deleteAll(java.lang.Iterable)
212+
*/
213+
@Override
214+
public void deleteAll(Iterable<? extends T> entities) {
215+
216+
Assert.notNull(entities, "The given Iterable of entities must not be null");
217+
218+
entities.forEach(this.operations::delete);
219+
}
220+
256221
/* (non-Javadoc)
257222
* @see org.springframework.data.repository.CrudRepository#deleteAll()
258223
*/
@@ -261,7 +226,51 @@ public void deleteAll() {
261226
this.operations.truncate(this.entityInformation.getJavaType());
262227
}
263228

229+
// -------------------------------------------------------------------------
230+
// Methods from CassandraRepository
231+
// -------------------------------------------------------------------------
232+
233+
/* (non-Javadoc)
234+
* @see org.springframework.data.cassandra.repository.CassandraRepository#findAll(org.springframework.data.domain.Pageable)
235+
*/
236+
@Override
237+
public Slice<T> findAll(Pageable pageable) {
238+
239+
Assert.notNull(pageable, "Pageable must not be null");
240+
241+
return this.operations.slice(Query.empty().pageRequest(pageable), this.entityInformation.getJavaType());
242+
}
243+
244+
/* (non-Javadoc)
245+
* @see org.springframework.data.cassandra.repository.TypedIdCassandraRepository#insert(java.lang.Object)
246+
*/
247+
@Override
248+
public <S extends T> S insert(S entity) {
249+
250+
Assert.notNull(entity, "Entity must not be null");
251+
252+
return this.operations.insert(entity);
253+
}
254+
255+
/* (non-Javadoc)
256+
* @see org.springframework.data.cassandra.repository.TypedIdCassandraRepository#insert(java.lang.Iterable)
257+
*/
258+
@Override
259+
public <S extends T> List<S> insert(Iterable<S> entities) {
260+
261+
Assert.notNull(entities, "The given Iterable of entities must not be null");
262+
263+
List<S> result = new ArrayList<>();
264+
265+
for (S entity : entities) {
266+
result.add(this.operations.insert(entity));
267+
}
268+
269+
return result;
270+
}
271+
264272
private Query createIdsInQuery(Iterable<? extends ID> ids) {
273+
265274
FindByIdQuery mapIdQuery = FindByIdQuery.forIds(ids);
266275
List<Object> idCollection = mapIdQuery.getIdCollection();
267276
String idField = mapIdQuery.getIdProperty();

0 commit comments

Comments
 (0)