Merge branch 'feature/grpc-router' of gitlab.eolink.com:goku/eosc int…

…o feature/dubbo
xkm123 · Feb 14, 2023 · 74b6bb6 · 74b6bb6
2 parents ea476ca + c1fd4d7
commit 74b6bb6
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/eocontext/grpc-context/context.go b/eocontext/grpc-context/context.go
@@ -6,7 +6,6 @@ import (
 	"github.com/eolinker/eosc/eocontext"
 	"github.com/jhump/protoreflect/desc"
 	"github.com/jhump/protoreflect/dynamic"
-	"google.golang.org/grpc"
 	"google.golang.org/grpc/metadata"
 )
 
@@ -28,6 +27,7 @@ type IGrpcContext interface {
 	InsecureCertificateVerify(bool)
 	// Invoke grpc调用
 	Invoke(address string, timeout time.Duration) error
+	FastFinish() error
 }
 
 type IRequest interface {
@@ -38,15 +38,18 @@ type IRequest interface {
 	Method() string
 	SetMethod(string)
 	FullMethodName() string
+	RealIP() string
+	ForwardIP() string
 	// Message 获取原始请求内容，在grpc协议需要转其他协议时使用
 	Message(*desc.MessageDescriptor) *dynamic.Message
 }
 
 type IResponse interface {
 	Headers() metadata.MD
-	Message(*desc.MessageDescriptor) *dynamic.Message
+	Message() *dynamic.Message
 	Trailer() metadata.MD
-	ClientStream() grpc.ClientStream
+	Write(msg *dynamic.Message)
+	//ClientStream() grpc.ClientStream
 }
 
 type ITrailer interface {