feat: add deleteByIdBatch interface (#2)

Author: shbone

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

@@ -331,6 +331,31 @@ public void deleteById() {
     int row = repository.deleteById("赵小洋");
     System.out.println(row);
   }
+
+  @Test
+  public void deleteByIdBatch() {
+    long now = System.currentTimeMillis();
+    Person person1 = new Person();
+    person1.setName("UBB" + now);
+
+    Person person2 = new Person();
+    person2.setName("UBB" + (now + 1));
+
+    Person person3 = new Person();
+    person3.setName("UBB" + (now + 2));
+
+    List<Person> people = new ArrayList<>();
+    people.add(person1);
+    people.add(person2);
+    people.add(person3);
+    repository.insertBatch(people);
+
+    List<String> peopleIds = new ArrayList<>();
+    peopleIds.add(person1.getName());
+    peopleIds.add(person2.getName());
+    peopleIds.add(person3.getName());
+    Assert.equals(repository.deleteByIdBatch(peopleIds),1);
+  }
   // endregion
 
   // region graph special

pom.xml

@@ -32,6 +32,10 @@
             <name>dieyi</name>
             <email>[email protected]</email>
         </developer>
+        <developer>
+            <name>shbone</name>
+            <email>[email protected]</email>
+        </developer>
     </developers>
 
     <properties>

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

@@ -314,7 +314,22 @@ default int deleteById(I id) {
     ResultSet resultSet = (ResultSet) MapperProxy.invoke(classModel, methodModel, id);
     return resultSet.isSucceeded() ? 1 : 0;
   }
+
   // endregion
+  /**
+   * <p>通过 主键批量删除当前记录</p>
+   *
+   * @param ids 表记录主键列表
+   * @return 是否删除成功，成功 1，失败 0
+   */
+  default int deleteByIdBatch(List<I> ids) {
+    MethodModel methodModel = getMethodModel();
+    methodModel.setReturnType(ResultSet.class);
+    methodModel.setResultType(ResultSet.class);
+    ClassModel classModel = getClassModel(this.getClass());
+    ResultSet resultSet = (ResultSet) MapperProxy.invoke(classModel, methodModel, ids);
+    return resultSet.isSucceeded() ? 1 : 0;
+  }
 
   // region graph special
   /**
@@ -332,19 +347,18 @@ default void insertEdge(@NotNull Object v1, @NotNull Object e, @NotNull Object v
     ClassModel classModel = getClassModel(this.getClass());
     MapperProxy.invoke(classModel, methodModel, v1, e, v2);
   }
+
   /**
    * @Author sunhb
    * @Description 根据三元组列表的头结点，尾节点和尾节点进行插入
    * @Date 2023/10/10 上午11:03
-   * @Param
-   * @param triplets 三元组列表
-   * @return void
    **/
-  default void insertEdgeBatch(List triplets){
+  default void insertEdgeBatch(List triplets) {
     MethodModel methodModel = getMethodModel();
     ClassModel classModel = getClassModel(this.getClass());
     MapperProxy.invoke(classModel, methodModel, triplets);
   }
+
   /**
    * 根据三元组值, 插入关系
    * <p>Selective: 仅处理非空字段</p>
@@ -462,7 +476,9 @@ default <E> E startNode(Class<E> startType, Class<?> edgeType, I endId) {
    * Find the shortest path by srcId and dstId.
    * @param srcId the start node's id
    * @param dstId the end node's id
-   * @return Shortest path list. entities containing vertext in path. If you want to obtain attributes within an entity, you need to use “with prop” in the nGQL.
+   * @return Shortest path list. entities containing vertext in path.
+   *         If you want to obtain attributes within an entity,
+   *         you need to use “with prop” in the nGQL.
    */
   default List<NgPath<I>> shortestPath(@Param("srcId") I srcId, @Param("dstId") I dstId) {
     MethodModel methodModel = getMethodModel();
@@ -471,6 +487,8 @@ default List<NgPath<I>> shortestPath(@Param("srcId") I srcId, @Param("dstId") I
     ClassModel classModel = getClassModel(this.getClass());
     return (List<NgPath<I>>) MapperProxy.invoke(classModel, methodModel, srcId, dstId);
   }
+
+
   // endregion
 
 }

src/main/resources/NebulaDaoBasic.xml

@@ -285,7 +285,11 @@
         DELETE VERTEX ${ ng.valueFmt( p0 ) }
     </delete>
     <!--endregion-->
-
+    <delete id="deleteByIdBatch">
+        @for ( v in ng_args[0] ) {
+            DELETE VERTEX ${ ng.valueFmt( v )};
+        @}
+    </delete>
     <!--region graph special-->
     <insert id="insertEdge">
         @var kv = ng.kv( ng_args[1], '', null, null, false );