fix: Datetime index creation (#37)

Author: Anush008

README.md

@@ -14,7 +14,7 @@
     <a href="https://github.com/qdrant/java-client/actions/workflows/cd.yml"><img src="https://github.com/qdrant/java-client/actions/workflows/cd.yml/badge.svg?branch=master" alt="Tests"></a>
     <a href="https://github.com/qdrant/java-client/blob/master/LICENSE"><img src="https://img.shields.io/badge/License-Apache%202.0-success" alt="Apache 2.0 License"></a>
     <a href="https://qdrant.to/discord"><img src="https://img.shields.io/badge/Discord-Qdrant-5865F2.svg?logo=discord" alt="Discord"></a>
-    <a href="https://qdrant.to/roadmap"><img src="https://img.shields.io/badge/Roadmap-2023-bc1439.svg" alt="Roadmap 2023"></a>
+    <a href="https://qdrant.to/roadmap"><img src="https://img.shields.io/badge/Roadmap-2024-bc1439.svg" alt="Roadmap 2024"></a>
 </p>
 
 # Qdrant Java Client
@@ -184,4 +184,4 @@ List<ScoredPoint> points = client.searchAsync(SearchPoints.newBuilder()
 
 ## ⚖️ LICENSE
 
-Apache 2.0 © [2023](https://github.com/qdrant/java-client/blob/master/LICENSE)
+Apache 2.0 © [2024](https://github.com/qdrant/java-client/blob/master/LICENSE)

src/main/java/io/qdrant/client/QdrantClient.java

@@ -2212,6 +2212,9 @@ public ListenableFuture<UpdateResult> createPayloadIndexAsync(
 			case Bool:
 				requestBuilder.setFieldType(FieldType.FieldTypeBool);
 				break;
+			case Datetime:
+				requestBuilder.setFieldType(FieldType.FieldTypeDatetime);
+				break;
 			default:
 				throw new IllegalArgumentException("Invalid schemaType: '" + schemaType + "'");
 		}
@@ -2224,8 +2227,22 @@ public ListenableFuture<UpdateResult> createPayloadIndexAsync(
 			requestBuilder.setOrdering(WriteOrdering.newBuilder().setType(ordering).build());
 		}
 
-		logger.debug("Create payload field index for '{}' in '{}'", field, collectionName);
-		ListenableFuture<PointsOperationResponse> future = getPoints(timeout).createFieldIndex(requestBuilder.build());
+		return createPayloadIndexAsync(requestBuilder.build(), timeout);
+	}
+
+	/**
+	 * Creates a payload field index in a collection.
+	 *
+	 * @param request The create field index request.
+	 * @param timeout The timeout for the call.
+	 * @return a new instance of {@link ListenableFuture}
+	 */
+	public ListenableFuture<UpdateResult> createPayloadIndexAsync(
+		CreateFieldIndexCollection request,
+		@Nullable Duration timeout
+	) {
+		logger.debug("Create payload field index for '{}' in '{}'", request.getFieldName(), request.getCollectionName());
+		ListenableFuture<PointsOperationResponse> future = getPoints(timeout).createFieldIndex(request);
 		addLogFailureCallback(future, "Create payload field index");
 		return Futures.transform(future, PointsOperationResponse::getResult, MoreExecutors.directExecutor());
 	}

src/test/java/io/qdrant/client/PointsTest.java

@@ -99,7 +99,7 @@ public void retrieve() throws ExecutionException, InterruptedException {
 		assertEquals(1, points.size());
 		RetrievedPoint point = points.get(0);
 		assertEquals(id(9), point.getId());
-		assertEquals(ImmutableSet.of("foo", "bar"), point.getPayloadMap().keySet());
+		assertEquals(ImmutableSet.of("foo", "bar", "date"), point.getPayloadMap().keySet());
 		assertEquals(value("goodbye"), point.getPayloadMap().get("foo"));
 		assertEquals(value(2), point.getPayloadMap().get("bar"));
 		assertEquals(Vectors.getDefaultInstance(), point.getVectors());
@@ -141,7 +141,7 @@ public void setPayload() throws ExecutionException, InterruptedException {
 		assertEquals(1, points.size());
 		RetrievedPoint point = points.get(0);
 		assertEquals(id(9), point.getId());
-		assertEquals(ImmutableSet.of("foo", "bar"), point.getPayloadMap().keySet());
+		assertEquals(ImmutableSet.of("foo", "bar", "date"), point.getPayloadMap().keySet());
 		assertEquals(value("some bar"), point.getPayloadMap().get("bar"));
 		assertEquals(value("goodbye"), point.getPayloadMap().get("foo"));
 	}
@@ -188,7 +188,7 @@ public void deletePayload() throws ExecutionException, InterruptedException {
 		assertEquals(1, points.size());
 		RetrievedPoint point = points.get(0);
 		assertEquals(id(9), point.getId());
-		assertEquals(ImmutableSet.of("bar"), point.getPayloadMap().keySet());
+		assertEquals(ImmutableSet.of("bar", "date"), point.getPayloadMap().keySet());
 		assertEquals(value("some bar"), point.getPayloadMap().get("bar"));
 	}
 
@@ -239,6 +239,26 @@ public void createFieldIndex() throws ExecutionException, InterruptedException {
 		assertEquals(PayloadSchemaType.Integer, collectionInfo.getPayloadSchemaMap().get("bar").getDataType());
 	}
 
+	@Test
+	public void createDatetimeFieldIndex() throws ExecutionException, InterruptedException {
+		createAndSeedCollection(testName);
+
+		UpdateResult result = client.createPayloadIndexAsync(
+			testName,
+			"date",
+			PayloadSchemaType.Datetime,
+			null,
+			null,
+			null,
+			null).get();
+
+		assertEquals(UpdateStatus.Completed, result.getStatus());
+
+		CollectionInfo collectionInfo = client.getCollectionInfoAsync(testName).get();
+		assertEquals(ImmutableSet.of("date"), collectionInfo.getPayloadSchemaMap().keySet());
+		assertEquals(PayloadSchemaType.Datetime, collectionInfo.getPayloadSchemaMap().get("date").getDataType());
+	}
+
 	@Test
 	public void deleteFieldIndex() throws ExecutionException, InterruptedException {
 		createAndSeedCollection(testName);
@@ -277,7 +297,7 @@ public void search() throws ExecutionException, InterruptedException {
 		assertEquals(1, points.size());
 		ScoredPoint point = points.get(0);
 		assertEquals(id(9), point.getId());
-		assertEquals(ImmutableSet.of("foo", "bar"), point.getPayloadMap().keySet());
+		assertEquals(ImmutableSet.of("foo", "bar", "date"), point.getPayloadMap().keySet());
 		assertEquals(value("goodbye"), point.getPayloadMap().get("foo"));
 		assertEquals(value(2), point.getPayloadMap().get("bar"));
 		assertFalse(point.getVectors().hasVector());
@@ -589,15 +609,17 @@ private void createAndSeedCollection(String collectionName) throws ExecutionExce
 				.setVectors(VectorsFactory.vectors(ImmutableList.of(3.5f, 4.5f)))
 				.putAllPayload(ImmutableMap.of(
 					"foo", value("hello"),
-					"bar", value(1)
+					"bar", value(1),
+					"date", value("2021-01-01T00:00:00Z")
 				))
 				.build(),
 			PointStruct.newBuilder()
 				.setId(id(9))
 				.setVectors(VectorsFactory.vectors(ImmutableList.of(10.5f, 11.5f)))
 				.putAllPayload(ImmutableMap.of(
 					"foo", value("goodbye"),
-					"bar", value(2)
+					"bar", value(2),
+					"date", value("2024-01-02T00:00:00Z")
 				))
 				.build()
 		)).get();