Skip to content

Commit 86d5ce7

Browse files
Alchuang22-devCRZbulabula
Alchuang22-dev
authored and
CRZbulabula
committed
[AINode] Update AINodeClient for DataNode to borrow (#16647)
(cherry picked from commit d49d7dd)
1 parent d52bf55 commit 86d5ce7

File tree

25 files changed

+486
-450
lines changed

25 files changed

+486
-450
lines changed

‎iotdb-core/confignode/src/main/java/org/apache/iotdb/confignode/client/async/AsyncAINodeHeartbeatClientPool.java‎

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -21,10 +21,10 @@
2121

2222
import org.apache.iotdb.ainode.rpc.thrift.TAIHeartbeatReq;
2323
import org.apache.iotdb.common.rpc.thrift.TEndPoint;
24-
import org.apache.iotdb.commons.client.ClientPoolFactory;
2524
import org.apache.iotdb.commons.client.IClientManager;
26-
import org.apache.iotdb.commons.client.ainode.AsyncAINodeServiceClient;
2725
import org.apache.iotdb.confignode.client.async.handlers.heartbeat.AINodeHeartbeatHandler;
26+
import org.apache.iotdb.db.protocol.client.AINodeClientFactory;
27+
import org.apache.iotdb.db.protocol.client.ainode.AsyncAINodeServiceClient;
2828

2929
public class AsyncAINodeHeartbeatClientPool {
3030

@@ -33,8 +33,7 @@ public class AsyncAINodeHeartbeatClientPool {
3333
private AsyncAINodeHeartbeatClientPool() {
3434
clientManager =
3535
new IClientManager.Factory<TEndPoint, AsyncAINodeServiceClient>()
36-
.createClientManager(
37-
new ClientPoolFactory.AsyncAINodeHeartbeatServiceClientPoolFactory());
36+
.createClientManager(new AINodeClientFactory.AINodeHeartbeatClientPoolFactory());
3837
}
3938

4039
public void getAINodeHeartBeat(

‎iotdb-core/confignode/src/main/java/org/apache/iotdb/confignode/consensus/request/read/model/ShowModelPlan.java‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,9 +19,9 @@
1919

2020
package org.apache.iotdb.confignode.consensus.request.read.model;
2121

22+
import org.apache.iotdb.ainode.rpc.thrift.TShowModelsReq;
2223
import org.apache.iotdb.confignode.consensus.request.ConfigPhysicalPlanType;
2324
import org.apache.iotdb.confignode.consensus.request.read.ConfigPhysicalReadPlan;
24-
import org.apache.iotdb.confignode.rpc.thrift.TShowModelReq;
2525

2626
import java.util.Objects;
2727

@@ -33,7 +33,7 @@ public ShowModelPlan() {
3333
super(ConfigPhysicalPlanType.ShowModel);
3434
}
3535

36-
public ShowModelPlan(final TShowModelReq showModelReq) {
36+
public ShowModelPlan(final TShowModelsReq showModelReq) {
3737
super(ConfigPhysicalPlanType.ShowModel);
3838
if (showModelReq.isSetModelId()) {
3939
this.modelName = showModelReq.getModelId();

‎iotdb-core/confignode/src/main/java/org/apache/iotdb/confignode/manager/ConfigManager.java‎

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,14 @@
2020
package org.apache.iotdb.confignode.manager;
2121

2222
import org.apache.iotdb.ainode.rpc.thrift.IDataSchema;
23+
import org.apache.iotdb.ainode.rpc.thrift.TLoadModelReq;
24+
import org.apache.iotdb.ainode.rpc.thrift.TShowAIDevicesResp;
25+
import org.apache.iotdb.ainode.rpc.thrift.TShowLoadedModelsReq;
26+
import org.apache.iotdb.ainode.rpc.thrift.TShowLoadedModelsResp;
27+
import org.apache.iotdb.ainode.rpc.thrift.TShowModelsReq;
28+
import org.apache.iotdb.ainode.rpc.thrift.TShowModelsResp;
2329
import org.apache.iotdb.ainode.rpc.thrift.TTrainingReq;
30+
import org.apache.iotdb.ainode.rpc.thrift.TUnloadModelReq;
2431
import org.apache.iotdb.common.rpc.thrift.TAINodeConfiguration;
2532
import org.apache.iotdb.common.rpc.thrift.TAINodeLocation;
2633
import org.apache.iotdb.common.rpc.thrift.TConfigNodeLocation;
@@ -41,8 +48,6 @@
4148
import org.apache.iotdb.common.rpc.thrift.TTimePartitionSlot;
4249
import org.apache.iotdb.commons.auth.AuthException;
4350
import org.apache.iotdb.commons.auth.entity.PrivilegeUnion;
44-
import org.apache.iotdb.commons.client.ainode.AINodeClient;
45-
import org.apache.iotdb.commons.client.ainode.AINodeClientManager;
4651
import org.apache.iotdb.commons.cluster.NodeStatus;
4752
import org.apache.iotdb.commons.cluster.NodeType;
4853
import org.apache.iotdb.commons.conf.CommonConfig;
@@ -210,7 +215,6 @@
210215
import org.apache.iotdb.confignode.rpc.thrift.TGetTriggerTableResp;
211216
import org.apache.iotdb.confignode.rpc.thrift.TGetUDFTableResp;
212217
import org.apache.iotdb.confignode.rpc.thrift.TGetUdfTableReq;
213-
import org.apache.iotdb.confignode.rpc.thrift.TLoadModelReq;
214218
import org.apache.iotdb.confignode.rpc.thrift.TMigrateRegionReq;
215219
import org.apache.iotdb.confignode.rpc.thrift.TNodeVersionInfo;
216220
import org.apache.iotdb.confignode.rpc.thrift.TPermissionInfoResp;
@@ -223,7 +227,6 @@
223227
import org.apache.iotdb.confignode.rpc.thrift.TSchemaPartitionTableResp;
224228
import org.apache.iotdb.confignode.rpc.thrift.TSetDataNodeStatusReq;
225229
import org.apache.iotdb.confignode.rpc.thrift.TSetSchemaTemplateReq;
226-
import org.apache.iotdb.confignode.rpc.thrift.TShowAIDevicesResp;
227230
import org.apache.iotdb.confignode.rpc.thrift.TShowAINodesResp;
228231
import org.apache.iotdb.confignode.rpc.thrift.TShowCQResp;
229232
import org.apache.iotdb.confignode.rpc.thrift.TShowClusterResp;
@@ -232,10 +235,6 @@
232235
import org.apache.iotdb.confignode.rpc.thrift.TShowDataNodes4InformationSchemaResp;
233236
import org.apache.iotdb.confignode.rpc.thrift.TShowDataNodesResp;
234237
import org.apache.iotdb.confignode.rpc.thrift.TShowDatabaseResp;
235-
import org.apache.iotdb.confignode.rpc.thrift.TShowLoadedModelReq;
236-
import org.apache.iotdb.confignode.rpc.thrift.TShowLoadedModelResp;
237-
import org.apache.iotdb.confignode.rpc.thrift.TShowModelReq;
238-
import org.apache.iotdb.confignode.rpc.thrift.TShowModelResp;
239238
import org.apache.iotdb.confignode.rpc.thrift.TShowPipePluginReq;
240239
import org.apache.iotdb.confignode.rpc.thrift.TShowPipeReq;
241240
import org.apache.iotdb.confignode.rpc.thrift.TShowPipeResp;
@@ -253,12 +252,13 @@
253252
import org.apache.iotdb.confignode.rpc.thrift.TSubscribeReq;
254253
import org.apache.iotdb.confignode.rpc.thrift.TThrottleQuotaResp;
255254
import org.apache.iotdb.confignode.rpc.thrift.TTimeSlotList;
256-
import org.apache.iotdb.confignode.rpc.thrift.TUnloadModelReq;
257255
import org.apache.iotdb.confignode.rpc.thrift.TUnsetSchemaTemplateReq;
258256
import org.apache.iotdb.confignode.rpc.thrift.TUnsubscribeReq;
259257
import org.apache.iotdb.confignode.rpc.thrift.TUpdateModelInfoReq;
260258
import org.apache.iotdb.consensus.common.DataSet;
261259
import org.apache.iotdb.consensus.exception.ConsensusException;
260+
import org.apache.iotdb.db.protocol.client.ainode.AINodeClient;
261+
import org.apache.iotdb.db.protocol.client.ainode.AINodeClientManager;
262262
import org.apache.iotdb.db.schemaengine.template.Template;
263263
import org.apache.iotdb.db.schemaengine.template.TemplateAlterOperationType;
264264
import org.apache.iotdb.db.schemaengine.template.alter.TemplateAlterOperationUtil;
@@ -2753,19 +2753,19 @@ public TSStatus unloadModel(TUnloadModelReq req) {
27532753
}
27542754

27552755
@Override
2756-
public TShowModelResp showModel(TShowModelReq req) {
2756+
public TShowModelsResp showModel(TShowModelsReq req) {
27572757
TSStatus status = confirmLeader();
27582758
return status.getCode() == TSStatusCode.SUCCESS_STATUS.getStatusCode()
27592759
? modelManager.showModel(req)
2760-
: new TShowModelResp(status);
2760+
: new TShowModelsResp(status);
27612761
}
27622762

27632763
@Override
2764-
public TShowLoadedModelResp showLoadedModel(TShowLoadedModelReq req) {
2764+
public TShowLoadedModelsResp showLoadedModel(TShowLoadedModelsReq req) {
27652765
TSStatus status = confirmLeader();
27662766
return status.getCode() == TSStatusCode.SUCCESS_STATUS.getStatusCode()
27672767
? modelManager.showLoadedModel(req)
2768-
: new TShowLoadedModelResp(status, Collections.emptyMap());
2768+
: new TShowLoadedModelsResp(status, Collections.emptyMap());
27692769
}
27702770

27712771
@Override

‎iotdb-core/confignode/src/main/java/org/apache/iotdb/confignode/manager/IManager.java‎

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,13 @@
1919

2020
package org.apache.iotdb.confignode.manager;
2121

22+
import org.apache.iotdb.ainode.rpc.thrift.TLoadModelReq;
23+
import org.apache.iotdb.ainode.rpc.thrift.TShowAIDevicesResp;
24+
import org.apache.iotdb.ainode.rpc.thrift.TShowLoadedModelsReq;
25+
import org.apache.iotdb.ainode.rpc.thrift.TShowLoadedModelsResp;
26+
import org.apache.iotdb.ainode.rpc.thrift.TShowModelsReq;
27+
import org.apache.iotdb.ainode.rpc.thrift.TShowModelsResp;
28+
import org.apache.iotdb.ainode.rpc.thrift.TUnloadModelReq;
2229
import org.apache.iotdb.common.rpc.thrift.TConfigNodeLocation;
2330
import org.apache.iotdb.common.rpc.thrift.TDataNodeLocation;
2431
import org.apache.iotdb.common.rpc.thrift.TFlushReq;
@@ -124,7 +131,6 @@
124131
import org.apache.iotdb.confignode.rpc.thrift.TGetTriggerTableResp;
125132
import org.apache.iotdb.confignode.rpc.thrift.TGetUDFTableResp;
126133
import org.apache.iotdb.confignode.rpc.thrift.TGetUdfTableReq;
127-
import org.apache.iotdb.confignode.rpc.thrift.TLoadModelReq;
128134
import org.apache.iotdb.confignode.rpc.thrift.TMigrateRegionReq;
129135
import org.apache.iotdb.confignode.rpc.thrift.TPermissionInfoResp;
130136
import org.apache.iotdb.confignode.rpc.thrift.TPipeConfigTransferReq;
@@ -136,7 +142,6 @@
136142
import org.apache.iotdb.confignode.rpc.thrift.TSchemaPartitionTableResp;
137143
import org.apache.iotdb.confignode.rpc.thrift.TSetDataNodeStatusReq;
138144
import org.apache.iotdb.confignode.rpc.thrift.TSetSchemaTemplateReq;
139-
import org.apache.iotdb.confignode.rpc.thrift.TShowAIDevicesResp;
140145
import org.apache.iotdb.confignode.rpc.thrift.TShowAINodesResp;
141146
import org.apache.iotdb.confignode.rpc.thrift.TShowCQResp;
142147
import org.apache.iotdb.confignode.rpc.thrift.TShowClusterResp;
@@ -145,10 +150,6 @@
145150
import org.apache.iotdb.confignode.rpc.thrift.TShowDataNodes4InformationSchemaResp;
146151
import org.apache.iotdb.confignode.rpc.thrift.TShowDataNodesResp;
147152
import org.apache.iotdb.confignode.rpc.thrift.TShowDatabaseResp;
148-
import org.apache.iotdb.confignode.rpc.thrift.TShowLoadedModelReq;
149-
import org.apache.iotdb.confignode.rpc.thrift.TShowLoadedModelResp;
150-
import org.apache.iotdb.confignode.rpc.thrift.TShowModelReq;
151-
import org.apache.iotdb.confignode.rpc.thrift.TShowModelResp;
152153
import org.apache.iotdb.confignode.rpc.thrift.TShowPipePluginReq;
153154
import org.apache.iotdb.confignode.rpc.thrift.TShowPipeReq;
154155
import org.apache.iotdb.confignode.rpc.thrift.TShowPipeResp;
@@ -162,7 +163,6 @@
162163
import org.apache.iotdb.confignode.rpc.thrift.TStartPipeReq;
163164
import org.apache.iotdb.confignode.rpc.thrift.TStopPipeReq;
164165
import org.apache.iotdb.confignode.rpc.thrift.TSubscribeReq;
165-
import org.apache.iotdb.confignode.rpc.thrift.TUnloadModelReq;
166166
import org.apache.iotdb.confignode.rpc.thrift.TUnsetSchemaTemplateReq;
167167
import org.apache.iotdb.confignode.rpc.thrift.TUnsubscribeReq;
168168
import org.apache.iotdb.consensus.common.DataSet;
@@ -874,10 +874,10 @@ TDataPartitionTableResp getOrCreateDataPartition(
874874
TSStatus unloadModel(TUnloadModelReq req);
875875

876876
/** Return the model table. */
877-
TShowModelResp showModel(TShowModelReq req);
877+
TShowModelsResp showModel(TShowModelsReq req);
878878

879879
/** Return the loaded model instances. */
880-
TShowLoadedModelResp showLoadedModel(TShowLoadedModelReq req);
880+
TShowLoadedModelsResp showLoadedModel(TShowLoadedModelsReq req);
881881

882882
/** Return all available AI devices. */
883883
TShowAIDevicesResp showAIDevices();

‎iotdb-core/confignode/src/main/java/org/apache/iotdb/confignode/manager/ModelManager.java‎

Lines changed: 19 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -19,14 +19,15 @@
1919

2020
package org.apache.iotdb.confignode.manager;
2121

22+
import org.apache.iotdb.ainode.rpc.thrift.TLoadModelReq;
23+
import org.apache.iotdb.ainode.rpc.thrift.TShowAIDevicesResp;
2224
import org.apache.iotdb.ainode.rpc.thrift.TShowLoadedModelsReq;
2325
import org.apache.iotdb.ainode.rpc.thrift.TShowLoadedModelsResp;
2426
import org.apache.iotdb.ainode.rpc.thrift.TShowModelsReq;
2527
import org.apache.iotdb.ainode.rpc.thrift.TShowModelsResp;
28+
import org.apache.iotdb.ainode.rpc.thrift.TUnloadModelReq;
2629
import org.apache.iotdb.common.rpc.thrift.TEndPoint;
2730
import org.apache.iotdb.common.rpc.thrift.TSStatus;
28-
import org.apache.iotdb.commons.client.ainode.AINodeClient;
29-
import org.apache.iotdb.commons.client.ainode.AINodeClientManager;
3031
import org.apache.iotdb.commons.client.exception.ClientManagerException;
3132
import org.apache.iotdb.commons.model.ModelInformation;
3233
import org.apache.iotdb.commons.model.ModelStatus;
@@ -40,15 +41,10 @@
4041
import org.apache.iotdb.confignode.rpc.thrift.TDropModelReq;
4142
import org.apache.iotdb.confignode.rpc.thrift.TGetModelInfoReq;
4243
import org.apache.iotdb.confignode.rpc.thrift.TGetModelInfoResp;
43-
import org.apache.iotdb.confignode.rpc.thrift.TLoadModelReq;
44-
import org.apache.iotdb.confignode.rpc.thrift.TShowAIDevicesResp;
45-
import org.apache.iotdb.confignode.rpc.thrift.TShowLoadedModelReq;
46-
import org.apache.iotdb.confignode.rpc.thrift.TShowLoadedModelResp;
47-
import org.apache.iotdb.confignode.rpc.thrift.TShowModelReq;
48-
import org.apache.iotdb.confignode.rpc.thrift.TShowModelResp;
49-
import org.apache.iotdb.confignode.rpc.thrift.TUnloadModelReq;
5044
import org.apache.iotdb.confignode.rpc.thrift.TUpdateModelInfoReq;
5145
import org.apache.iotdb.consensus.exception.ConsensusException;
46+
import org.apache.iotdb.db.protocol.client.ainode.AINodeClient;
47+
import org.apache.iotdb.db.protocol.client.ainode.AINodeClientManager;
5248
import org.apache.iotdb.rpc.TSStatusCode;
5349

5450
import org.slf4j.Logger;
@@ -124,42 +120,43 @@ public TSStatus unloadModel(TUnloadModelReq req) {
124120
}
125121
}
126122

127-
public TShowModelResp showModel(final TShowModelReq req) {
123+
public TShowModelsResp showModel(final TShowModelsReq req) {
128124
try (AINodeClient client = getAINodeClient()) {
129125
TShowModelsReq showModelsReq = new TShowModelsReq();
130126
if (req.isSetModelId()) {
131127
showModelsReq.setModelId(req.getModelId());
132128
}
133129
TShowModelsResp resp = client.showModels(showModelsReq);
134-
TShowModelResp res =
135-
new TShowModelResp().setStatus(new TSStatus(TSStatusCode.SUCCESS_STATUS.getStatusCode()));
130+
TShowModelsResp res =
131+
new TShowModelsResp()
132+
.setStatus(new TSStatus(TSStatusCode.SUCCESS_STATUS.getStatusCode()));
136133
res.setModelIdList(resp.getModelIdList());
137134
res.setModelTypeMap(resp.getModelTypeMap());
138135
res.setCategoryMap(resp.getCategoryMap());
139136
res.setStateMap(resp.getStateMap());
140137
return res;
141138
} catch (Exception e) {
142139
LOGGER.warn("Failed to show models due to", e);
143-
return new TShowModelResp()
140+
return new TShowModelsResp()
144141
.setStatus(
145142
new TSStatus(TSStatusCode.AI_NODE_INTERNAL_ERROR.getStatusCode())
146143
.setMessage(e.getMessage()));
147144
}
148145
}
149146

150-
public TShowLoadedModelResp showLoadedModel(final TShowLoadedModelReq req) {
147+
public TShowLoadedModelsResp showLoadedModel(final TShowLoadedModelsReq req) {
151148
try (AINodeClient client = getAINodeClient()) {
152149
TShowLoadedModelsReq showModelsReq =
153150
new TShowLoadedModelsReq().setDeviceIdList(req.getDeviceIdList());
154151
TShowLoadedModelsResp resp = client.showLoadedModels(showModelsReq);
155-
TShowLoadedModelResp res =
156-
new TShowLoadedModelResp()
152+
TShowLoadedModelsResp res =
153+
new TShowLoadedModelsResp()
157154
.setStatus(new TSStatus(TSStatusCode.SUCCESS_STATUS.getStatusCode()));
158155
res.setDeviceLoadedModelsMap(resp.getDeviceLoadedModelsMap());
159156
return res;
160157
} catch (Exception e) {
161158
LOGGER.warn("Failed to show loaded models due to", e);
162-
return new TShowLoadedModelResp()
159+
return new TShowLoadedModelsResp()
163160
.setStatus(
164161
new TSStatus(TSStatusCode.AI_NODE_INTERNAL_ERROR.getStatusCode())
165162
.setMessage(e.getMessage()));
@@ -235,7 +232,11 @@ private AINodeClient getAINodeClient() throws NoAvailableAINodeException, Client
235232
}
236233
TEndPoint targetAINodeEndPoint =
237234
new TEndPoint(aiNodeInfo.get(0).getInternalAddress(), aiNodeInfo.get(0).getInternalPort());
238-
return AINodeClientManager.getInstance().borrowClient(targetAINodeEndPoint);
235+
try {
236+
return AINodeClientManager.getInstance().borrowClient(targetAINodeEndPoint);
237+
} catch (Exception e) {
238+
throw new RuntimeException(e);
239+
}
239240
}
240241

241242
public List<Integer> getModelDistributions(String modelName) {

‎iotdb-core/confignode/src/main/java/org/apache/iotdb/confignode/procedure/impl/model/CreateModelProcedure.java‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,6 @@
2121

2222
import org.apache.iotdb.common.rpc.thrift.TAINodeConfiguration;
2323
import org.apache.iotdb.common.rpc.thrift.TSStatus;
24-
import org.apache.iotdb.commons.client.ainode.AINodeClient;
25-
import org.apache.iotdb.commons.client.ainode.AINodeClientManager;
2624
import org.apache.iotdb.commons.exception.ainode.LoadModelException;
2725
import org.apache.iotdb.commons.model.ModelInformation;
2826
import org.apache.iotdb.commons.model.ModelStatus;
@@ -36,6 +34,8 @@
3634
import org.apache.iotdb.confignode.procedure.state.model.CreateModelState;
3735
import org.apache.iotdb.confignode.procedure.store.ProcedureType;
3836
import org.apache.iotdb.consensus.exception.ConsensusException;
37+
import org.apache.iotdb.db.protocol.client.ainode.AINodeClient;
38+
import org.apache.iotdb.db.protocol.client.ainode.AINodeClientManager;
3939
import org.apache.iotdb.rpc.TSStatusCode;
4040

4141
import org.apache.tsfile.utils.ReadWriteIOUtils;

‎iotdb-core/confignode/src/main/java/org/apache/iotdb/confignode/procedure/impl/model/DropModelProcedure.java‎

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,17 +19,18 @@
1919

2020
package org.apache.iotdb.confignode.procedure.impl.model;
2121

22+
import org.apache.iotdb.ainode.rpc.thrift.TDeleteModelReq;
2223
import org.apache.iotdb.common.rpc.thrift.TAINodeConfiguration;
2324
import org.apache.iotdb.common.rpc.thrift.TSStatus;
24-
import org.apache.iotdb.commons.client.ainode.AINodeClient;
25-
import org.apache.iotdb.commons.client.ainode.AINodeClientManager;
2625
import org.apache.iotdb.commons.model.exception.ModelManagementException;
2726
import org.apache.iotdb.confignode.consensus.request.write.model.DropModelPlan;
2827
import org.apache.iotdb.confignode.procedure.env.ConfigNodeProcedureEnv;
2928
import org.apache.iotdb.confignode.procedure.exception.ProcedureException;
3029
import org.apache.iotdb.confignode.procedure.impl.node.AbstractNodeProcedure;
3130
import org.apache.iotdb.confignode.procedure.state.model.DropModelState;
3231
import org.apache.iotdb.confignode.procedure.store.ProcedureType;
32+
import org.apache.iotdb.db.protocol.client.ainode.AINodeClient;
33+
import org.apache.iotdb.db.protocol.client.ainode.AINodeClientManager;
3334
import org.apache.iotdb.rpc.TSStatusCode;
3435

3536
import org.apache.thrift.TException;
@@ -115,7 +116,7 @@ private void dropModelOnAINode(ConfigNodeProcedureEnv env) {
115116
.getRegisteredAINode(nodeId)
116117
.getLocation()
117118
.getInternalEndPoint())) {
118-
TSStatus status = client.deleteModel(modelName);
119+
TSStatus status = client.deleteModel(new TDeleteModelReq(modelName));
119120
if (status.getCode() != TSStatusCode.SUCCESS_STATUS.getStatusCode()) {
120121
LOGGER.warn(
121122
"Failed to drop model [{}] on AINode [{}], status: {}",

‎iotdb-core/confignode/src/main/java/org/apache/iotdb/confignode/procedure/impl/node/RemoveAINodeProcedure.java‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,15 +21,15 @@
2121

2222
import org.apache.iotdb.common.rpc.thrift.TAINodeLocation;
2323
import org.apache.iotdb.common.rpc.thrift.TSStatus;
24-
import org.apache.iotdb.commons.client.ainode.AINodeClient;
25-
import org.apache.iotdb.commons.client.ainode.AINodeClientManager;
2624
import org.apache.iotdb.commons.utils.ThriftCommonsSerDeUtils;
2725
import org.apache.iotdb.confignode.consensus.request.write.ainode.RemoveAINodePlan;
2826
import org.apache.iotdb.confignode.consensus.request.write.model.DropModelInNodePlan;
2927
import org.apache.iotdb.confignode.procedure.env.ConfigNodeProcedureEnv;
3028
import org.apache.iotdb.confignode.procedure.exception.ProcedureException;
3129
import org.apache.iotdb.confignode.procedure.state.RemoveAINodeState;
3230
import org.apache.iotdb.confignode.procedure.store.ProcedureType;
31+
import org.apache.iotdb.db.protocol.client.ainode.AINodeClient;
32+
import org.apache.iotdb.db.protocol.client.ainode.AINodeClientManager;
3333
import org.apache.iotdb.rpc.TSStatusCode;
3434

3535
import org.slf4j.Logger;

0 commit comments

Comments
 (0)