fix direct load trace id generator (#400)

Author: suz-yang

src/main/java/com/alipay/oceanbase/rpc/direct_load/ObDirectLoadConnection.java

@@ -64,7 +64,7 @@ public class ObDirectLoadConnection {
 
     ObDirectLoadConnection(ObDirectLoadConnectionFactory connectionFactory) {
         this.connectionFactory = connectionFactory;
-        this.traceId = ObDirectLoadTraceId.generateTraceId();
+        this.traceId = ObDirectLoadTraceIdGenerator.generate();
         this.logger = ObDirectLoadLogger.getLogger(this.traceId);
     }
 
@@ -259,7 +259,7 @@ ObDirectLoadStatement buildStatement(ObDirectLoadStatement.Builder builder)
         ObDirectLoadStatement stmt = null;
         try {
             final ObDirectLoadTraceId traceId = builder.getTraceId() != null ? builder.getTraceId()
-                : ObDirectLoadTraceId.generateTraceId();
+                : ObDirectLoadTraceIdGenerator.generate();
             stmt = createStatement(traceId);
             stmt.init(builder);
         } catch (Exception e) {

src/main/java/com/alipay/oceanbase/rpc/direct_load/ObDirectLoadTraceId.java

@@ -17,9 +17,6 @@
 
 package com.alipay.oceanbase.rpc.direct_load;
 
-import java.net.InetAddress;
-import java.util.concurrent.atomic.AtomicLong;
-
 import com.alipay.oceanbase.rpc.util.ObByteBuf;
 import com.alipay.oceanbase.rpc.util.Serialization;
 
@@ -79,50 +76,9 @@ public int getEncodedSize() {
     }
 
     public static final ObDirectLoadTraceId DEFAULT_TRACE_ID;
-    public static TraceIdGenerator          traceIdGenerator;
 
     static {
         DEFAULT_TRACE_ID = new ObDirectLoadTraceId(0, 0);
-        traceIdGenerator = new TraceIdGenerator();
-    }
-
-    public static ObDirectLoadTraceId generateTraceId() {
-        return traceIdGenerator.generate();
     }
 
-    public static class TraceIdGenerator {
-
-        private final ObDirectLoadLogger logger = ObDirectLoadLogger.getLogger();
-
-        private final long               uniqueId;
-        private AtomicLong               sequence;
-
-        public TraceIdGenerator() {
-            long ip = 0;
-            try {
-                ip = ipToLong(InetAddress.getLocalHost().getHostAddress());
-            } catch (Exception e) {
-                logger.warn("get local host address failed", e);
-            }
-            long port = (long) (Math.random() % 65536) << 32;
-            long isUserRequest = (1l << (32 + 16));
-            long reserved = 0;
-            uniqueId = ip | port | isUserRequest | reserved;
-            sequence = new AtomicLong(0);
-        }
-
-        private static long ipToLong(String strIp) {
-            String[] ip = strIp.split("\\.");
-            return (Long.parseLong(ip[0]) << 24) + (Long.parseLong(ip[1]) << 16)
-                   + (Long.parseLong(ip[2]) << 8) + (Long.parseLong(ip[3]));
-        }
-
-        public ObDirectLoadTraceId generate() {
-            long newSequence = System.currentTimeMillis() * 1000 + sequence.incrementAndGet()
-                               % 1000;
-            return new ObDirectLoadTraceId(uniqueId, newSequence);
-        }
-
-    };
-
 }

src/main/java/com/alipay/oceanbase/rpc/direct_load/ObDirectLoadTraceIdGenerator.java

@@ -0,0 +1,60 @@
+/*-
+ * #%L
+ * com.oceanbase:obkv-table-client
+ * %%
+ * Copyright (C) 2021 - 2025 OceanBase
+ * %%
+ * OBKV Table Client Framework is licensed under Mulan PSL v2.
+ * You can use this software according to the terms and conditions of the Mulan PSL v2.
+ * You may obtain a copy of Mulan PSL v2 at:
+ *          http://license.coscl.org.cn/MulanPSL2
+ * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+ * EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+ * MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+ * See the Mulan PSL v2 for more details.
+ * #L%
+ */
+
+package com.alipay.oceanbase.rpc.direct_load;
+
+import java.net.InetAddress;
+import java.util.concurrent.atomic.AtomicLong;
+
+public class ObDirectLoadTraceIdGenerator {
+
+    private final ObDirectLoadLogger            logger   = ObDirectLoadLogger.getLogger();
+    private static ObDirectLoadTraceIdGenerator instance = new ObDirectLoadTraceIdGenerator();
+
+    private final long                          uniqueId;
+    private AtomicLong                          sequence;
+
+    public ObDirectLoadTraceIdGenerator() {
+        long ip = 0;
+        try {
+            ip = ipToLong(InetAddress.getLocalHost().getHostAddress());
+        } catch (Exception e) {
+            logger.warn("get local host address failed", e);
+        }
+        long port = (long) (Math.random() % 65536) << 32;
+        long isUserRequest = (1l << (32 + 16));
+        long reserved = 0;
+        uniqueId = ip | port | isUserRequest | reserved;
+        sequence = new AtomicLong(0);
+    }
+
+    private static long ipToLong(String strIp) {
+        String[] ip = strIp.split("\\.");
+        return (Long.parseLong(ip[0]) << 24) + (Long.parseLong(ip[1]) << 16)
+               + (Long.parseLong(ip[2]) << 8) + (Long.parseLong(ip[3]));
+    }
+
+    public ObDirectLoadTraceId generateNextId() {
+        long newSequence = System.currentTimeMillis() * 1000 + sequence.incrementAndGet() % 1000;
+        return new ObDirectLoadTraceId(uniqueId, newSequence);
+    }
+
+    public static ObDirectLoadTraceId generate() {
+        return instance.generateNextId();
+    }
+
+}