Added List<Float> constructor

ankane · ankane · commit 9e2050eb7fe5 · 2023-11-28T22:59:29.000-08:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,3 +1,7 @@
+## 0.1.4 (unreleased)
+
+- Added `List<Float>` constructor
+
 ## 0.1.3 (2023-07-18)
 
 - Added support for binary representation
diff --git a/src/main/java/com/pgvector/PGvector.java b/src/main/java/com/pgvector/PGvector.java
@@ -4,6 +4,7 @@
 import java.sql.Connection;
 import java.sql.SQLException;
 import java.util.Arrays;
+import java.util.List;
 import org.postgresql.PGConnection;
 import org.postgresql.util.ByteConverter;
 import org.postgresql.util.PGBinaryObject;
@@ -30,6 +31,22 @@ public PGvector(float[] v) {
         vec = v;
     }
 
+    /**
+     * Constructor
+     */
+    public PGvector(List<Float> v) {
+        this();
+        if (v == null) {
+            vec = null;
+        } else {
+            vec = new float[v.size()];
+            int i = 0;
+            for (Float f : v) {
+                vec[i++] = f.floatValue();
+            }
+        }
+    }
+
     /**
      * Constructor
      */
diff --git a/src/test/java/com/pgvector/PGvectorTest.java b/src/test/java/com/pgvector/PGvectorTest.java
@@ -1,6 +1,7 @@
 package com.pgvector;
 
 import java.sql.SQLException;
+import java.util.Arrays;
 import com.pgvector.PGvector;
 import org.junit.jupiter.api.Test;
 
@@ -19,6 +20,13 @@ void testStringConstructor() throws SQLException {
         assertArrayEquals(new float[] {1, 2, 3}, vec.toArray());
     }
 
+    @Test
+    void testListConstructor() {
+        Float[] a = new Float[] {Float.valueOf(1), Float.valueOf(2), Float.valueOf(3)};
+        PGvector vec = new PGvector(Arrays.asList(a));
+        assertArrayEquals(new float[] {1, 2, 3}, vec.toArray());
+    }
+
     @Test
     void testGetValue() throws SQLException {
         PGvector vec = new PGvector(new float[] {1, 2, 3});
