Fix URI parsed failed bug. (#1561)

sogou · May 22, 2024 · fa85db0 · fa85db0
1 parent c2eb095
commit fa85db0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/factory/KafkaTaskImpl.cc b/src/factory/KafkaTaskImpl.cc
@@ -774,6 +774,8 @@ __WFKafkaTask *__WFKafkaTaskFactory::create_kafka_task(enum TransportType type,
 		uri.state = URI_STATE_ERROR;
 		uri.error = errno;
 	}
+	else
+		uri.state = URI_STATE_SUCCESS;
 
 	task->init(std::move(uri));
 	task->set_keep_alive(KAFKA_KEEPALIVE_DEFAULT);

diff --git a/src/factory/WFTaskFactory.inl b/src/factory/WFTaskFactory.inl
@@ -508,6 +508,8 @@ WFNetworkTaskFactory<REQ, RESP>::create_client_task(enum TransportType type,
 		uri.state = URI_STATE_ERROR;
 		uri.error = errno;
 	}
+	else
+		uri.state = URI_STATE_SUCCESS;
 
 	task->init(std::move(uri));
 	task->set_transport_type(type);