update indexer to perform replacement for existing nodes

Southclaws · Jul 3, 2024 · 8529f57 · 8529f57
1 parent 2802bbc
commit 8529f57
Showing 1 changed file with 31 additions and 11 deletions.
diff --git a/app/services/semdex/weaviate/indexer.go b/app/services/semdex/weaviate/indexer.go
@@ -24,17 +24,37 @@ func (s *weaviateSemdexer) Index(ctx context.Context, object datagraph.Indexable
 		return nil
 	}
 
-	_, err := s.wc.Data().Creator().
-		WithClassName(s.cn.String()).
-		WithID(wid).
-		WithProperties(map[string]any{
-			"datagraph_id":   sid.String(),
-			"datagraph_type": object.GetKind(),
-			"name":           object.GetName(),
-			"content":        content,
-			"props":          object.GetProps(),
-		}).
-		Do(ctx)
+	existing, err := s.wc.Data().ObjectsGetter().WithID(wid).Do(ctx)
+	if err != nil {
+		return fault.Wrap(err, fctx.With(ctx))
+	}
+
+	if len(existing) > 0 {
+		err = s.wc.Data().Updater().
+			WithClassName(s.cn.String()).
+			WithID(wid).
+			WithProperties(map[string]any{
+				"datagraph_id":   sid.String(),
+				"datagraph_type": object.GetKind(),
+				"name":           object.GetName(),
+				"content":        content,
+				"props":          object.GetProps(),
+			}).
+			Do(ctx)
+	} else {
+		_, err = s.wc.Data().Creator().
+			WithClassName(s.cn.String()).
+			WithID(wid).
+			WithProperties(map[string]any{
+				"datagraph_id":   sid.String(),
+				"datagraph_type": object.GetKind(),
+				"name":           object.GetName(),
+				"content":        content,
+				"props":          object.GetProps(),
+			}).
+			Do(ctx)
+	}
+
 	if err != nil {
 		we := &weaviate_errors.WeaviateClientError{}
 		if errors.As(err, &we) {