From 5b1bd363525f7c4b3720262f5a70b1bb815bf354 Mon Sep 17 00:00:00 2001
From: Kyle Xiao <xiaoguoqiao@bytedance.com>
Date: Fri, 15 Nov 2024 15:49:36 +0800
Subject: [PATCH] fix: work with latest kitex (#100)

---
 run.sh                       | 1 +
 thrift_streaming/generate.sh | 1 +
 2 files changed, 2 insertions(+)

diff --git a/run.sh b/run.sh
index 587f92a..abc7923 100755
--- a/run.sh
+++ b/run.sh
@@ -146,6 +146,7 @@ go get github.com/cloudwego/kitex@develop
 
 if [[ -n $LOCAL_REPO ]]; then
     go mod edit -replace github.com/cloudwego/kitex=${LOCAL_REPO}
+    go mod edit -replace github.com/cloudwego/kitex/pkg/protocol/bthrift=${LOCAL_REPO}/pkg/protocol/bthrift
 fi
 
 go mod tidy
diff --git a/thrift_streaming/generate.sh b/thrift_streaming/generate.sh
index 482be2b..eda7750 100755
--- a/thrift_streaming/generate.sh
+++ b/thrift_streaming/generate.sh
@@ -136,6 +136,7 @@ function generate_new_thriftgo_old_kitex() {
 go get github.com/cloudwego/kitex@develop
 if [ -d "$LOCAL_REPO" ]; then
     go mod edit -replace github.com/cloudwego/kitex=$LOCAL_REPO
+    go mod edit -replace github.com/cloudwego/kitex/pkg/protocol/bthrift=${LOCAL_REPO}/pkg/protocol/bthrift
 fi
 
 generate_new