From fa85db076a4969b9d1b613904e2ac02715710419 Mon Sep 17 00:00:00 2001
From: xiehan <52160700+Barenboim@users.noreply.github.com>
Date: Wed, 22 May 2024 18:03:16 +0800
Subject: [PATCH] Fix URI parsed failed bug. (#1561)

---
 src/factory/KafkaTaskImpl.cc  | 2 ++
 src/factory/WFTaskFactory.inl | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/src/factory/KafkaTaskImpl.cc b/src/factory/KafkaTaskImpl.cc
index ccd45304ec..6b87e784af 100644
--- 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
index ed24767c12..acd786e5df 100644
--- 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);