From 3a8c663bcd5b03d2fa457df1732e92e3117d03ab Mon Sep 17 00:00:00 2001
From: jingshi-ant <124752045+jingshi-ant@users.noreply.github.com>
Date: Fri, 10 May 2024 17:22:07 +0800
Subject: [PATCH] fix uri config for more (#276)

---
 docker/version.txt                         | 2 +-
 pkg/broker/services/intra/query_handler.go | 7 +------
 2 files changed, 2 insertions(+), 7 deletions(-)

diff --git a/docker/version.txt b/docker/version.txt
index 9f1f81c9..cae6a840 100644
--- a/docker/version.txt
+++ b/docker/version.txt
@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-version = "0.6.0b1"
\ No newline at end of file
+version = "0.6.0b2"
\ No newline at end of file
diff --git a/pkg/broker/services/intra/query_handler.go b/pkg/broker/services/intra/query_handler.go
index 799ae8d2..75503775 100644
--- a/pkg/broker/services/intra/query_handler.go
+++ b/pkg/broker/services/intra/query_handler.go
@@ -229,12 +229,7 @@ func distributeQueryToOtherParty(session *application.Session, enginesInfo *tran
 	selfCode := session.GetSelfPartyCode()
 	var selfEndpoint string
 	if !session.DryRun {
-		endpoint, err := session.GetEndpoint(selfCode)
-		if err != nil {
-			ret.err = err
-			return
-		}
-		selfEndpoint = endpoint
+		selfEndpoint = session.Engine.GetEndpointForPeer()
 	}
 
 	// distribute queries to other participants