Merge pull request #108 from DataSQRL/fix/auto_register

mark aggregation functions as auto-service

Author: mbroecheler

types/json-type/src/main/java/com/datasqrl/flinkrunner/functions/json/jsonb_array_agg.java

@@ -15,15 +15,19 @@
  */
 package com.datasqrl.flinkrunner.functions.json;
 
+import com.datasqrl.flinkrunner.functions.AutoRegisterSystemFunction;
 import com.datasqrl.flinkrunner.types.json.FlinkJsonType;
+import com.google.auto.service.AutoService;
 import java.util.ArrayList;
 import lombok.SneakyThrows;
 import org.apache.flink.shaded.jackson2.com.fasterxml.jackson.databind.ObjectMapper;
 import org.apache.flink.table.functions.AggregateFunction;
 import org.apache.flink.util.jackson.JacksonMapperFactory;
 
 /** Aggregation function that aggregates JSON objects into a JSON array. */
-public class jsonb_array_agg extends AggregateFunction<FlinkJsonType, ArrayAgg> {
+@AutoService(AutoRegisterSystemFunction.class)
+public class jsonb_array_agg extends AggregateFunction<FlinkJsonType, ArrayAgg>
+    implements AutoRegisterSystemFunction {
 
   private static final ObjectMapper mapper = JacksonMapperFactory.createObjectMapper();
 

types/json-type/src/main/java/com/datasqrl/flinkrunner/functions/json/jsonb_object_agg.java

@@ -15,8 +15,10 @@
  */
 package com.datasqrl.flinkrunner.functions.json;
 
+import com.datasqrl.flinkrunner.functions.AutoRegisterSystemFunction;
 import com.datasqrl.flinkrunner.types.json.FlinkJsonType;
 import com.datasqrl.flinkrunner.types.json.FlinkJsonTypeSerializer;
+import com.google.auto.service.AutoService;
 import java.util.LinkedHashMap;
 import org.apache.flink.shaded.jackson2.com.fasterxml.jackson.databind.ObjectMapper;
 import org.apache.flink.table.annotation.DataTypeHint;
@@ -35,7 +37,9 @@
             value = "RAW",
             bridgedTo = FlinkJsonType.class,
             rawSerializer = FlinkJsonTypeSerializer.class))
-public class jsonb_object_agg extends AggregateFunction<Object, ObjectAgg> {
+@AutoService(AutoRegisterSystemFunction.class)
+public class jsonb_object_agg extends AggregateFunction<Object, ObjectAgg>
+    implements AutoRegisterSystemFunction {
 
   private static final ObjectMapper mapper = JacksonMapperFactory.createObjectMapper();
 

types/vector-type/src/main/java/com/datasqrl/flinkrunner/functions/vector/center.java

@@ -18,14 +18,18 @@
 import static com.datasqrl.flinkrunner.functions.vector.VectorFunctions.VEC_TO_DOUBLE;
 import static com.datasqrl.flinkrunner.functions.vector.VectorFunctions.convert;
 
+import com.datasqrl.flinkrunner.functions.AutoRegisterSystemFunction;
 import com.datasqrl.flinkrunner.types.vector.FlinkVectorType;
+import com.google.auto.service.AutoService;
 import org.apache.flink.table.functions.AggregateFunction;
 
 /**
  * Aggregates vectors by computing the centroid, i.e. summing up all vectors and dividing the
  * resulting vector by the number of vectors.
  */
-public class center extends AggregateFunction<FlinkVectorType, CenterAccumulator> {
+@AutoService(AutoRegisterSystemFunction.class)
+public class center extends AggregateFunction<FlinkVectorType, CenterAccumulator>
+    implements AutoRegisterSystemFunction {
 
   @Override
   public CenterAccumulator createAccumulator() {

types/vector-type/src/main/java/com/datasqrl/flinkrunner/functions/vector/cosine_distance.java

@@ -15,9 +15,12 @@
  */
 package com.datasqrl.flinkrunner.functions.vector;
 
+import com.datasqrl.flinkrunner.functions.AutoRegisterSystemFunction;
 import com.datasqrl.flinkrunner.types.vector.FlinkVectorType;
+import com.google.auto.service.AutoService;
 
 /** Computes the cosine distance between two vectors */
+@AutoService(AutoRegisterSystemFunction.class)
 public class cosine_distance extends cosine_similarity {
 
   @Override