Skip to content

Commit a6939be

Browse files
committed
Remove List to CqlVector converters.
1 parent 696231d commit a6939be

File tree

3 files changed

+4
-45
lines changed

3 files changed

+4
-45
lines changed

spring-data-cassandra/src/main/java/org/springframework/data/cassandra/core/convert/CassandraConverters.java

Lines changed: 0 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -73,11 +73,9 @@ static Collection<Object> getConvertersToRegister() {
7373

7474
converters.add(VectorToFloatArrayConverter.INSTANCE);
7575
converters.add(VectorToDoubleArrayConverter.INSTANCE);
76-
converters.add(VectorToFloatListConverter.INSTANCE);
7776

7877
converters.add(FloatArrayToVectorConverter.INSTANCE);
7978
converters.add(DoubleArrayToVectorConverter.INSTANCE);
80-
converters.add(NumberListToVectorConverter.INSTANCE);
8179

8280
converters.add(VectorToCqlVectorConverter.INSTANCE);
8381
converters.add(CqlVectorToVectorConverter.INSTANCE);
@@ -307,17 +305,6 @@ public CqlVector<Float> convert(float[] source) {
307305
}
308306
}
309307

310-
@ReadingConverter
311-
public enum NumberListToVectorConverter implements Converter<List<Number>, CqlVector<Number>> {
312-
313-
INSTANCE;
314-
315-
@Override
316-
public CqlVector<Number> convert(List<Number> source) {
317-
return CqlVector.newInstance(source);
318-
}
319-
}
320-
321308
@ReadingConverter
322309
public enum VectorToFloatArrayConverter implements Converter<CqlVector<Number>, float[]> {
323310

@@ -348,18 +335,4 @@ public double[] convert(CqlVector<Number> source) {
348335
}
349336
}
350337

351-
@ReadingConverter
352-
public enum VectorToFloatListConverter implements Converter<CqlVector<Number>, List<Float>> {
353-
354-
INSTANCE;
355-
356-
@Override
357-
public List<Float> convert(CqlVector<Number> source) {
358-
List<Float> values = new ArrayList<>(source.size());
359-
for (int i = 0; i < source.size(); i++) {
360-
values.add(source.get(i).floatValue());
361-
}
362-
return values;
363-
}
364-
}
365338
}

spring-data-cassandra/src/test/java/org/springframework/data/cassandra/core/StatementFactoryUnitTests.java

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -921,16 +921,14 @@ void shouldUpdateCqlVector() {
921921
withVector.id = "foo";
922922
withVector.vector = CqlVector.newInstance(1.2f, 1.3f);
923923
withVector.array = new float[] { 2.2f, 2.3f };
924-
withVector.list = Arrays.asList(3.2f, 3.3f);
925924

926925
SimpleStatement statement = statementFactory.update(withVector, WriteOptions.empty())
927926
.build(ParameterHandling.BY_NAME);
928927

929-
assertThat(statement.getQuery()).isEqualTo("UPDATE withvector SET vector=:p0, array=:p1, list=:p2 WHERE id=:p3");
928+
assertThat(statement.getQuery()).isEqualTo("UPDATE withvector SET vector=:p0, array=:p1 WHERE id=:p2");
930929

931930
assertThat(statement.getNamedValues().get(CqlIdentifier.fromCql("p0"))).isInstanceOf(CqlVector.class).hasToString("[1.2, 1.3]");
932931
assertThat(statement.getNamedValues().get(CqlIdentifier.fromCql("p1"))).isInstanceOf(CqlVector.class).hasToString("[2.2, 2.3]");
933-
assertThat(statement.getNamedValues().get(CqlIdentifier.fromCql("p2"))).isInstanceOf(CqlVector.class).hasToString("[3.2, 3.3]");
934932
}
935933

936934
@Test // GH-1504
@@ -990,6 +988,5 @@ static class WithVector {
990988

991989
@VectorType(dimensions = 12) CqlVector<Number> vector;
992990
@VectorType(dimensions = 12) float[] array;
993-
@VectorType(dimensions = 12) List<Float> list;
994991
}
995992
}

spring-data-cassandra/src/test/java/org/springframework/data/cassandra/core/convert/QueryMapperUnitTests.java

Lines changed: 3 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@
2121

2222
import java.time.LocalDate;
2323
import java.time.LocalTime;
24-
import java.util.Arrays;
2524
import java.util.Collection;
2625
import java.util.Collections;
2726
import java.util.Currency;
@@ -466,7 +465,7 @@ void shouldConvertPrimaryKeyValues() {
466465
assertThat(mappedObject.iterator().next().getPredicate().getValue()).isEqualTo(42L);
467466
}
468467

469-
@Test //
468+
@Test // GH-1504
470469
void shouldConvertVectorValues() {
471470

472471
Filter filter = Filter.from(Criteria.where("array").is(new float[] { 1.1f, 2.2f }));
@@ -478,23 +477,13 @@ void shouldConvertVectorValues() {
478477
assertThat(mappedObject.iterator().next().getPredicate().getValue()).isEqualTo(new float[] { 1.1f, 2.2f });
479478
}
480479

481-
@Test // GH-1504
482-
void shouldConvertVectorValuesFromList() {
483-
484-
Filter filter = Filter.from(Criteria.where("list").is(Arrays.asList(1.1f, 2.2f)));
485-
486-
Filter mappedObject = this.queryMapper.getMappedObject(filter,
487-
this.mappingContext.getRequiredPersistentEntity(WithVector.class));
488-
489-
assertThat(mappedObject.iterator().next().getColumnName()).isEqualTo(ColumnName.from("list"));
490-
assertThat(mappedObject.iterator().next().getPredicate().getValue()).isEqualTo(CqlVector.newInstance(1.1f, 2.2f));
491-
}
492480

493481
@Test // GH-1504
494482
void shouldConvertVectorSelectorFunction() {
495483

496484
Columns columns = Columns.empty();
497-
Columns.FunctionCall similarity = Columns.FunctionCall.from("similarity_cosine", CqlIdentifier.fromCql("array"), Arrays.asList(1.1f, 2.2f));
485+
Columns.FunctionCall similarity = Columns.FunctionCall.from("similarity_cosine", CqlIdentifier.fromCql("array"),
486+
CqlVector.newInstance(1.1f, 2.2f));
498487

499488
Query query = Query.empty().columns(columns.select("array", similarity));
500489

0 commit comments

Comments
 (0)