Skip to content

Commit d49d7dd

Browse files
Alchuang22-devAlchuang22-dev
authored
[AINode] Update AINodeClient for DataNode to borrow (apache#16647)
1 parent a7f1527 commit d49d7dd

File tree

26 files changed

+487
-455
lines changed

26 files changed

+487
-455
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;
@@ -42,8 +49,6 @@
4249
import org.apache.iotdb.common.rpc.thrift.TTimePartitionSlot;
4350
import org.apache.iotdb.commons.auth.AuthException;
4451
import org.apache.iotdb.commons.auth.entity.PrivilegeUnion;
45-
import org.apache.iotdb.commons.client.ainode.AINodeClient;
46-
import org.apache.iotdb.commons.client.ainode.AINodeClientManager;
4752
import org.apache.iotdb.commons.cluster.NodeStatus;
4853
import org.apache.iotdb.commons.cluster.NodeType;
4954
import org.apache.iotdb.commons.conf.CommonConfig;
@@ -213,7 +218,6 @@
213218
import org.apache.iotdb.confignode.rpc.thrift.TGetTriggerTableResp;
214219
import org.apache.iotdb.confignode.rpc.thrift.TGetUDFTableResp;
215220
import org.apache.iotdb.confignode.rpc.thrift.TGetUdfTableReq;
216-
import org.apache.iotdb.confignode.rpc.thrift.TLoadModelReq;
217221
import org.apache.iotdb.confignode.rpc.thrift.TMigrateRegionReq;
218222
import org.apache.iotdb.confignode.rpc.thrift.TNodeVersionInfo;
219223
import org.apache.iotdb.confignode.rpc.thrift.TPermissionInfoResp;
@@ -226,7 +230,6 @@
226230
import org.apache.iotdb.confignode.rpc.thrift.TSchemaPartitionTableResp;
227231
import org.apache.iotdb.confignode.rpc.thrift.TSetDataNodeStatusReq;
228232
import org.apache.iotdb.confignode.rpc.thrift.TSetSchemaTemplateReq;
229-
import org.apache.iotdb.confignode.rpc.thrift.TShowAIDevicesResp;
230233
import org.apache.iotdb.confignode.rpc.thrift.TShowAINodesResp;
231234
import org.apache.iotdb.confignode.rpc.thrift.TShowCQResp;
232235
import org.apache.iotdb.confignode.rpc.thrift.TShowClusterResp;
@@ -235,10 +238,6 @@
235238
import org.apache.iotdb.confignode.rpc.thrift.TShowDataNodes4InformationSchemaResp;
236239
import org.apache.iotdb.confignode.rpc.thrift.TShowDataNodesResp;
237240
import org.apache.iotdb.confignode.rpc.thrift.TShowDatabaseResp;
238-
import org.apache.iotdb.confignode.rpc.thrift.TShowLoadedModelReq;
239-
import org.apache.iotdb.confignode.rpc.thrift.TShowLoadedModelResp;
240-
import org.apache.iotdb.confignode.rpc.thrift.TShowModelReq;
241-
import org.apache.iotdb.confignode.rpc.thrift.TShowModelResp;
242241
import org.apache.iotdb.confignode.rpc.thrift.TShowPipePluginReq;
243242
import org.apache.iotdb.confignode.rpc.thrift.TShowPipeReq;
244243
import org.apache.iotdb.confignode.rpc.thrift.TShowPipeResp;
@@ -256,12 +255,13 @@
256255
import org.apache.iotdb.confignode.rpc.thrift.TSubscribeReq;
257256
import org.apache.iotdb.confignode.rpc.thrift.TThrottleQuotaResp;
258257
import org.apache.iotdb.confignode.rpc.thrift.TTimeSlotList;
259-
import org.apache.iotdb.confignode.rpc.thrift.TUnloadModelReq;
260258
import org.apache.iotdb.confignode.rpc.thrift.TUnsetSchemaTemplateReq;
261259
import org.apache.iotdb.confignode.rpc.thrift.TUnsubscribeReq;
262260
import org.apache.iotdb.confignode.rpc.thrift.TUpdateModelInfoReq;
263261
import org.apache.iotdb.consensus.common.DataSet;
264262
import org.apache.iotdb.consensus.exception.ConsensusException;
263+
import org.apache.iotdb.db.protocol.client.ainode.AINodeClient;
264+
import org.apache.iotdb.db.protocol.client.ainode.AINodeClientManager;
265265
import org.apache.iotdb.db.schemaengine.template.Template;
266266
import org.apache.iotdb.db.schemaengine.template.TemplateAlterOperationType;
267267
import org.apache.iotdb.db.schemaengine.template.alter.TemplateAlterOperationUtil;
@@ -2863,19 +2863,19 @@ public TSStatus unloadModel(TUnloadModelReq req) {
28632863
}
28642864

28652865
@Override
2866-
public TShowModelResp showModel(TShowModelReq req) {
2866+
public TShowModelsResp showModel(TShowModelsReq req) {
28672867
TSStatus status = confirmLeader();
28682868
return status.getCode() == TSStatusCode.SUCCESS_STATUS.getStatusCode()
28692869
? modelManager.showModel(req)
2870-
: new TShowModelResp(status);
2870+
: new TShowModelsResp(status);
28712871
}
28722872

28732873
@Override
2874-
public TShowLoadedModelResp showLoadedModel(TShowLoadedModelReq req) {
2874+
public TShowLoadedModelsResp showLoadedModel(TShowLoadedModelsReq req) {
28752875
TSStatus status = confirmLeader();
28762876
return status.getCode() == TSStatusCode.SUCCESS_STATUS.getStatusCode()
28772877
? modelManager.showLoadedModel(req)
2878-
: new TShowLoadedModelResp(status, Collections.emptyMap());
2878+
: new TShowLoadedModelsResp(status, Collections.emptyMap());
28792879
}
28802880

28812881
@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.TConsensusGroupId;
2431
import org.apache.iotdb.common.rpc.thrift.TDataNodeLocation;
@@ -128,7 +135,6 @@
128135
import org.apache.iotdb.confignode.rpc.thrift.TGetTriggerTableResp;
129136
import org.apache.iotdb.confignode.rpc.thrift.TGetUDFTableResp;
130137
import org.apache.iotdb.confignode.rpc.thrift.TGetUdfTableReq;
131-
import org.apache.iotdb.confignode.rpc.thrift.TLoadModelReq;
132138
import org.apache.iotdb.confignode.rpc.thrift.TMigrateRegionReq;
133139
import org.apache.iotdb.confignode.rpc.thrift.TPermissionInfoResp;
134140
import org.apache.iotdb.confignode.rpc.thrift.TPipeConfigTransferReq;
@@ -140,7 +146,6 @@
140146
import org.apache.iotdb.confignode.rpc.thrift.TSchemaPartitionTableResp;
141147
import org.apache.iotdb.confignode.rpc.thrift.TSetDataNodeStatusReq;
142148
import org.apache.iotdb.confignode.rpc.thrift.TSetSchemaTemplateReq;
143-
import org.apache.iotdb.confignode.rpc.thrift.TShowAIDevicesResp;
144149
import org.apache.iotdb.confignode.rpc.thrift.TShowAINodesResp;
145150
import org.apache.iotdb.confignode.rpc.thrift.TShowCQResp;
146151
import org.apache.iotdb.confignode.rpc.thrift.TShowClusterResp;
@@ -149,10 +154,6 @@
149154
import org.apache.iotdb.confignode.rpc.thrift.TShowDataNodes4InformationSchemaResp;
150155
import org.apache.iotdb.confignode.rpc.thrift.TShowDataNodesResp;
151156
import org.apache.iotdb.confignode.rpc.thrift.TShowDatabaseResp;
152-
import org.apache.iotdb.confignode.rpc.thrift.TShowLoadedModelReq;
153-
import org.apache.iotdb.confignode.rpc.thrift.TShowLoadedModelResp;
154-
import org.apache.iotdb.confignode.rpc.thrift.TShowModelReq;
155-
import org.apache.iotdb.confignode.rpc.thrift.TShowModelResp;
156157
import org.apache.iotdb.confignode.rpc.thrift.TShowPipePluginReq;
157158
import org.apache.iotdb.confignode.rpc.thrift.TShowPipeReq;
158159
import org.apache.iotdb.confignode.rpc.thrift.TShowPipeResp;
@@ -166,7 +167,6 @@
166167
import org.apache.iotdb.confignode.rpc.thrift.TStartPipeReq;
167168
import org.apache.iotdb.confignode.rpc.thrift.TStopPipeReq;
168169
import org.apache.iotdb.confignode.rpc.thrift.TSubscribeReq;
169-
import org.apache.iotdb.confignode.rpc.thrift.TUnloadModelReq;
170170
import org.apache.iotdb.confignode.rpc.thrift.TUnsetSchemaTemplateReq;
171171
import org.apache.iotdb.confignode.rpc.thrift.TUnsubscribeReq;
172172
import org.apache.iotdb.consensus.common.DataSet;
@@ -893,10 +893,10 @@ TDataPartitionTableResp getOrCreateDataPartition(
893893
TSStatus unloadModel(TUnloadModelReq req);
894894

895895
/** Return the model table. */
896-
TShowModelResp showModel(TShowModelReq req);
896+
TShowModelsResp showModel(TShowModelsReq req);
897897

898898
/** Return the loaded model instances. */
899-
TShowLoadedModelResp showLoadedModel(TShowLoadedModelReq req);
899+
TShowLoadedModelsResp showLoadedModel(TShowLoadedModelsReq req);
900900

901901
/** Return all available AI devices. */
902902
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)