Merge pull request #363 from CorvusYe/master

feat: expanding the `updateEdgeByIdBatchSelective` interface in `NebulaDaoBasic`

Author: CorvusYe

ngbatis-demo/src/test/java/ye/weicheng/ngbatis/demo/NebulaBasicDaoTests.java

@@ -25,6 +25,7 @@
 import ye.weicheng.ngbatis.demo.pojo.Like;
 import ye.weicheng.ngbatis.demo.pojo.LikeWithRank;
 import ye.weicheng.ngbatis.demo.pojo.Person;
+import ye.weicheng.ngbatis.demo.pojo.edge.Follow;
 import ye.weicheng.ngbatis.demo.pojo.vertex.Player;
 import ye.weicheng.ngbatis.demo.repository.TestRepository;
 
@@ -540,6 +541,37 @@ public void upsertEdgeSelective() {
 
     repository.upsertEdgeSelective(person1, likeWithRank, person2);
   }
+  
+  @Test
+  public void updateEdgeByIdBatchSelective() {
+    Person p1 = new Person();
+    Person p2 = new Person();
+    
+    repository.insert(p1);
+    repository.insert(p2);
+
+    Follow p1FollowP2 = new Follow();
+    Follow p2FollowP1 = new Follow();
+    
+    repository.insertEdge(p1, p1FollowP2, p2);
+    repository.insertEdge(p2, p2FollowP1, p1);
+    
+    p1FollowP2.setSrcId(p1.getName());
+    p1FollowP2.setDstId(p2.getName());
+    
+    p2FollowP1.setSrcId(p2.getName());
+    p2FollowP1.setDstId(p1.getName());
+
+    List<Follow> follows = new ArrayList<>();
+    follows.add(p1FollowP2);
+    follows.add(p2FollowP1);
+    follows.forEach((l) -> l.setDegree(3));
+    
+    repository.updateEdgeByIdBatchSelective(follows);
+    
+    repository.deleteWithEdgeById(p1.getName());
+    repository.deleteWithEdgeById(p2.getName());
+  }
 
 
   @Test

src/main/java/org/nebula/contrib/ngbatis/binding/beetl/functions/GetDstIdFn.java

@@ -0,0 +1,25 @@
+package org.nebula.contrib.ngbatis.binding.beetl.functions;
+
+import java.lang.reflect.Field;
+import org.nebula.contrib.ngbatis.annotations.DstId;
+import org.nebula.contrib.ngbatis.annotations.SrcId;
+import org.nebula.contrib.ngbatis.utils.ReflectUtil;
+
+/**
+ * @author yeweicheng
+ * @since 2025-12-13 8:56
+ * <br>Now is history!
+ */
+public class GetDstIdFn extends AbstractFunction<Object, Void, Void, Void, Void, Void> {
+
+  @Override
+  public Object call(Object o) {
+    try {
+      Field dstIdField = ReflectUtil.getAnnoField(o.getClass(), DstId.class);
+      Object dstId = ReflectUtil.getValue(o, dstIdField);
+      return fnCall(valueFmtFn, dstId);
+    } catch (Exception e) {
+      throw new RuntimeException(e);
+    }
+  }
+}

src/main/java/org/nebula/contrib/ngbatis/binding/beetl/functions/GetSrcIdFn.java

@@ -0,0 +1,24 @@
+package org.nebula.contrib.ngbatis.binding.beetl.functions;
+
+import java.lang.reflect.Field;
+import org.nebula.contrib.ngbatis.annotations.SrcId;
+import org.nebula.contrib.ngbatis.utils.ReflectUtil;
+
+/**
+ * @author yeweicheng
+ * @since 2025-12-13 9:02
+ * <br>Now is history!
+ */
+public class GetSrcIdFn extends AbstractFunction<Object, Void, Void, Void, Void, Void> {
+
+  @Override
+  public Object call(Object o) {
+    try {
+      Field srcIdField = ReflectUtil.getAnnoField(o.getClass(), SrcId.class);
+      Object srcId = ReflectUtil.getValue(o, srcIdField);
+      return fnCall(valueFmtFn, srcId);
+    } catch (Exception e) {
+      throw new RuntimeException(e);
+    }
+  }
+}

src/main/java/org/nebula/contrib/ngbatis/proxy/NebulaDaoBasic.java

@@ -444,6 +444,12 @@ default void upsertEdgeSelective(Object src, Object edge, Object dst) {
     ClassModel classModel = getClassModel(this.getClass());
     MapperProxy.invoke(classModel, methodModel, src, edge, dst);
   }
+  
+  default void updateEdgeByIdBatchSelective(List<?> edgeList) {
+    MethodModel methodModel = getMethodModel();
+    ClassModel classModel = getClassModel(this.getClass());
+    MapperProxy.invoke(classModel, methodModel, edgeList);
+  }
 
   /**
    * 提供开始节点的id、结束节点的id 与 关系名，判断是否已经建立关系

src/main/resources/NebulaDaoBasic.xml

@@ -415,6 +415,26 @@
         @}
         @}
     </insert>
+  
+    <update id="updateEdgeByIdBatchSelective">
+      @for ( row in ng_args[0] ) {
+        @var kv = ng.kv( row,  '', true, false );
+        @if ( isNotEmpty( @kv.columns ) ) {
+          @var rank = ng.id( row );
+          @var srcId = ng.srcId( row );
+          @var dstId = ng.dstId( row );
+          @var edgeName = ng.tagName( row );
+          UPDATE  EDGE ON `${ edgeName }` ${ srcId }-> ${ dstId } ${ isNotEmpty( rank ) ? ('@' + rank) : ''  }
+          SET
+          @for (col in @kv.columns ) {
+          @var vIdx = @kv.columns.indexOf( col );
+          @var val = @kv.values.get(@vIdx);
+            `${ col }` = ${ val } ${ colLP.last ? '' : ',' }
+          @}
+          ;
+        @}
+      @}
+    </update>
 
     <select id="existsEdge">
         MATCH (n)-[r: ${ p1 }]-(n2)

src/main/resources/beetl.properties

@@ -12,6 +12,8 @@ FN.ng.pkName =org.nebula.contrib.ngbatis.binding.beetl.functions.PkNameFn
 FN.ng.entityType =org.nebula.contrib.ngbatis.binding.beetl.functions.EntityTypeFn
 FN.ng.fieldNames =org.nebula.contrib.ngbatis.binding.beetl.functions.FieldNamesFn
 FN.ng.id =org.nebula.contrib.ngbatis.binding.beetl.functions.IdFn
+FN.ng.srcId =org.nebula.contrib.ngbatis.binding.beetl.functions.GetSrcIdFn
+FN.ng.dstId =org.nebula.contrib.ngbatis.binding.beetl.functions.GetDstIdFn
 FN.ng.kv =org.nebula.contrib.ngbatis.binding.beetl.functions.KvFn
 FN.ng.join = org.nebula.contrib.ngbatis.binding.beetl.functions.JoinFn
 FN.ng.ifStringLike = org.nebula.contrib.ngbatis.binding.beetl.functions.IfStringLike