Support for VectorSeach

Author: clun

src/test/java/org/apache/beam/sdk/io/astra/db/vectorsearch/ProductMapperFactoryFn.java

@@ -39,23 +39,31 @@ public AstraDbMapper<ProductDto> apply(CqlSession cqlSession) {
                 .getProductDao(cqlSession.getKeyspace().get());
 
         // Mapping to Serialize
-        return new AstraDbMapper<ProductDto>() {
-
-            @Override
-            public ProductDto mapRow(Row row) {
-                return new ProductDto(dao.mapRow(row));
-            }
-
-            @Override
-            public CompletionStage<Void> deleteAsync(ProductDto entity) {
-                return dao.deleteAsync(entity.toProduct());
-            }
-
-            @Override
-            public CompletionStage<Void> saveAsync(ProductDto entity) {
-                return dao.saveAsync(entity.toProduct());
-            }
-        };
+        return new AstraDbMapperDelegate(dao);
+    }
+
+    public static class AstraDbMapperDelegate implements AstraDbMapper<ProductDto> {
+
+        ProductDao dao;
+
+        public AstraDbMapperDelegate(ProductDao dao) {
+            this.dao = dao;
+        }
+
+        @Override
+        public ProductDto mapRow(Row row) {
+            return new ProductDto(dao.mapRow(row));
+        }
+
+        @Override
+        public CompletionStage<Void> deleteAsync(ProductDto entity) {
+            return dao.deleteAsync(entity.toProduct());
+        }
+
+        @Override
+        public CompletionStage<Void> saveAsync(ProductDto entity) {
+            return dao.saveAsync(entity.toProduct());
+        }
     }
 
 }