Skip to content

Commit 48e338b

Browse files
CRZbulabulaCRZbulabula
committed
Fix conn bug
1 parent 157a0ec commit 48e338b

File tree

3 files changed

+6
-159
lines changed

3 files changed

+6
-159
lines changed

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

Lines changed: 6 additions & 118 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
import org.apache.iotdb.common.rpc.thrift.TConsensusGroupId;
2828
import org.apache.iotdb.common.rpc.thrift.TDataNodeConfiguration;
2929
import org.apache.iotdb.common.rpc.thrift.TDataNodeLocation;
30+
import org.apache.iotdb.common.rpc.thrift.TEndPoint;
3031
import org.apache.iotdb.common.rpc.thrift.TFlushReq;
3132
import org.apache.iotdb.common.rpc.thrift.TRegionReplicaSet;
3233
import org.apache.iotdb.common.rpc.thrift.TSStatus;
@@ -41,7 +42,6 @@
4142
import org.apache.iotdb.commons.auth.entity.PrivilegeUnion;
4243
import org.apache.iotdb.commons.client.ainode.AINodeClient;
4344
import org.apache.iotdb.commons.client.ainode.AINodeClientManager;
44-
import org.apache.iotdb.commons.client.ainode.AINodeInfo;
4545
import org.apache.iotdb.commons.cluster.NodeStatus;
4646
import org.apache.iotdb.commons.cluster.NodeType;
4747
import org.apache.iotdb.commons.conf.CommonConfig;
@@ -136,122 +136,7 @@
136136
import org.apache.iotdb.confignode.persistence.schema.ClusterSchemaInfo;
137137
import org.apache.iotdb.confignode.persistence.subscription.SubscriptionInfo;
138138
import org.apache.iotdb.confignode.procedure.impl.schema.SchemaUtils;
139-
import org.apache.iotdb.confignode.rpc.thrift.TAINodeRegisterReq;
140-
import org.apache.iotdb.confignode.rpc.thrift.TAINodeRestartReq;
141-
import org.apache.iotdb.confignode.rpc.thrift.TAINodeRestartResp;
142-
import org.apache.iotdb.confignode.rpc.thrift.TAlterLogicalViewReq;
143-
import org.apache.iotdb.confignode.rpc.thrift.TAlterOrDropTableReq;
144-
import org.apache.iotdb.confignode.rpc.thrift.TAlterPipeReq;
145-
import org.apache.iotdb.confignode.rpc.thrift.TAlterSchemaTemplateReq;
146-
import org.apache.iotdb.confignode.rpc.thrift.TAuthizedPatternTreeResp;
147-
import org.apache.iotdb.confignode.rpc.thrift.TCloseConsumerReq;
148-
import org.apache.iotdb.confignode.rpc.thrift.TClusterParameters;
149-
import org.apache.iotdb.confignode.rpc.thrift.TConfigNodeRegisterReq;
150-
import org.apache.iotdb.confignode.rpc.thrift.TConfigNodeRegisterResp;
151-
import org.apache.iotdb.confignode.rpc.thrift.TCountTimeSlotListReq;
152-
import org.apache.iotdb.confignode.rpc.thrift.TCountTimeSlotListResp;
153-
import org.apache.iotdb.confignode.rpc.thrift.TCreateCQReq;
154-
import org.apache.iotdb.confignode.rpc.thrift.TCreateConsumerReq;
155-
import org.apache.iotdb.confignode.rpc.thrift.TCreateFunctionReq;
156-
import org.apache.iotdb.confignode.rpc.thrift.TCreateModelReq;
157-
import org.apache.iotdb.confignode.rpc.thrift.TCreatePipePluginReq;
158-
import org.apache.iotdb.confignode.rpc.thrift.TCreatePipeReq;
159-
import org.apache.iotdb.confignode.rpc.thrift.TCreateSchemaTemplateReq;
160-
import org.apache.iotdb.confignode.rpc.thrift.TCreateTableViewReq;
161-
import org.apache.iotdb.confignode.rpc.thrift.TCreateTopicReq;
162-
import org.apache.iotdb.confignode.rpc.thrift.TCreateTrainingReq;
163-
import org.apache.iotdb.confignode.rpc.thrift.TCreateTriggerReq;
164-
import org.apache.iotdb.confignode.rpc.thrift.TDataNodeRegisterReq;
165-
import org.apache.iotdb.confignode.rpc.thrift.TDataNodeRestartReq;
166-
import org.apache.iotdb.confignode.rpc.thrift.TDataNodeRestartResp;
167-
import org.apache.iotdb.confignode.rpc.thrift.TDataPartitionTableResp;
168-
import org.apache.iotdb.confignode.rpc.thrift.TDataSchemaForTable;
169-
import org.apache.iotdb.confignode.rpc.thrift.TDatabaseSchema;
170-
import org.apache.iotdb.confignode.rpc.thrift.TDeactivateSchemaTemplateReq;
171-
import org.apache.iotdb.confignode.rpc.thrift.TDeleteDatabasesReq;
172-
import org.apache.iotdb.confignode.rpc.thrift.TDeleteLogicalViewReq;
173-
import org.apache.iotdb.confignode.rpc.thrift.TDeleteTableDeviceReq;
174-
import org.apache.iotdb.confignode.rpc.thrift.TDeleteTableDeviceResp;
175-
import org.apache.iotdb.confignode.rpc.thrift.TDeleteTimeSeriesReq;
176-
import org.apache.iotdb.confignode.rpc.thrift.TDescTable4InformationSchemaResp;
177-
import org.apache.iotdb.confignode.rpc.thrift.TDescTableResp;
178-
import org.apache.iotdb.confignode.rpc.thrift.TDropCQReq;
179-
import org.apache.iotdb.confignode.rpc.thrift.TDropFunctionReq;
180-
import org.apache.iotdb.confignode.rpc.thrift.TDropModelReq;
181-
import org.apache.iotdb.confignode.rpc.thrift.TDropPipePluginReq;
182-
import org.apache.iotdb.confignode.rpc.thrift.TDropPipeReq;
183-
import org.apache.iotdb.confignode.rpc.thrift.TDropSubscriptionReq;
184-
import org.apache.iotdb.confignode.rpc.thrift.TDropTopicReq;
185-
import org.apache.iotdb.confignode.rpc.thrift.TDropTriggerReq;
186-
import org.apache.iotdb.confignode.rpc.thrift.TExtendRegionReq;
187-
import org.apache.iotdb.confignode.rpc.thrift.TFetchTableResp;
188-
import org.apache.iotdb.confignode.rpc.thrift.TGetAllPipeInfoResp;
189-
import org.apache.iotdb.confignode.rpc.thrift.TGetAllSubscriptionInfoResp;
190-
import org.apache.iotdb.confignode.rpc.thrift.TGetAllTemplatesResp;
191-
import org.apache.iotdb.confignode.rpc.thrift.TGetAllTopicInfoResp;
192-
import org.apache.iotdb.confignode.rpc.thrift.TGetDataNodeLocationsResp;
193-
import org.apache.iotdb.confignode.rpc.thrift.TGetDatabaseReq;
194-
import org.apache.iotdb.confignode.rpc.thrift.TGetJarInListReq;
195-
import org.apache.iotdb.confignode.rpc.thrift.TGetJarInListResp;
196-
import org.apache.iotdb.confignode.rpc.thrift.TGetLocationForTriggerResp;
197-
import org.apache.iotdb.confignode.rpc.thrift.TGetModelInfoReq;
198-
import org.apache.iotdb.confignode.rpc.thrift.TGetModelInfoResp;
199-
import org.apache.iotdb.confignode.rpc.thrift.TGetPathsSetTemplatesReq;
200-
import org.apache.iotdb.confignode.rpc.thrift.TGetPathsSetTemplatesResp;
201-
import org.apache.iotdb.confignode.rpc.thrift.TGetPipePluginTableResp;
202-
import org.apache.iotdb.confignode.rpc.thrift.TGetRegionIdReq;
203-
import org.apache.iotdb.confignode.rpc.thrift.TGetRegionIdResp;
204-
import org.apache.iotdb.confignode.rpc.thrift.TGetSeriesSlotListReq;
205-
import org.apache.iotdb.confignode.rpc.thrift.TGetSeriesSlotListResp;
206-
import org.apache.iotdb.confignode.rpc.thrift.TGetTemplateResp;
207-
import org.apache.iotdb.confignode.rpc.thrift.TGetTimeSlotListReq;
208-
import org.apache.iotdb.confignode.rpc.thrift.TGetTimeSlotListResp;
209-
import org.apache.iotdb.confignode.rpc.thrift.TGetTriggerTableResp;
210-
import org.apache.iotdb.confignode.rpc.thrift.TGetUDFTableResp;
211-
import org.apache.iotdb.confignode.rpc.thrift.TGetUdfTableReq;
212-
import org.apache.iotdb.confignode.rpc.thrift.TMigrateRegionReq;
213-
import org.apache.iotdb.confignode.rpc.thrift.TNodeVersionInfo;
214-
import org.apache.iotdb.confignode.rpc.thrift.TPermissionInfoResp;
215-
import org.apache.iotdb.confignode.rpc.thrift.TPipeConfigTransferReq;
216-
import org.apache.iotdb.confignode.rpc.thrift.TPipeConfigTransferResp;
217-
import org.apache.iotdb.confignode.rpc.thrift.TReconstructRegionReq;
218-
import org.apache.iotdb.confignode.rpc.thrift.TRegionRouteMapResp;
219-
import org.apache.iotdb.confignode.rpc.thrift.TRemoveRegionReq;
220-
import org.apache.iotdb.confignode.rpc.thrift.TSchemaNodeManagementResp;
221-
import org.apache.iotdb.confignode.rpc.thrift.TSchemaPartitionTableResp;
222-
import org.apache.iotdb.confignode.rpc.thrift.TSetDataNodeStatusReq;
223-
import org.apache.iotdb.confignode.rpc.thrift.TSetSchemaTemplateReq;
224-
import org.apache.iotdb.confignode.rpc.thrift.TShowAINodesResp;
225-
import org.apache.iotdb.confignode.rpc.thrift.TShowCQResp;
226-
import org.apache.iotdb.confignode.rpc.thrift.TShowClusterResp;
227-
import org.apache.iotdb.confignode.rpc.thrift.TShowConfigNodes4InformationSchemaResp;
228-
import org.apache.iotdb.confignode.rpc.thrift.TShowConfigNodesResp;
229-
import org.apache.iotdb.confignode.rpc.thrift.TShowDataNodes4InformationSchemaResp;
230-
import org.apache.iotdb.confignode.rpc.thrift.TShowDataNodesResp;
231-
import org.apache.iotdb.confignode.rpc.thrift.TShowDatabaseResp;
232-
import org.apache.iotdb.confignode.rpc.thrift.TShowModelReq;
233-
import org.apache.iotdb.confignode.rpc.thrift.TShowModelResp;
234-
import org.apache.iotdb.confignode.rpc.thrift.TShowPipePluginReq;
235-
import org.apache.iotdb.confignode.rpc.thrift.TShowPipeReq;
236-
import org.apache.iotdb.confignode.rpc.thrift.TShowPipeResp;
237-
import org.apache.iotdb.confignode.rpc.thrift.TShowSubscriptionReq;
238-
import org.apache.iotdb.confignode.rpc.thrift.TShowSubscriptionResp;
239-
import org.apache.iotdb.confignode.rpc.thrift.TShowTable4InformationSchemaResp;
240-
import org.apache.iotdb.confignode.rpc.thrift.TShowTableResp;
241-
import org.apache.iotdb.confignode.rpc.thrift.TShowThrottleReq;
242-
import org.apache.iotdb.confignode.rpc.thrift.TShowTopicReq;
243-
import org.apache.iotdb.confignode.rpc.thrift.TShowTopicResp;
244-
import org.apache.iotdb.confignode.rpc.thrift.TShowVariablesResp;
245-
import org.apache.iotdb.confignode.rpc.thrift.TSpaceQuotaResp;
246-
import org.apache.iotdb.confignode.rpc.thrift.TStartPipeReq;
247-
import org.apache.iotdb.confignode.rpc.thrift.TStopPipeReq;
248-
import org.apache.iotdb.confignode.rpc.thrift.TSubscribeReq;
249-
import org.apache.iotdb.confignode.rpc.thrift.TTableInfo;
250-
import org.apache.iotdb.confignode.rpc.thrift.TThrottleQuotaResp;
251-
import org.apache.iotdb.confignode.rpc.thrift.TTimeSlotList;
252-
import org.apache.iotdb.confignode.rpc.thrift.TUnsetSchemaTemplateReq;
253-
import org.apache.iotdb.confignode.rpc.thrift.TUnsubscribeReq;
254-
import org.apache.iotdb.confignode.rpc.thrift.TUpdateModelInfoReq;
139+
import org.apache.iotdb.confignode.rpc.thrift.*;
255140
import org.apache.iotdb.consensus.common.DataSet;
256141
import org.apache.iotdb.consensus.exception.ConsensusException;
257142
import org.apache.iotdb.db.schemaengine.template.Template;
@@ -2710,8 +2595,11 @@ public TSStatus createTraining(TCreateTrainingReq req) {
27102595
updateModelInfo(new TUpdateModelInfoReq(req.modelId, ModelStatus.TRAINING.ordinal()));
27112596
trainingReq.setTargetDataSchema(dataSchema);
27122597

2598+
TAINodeInfo registeredAINode = getNodeManager().getRegisteredAINodeInfoList().get(0);
2599+
TEndPoint targetAINodeEndPoint =
2600+
new TEndPoint(registeredAINode.getInternalAddress(), registeredAINode.getInternalPort());
27132601
try (AINodeClient client =
2714-
AINodeClientManager.getInstance().borrowClient(AINodeInfo.endPoint)) {
2602+
AINodeClientManager.getInstance().borrowClient(targetAINodeEndPoint)) {
27152603
status = client.createTrainingTask(trainingReq);
27162604
if (status.getCode() != TSStatusCode.SUCCESS_STATUS.getStatusCode()) {
27172605
throw new IllegalArgumentException(status.message);

‎iotdb-core/node-commons/src/main/java/org/apache/iotdb/commons/client/ainode/AINodeInfo.java‎

Lines changed: 0 additions & 29 deletions
This file was deleted.

‎iotdb-core/node-commons/src/main/java/org/apache/iotdb/commons/conf/CommonConfig.java‎

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,6 @@
1919

2020
package org.apache.iotdb.commons.conf;
2121

22-
import org.apache.iotdb.common.rpc.thrift.TEndPoint;
2322
import org.apache.iotdb.commons.client.property.ClientPoolProperty.DefaultProperty;
2423
import org.apache.iotdb.commons.cluster.NodeStatus;
2524
import org.apache.iotdb.commons.enums.HandleSystemErrorStrategy;
@@ -174,9 +173,6 @@ public class CommonConfig {
174173
/** Disk Monitor. */
175174
private double diskSpaceWarningThreshold = 0.05;
176175

177-
/** Ip and port of target AI node. */
178-
private TEndPoint targetAINodeEndPoint = new TEndPoint("127.0.0.1", 10810);
179-
180176
/** Time partition origin in milliseconds. */
181177
private long timePartitionOrigin = 0;
182178

@@ -662,14 +658,6 @@ public void setStatusReason(String statusReason) {
662658
this.statusReason = statusReason;
663659
}
664660

665-
public TEndPoint getTargetAINodeEndPoint() {
666-
return targetAINodeEndPoint;
667-
}
668-
669-
public void setTargetAINodeEndPoint(TEndPoint targetAINodeEndPoint) {
670-
this.targetAINodeEndPoint = targetAINodeEndPoint;
671-
}
672-
673661
public int getTTimePartitionSlotTransmitLimit() {
674662
return TTimePartitionSlotTransmitLimit;
675663
}

0 commit comments

Comments
 (0)