fix[orm]: 修复当数据库key不存在的时候无法set数据

orm/src/main/java/com/zfoo/orm/util/MongoIdUtils.java

@@ -16,6 +16,7 @@ package com.zfoo.orm.util;
 import com.mongodb.client.model.Filters;
 import com.zfoo.orm.OrmContext;
 import com.zfoo.protocol.collection.CollectionUtils;
+import com.zfoo.protocol.util.AssertionUtils;
 import com.zfoo.protocol.util.StringUtils;
 import org.bson.Document;
 
@@ -54,7 +55,8 @@ public abstract class MongoIdUtils {
                 , new Document("$inc", new Document(COUNT, 1L)));
 
         if (document == null) {
-            collection.insertOne(new Document("_id", documentName).append(COUNT, INIT_ID));
+            var result = collection.insertOne(new Document("_id", documentName).append(COUNT, INIT_ID));
+            AssertionUtils.notNull(result.getInsertedId());
             return INIT_ID;
         }
 
@@ -78,20 +80,25 @@ public abstract class MongoIdUtils {
      * @param documentName 档id
      */
     public static void resetIncrementIdFromMongoDefault(String documentName) {
-        setIncrementIdFromMongo(0L, COLLECTION_NAME, documentName);
+        setIncrementIdFromMongo(COLLECTION_NAME, documentName, 0L);
     }
 
-    public static void setIncrementIdFromMongo(long value, Class<?> clazz) {
-        setIncrementIdFromMongo(value, COLLECTION_NAME, StringUtils.uncapitalize(clazz.getSimpleName()));
+    public static void setIncrementIdFromMongo(Class<?> clazz, long value) {
+        setIncrementIdFromMongo(COLLECTION_NAME, StringUtils.uncapitalize(clazz.getSimpleName()), value);
     }
 
-    public static void setIncrementIdFromMongo(long value, String documentName) {
-        setIncrementIdFromMongo(value, COLLECTION_NAME, documentName);
+    public static void setIncrementIdFromMongo(String documentName, long value) {
+        setIncrementIdFromMongo(COLLECTION_NAME, documentName, value);
     }
 
-    public static void setIncrementIdFromMongo(long value, String collectionName, String documentName) {
+    public static void setIncrementIdFromMongo(String collectionName, String documentName, long value) {
         var collection = OrmContext.getOrmManager().getCollection(collectionName);
-        collection.findOneAndUpdate(Filters.eq("_id", documentName), new Document("$set", new Document(COUNT, value)));
+        var document = collection.findOneAndUpdate(Filters.eq("_id", documentName), new Document("$set", new Document(COUNT, value)));
+
+        if (document == null) {
+            var result = collection.insertOne(new Document("_id", documentName).append(COUNT, value));
+            AssertionUtils.notNull(result.getInsertedId());
+        }
     }
 
 

orm/src/test/java/com/zfoo/orm/util/MongoIdUtilsTest.java

@@ -81,8 +81,6 @@ public class MongoIdUtilsTest {
     @Test
     public void set() {
         var context = new ClassPathXmlApplicationContext("application.xml");
-        var id = MongoIdUtils.getIncrementIdFromMongoDefault("myDocument");
-        System.out.println(id);
-        MongoIdUtils.setIncrementIdFromMongo(100, "myDocument");
+        MongoIdUtils.setIncrementIdFromMongo("myDocument", 100);
     }
 }

protocol/src/main/java/com/zfoo/protocol/util/StringUtils.java

@@ -35,7 +35,7 @@ public abstract class StringUtils {
     public static final String SPACE = " ";
 
     public static final String TAB = "    ";
-
+    public static final String TAB_ASCII = "\t";
 
     public static final String COMMA = ",";// [com·ma || 'kɒmə] n.  逗点; 逗号
     public static final String COMMA_REGEX = ",|，";