feat: add GetCallee to kitexutil

pkg/utils/kitexutil/kitexutil.go

@@ -37,6 +37,18 @@ func GetCaller(ctx context.Context) (string, bool) {
 	return ri.From().ServiceName(), true
 }
 
+// GetCallee is used to get the Service Name of the callee.
+// Return false if failed to get the information.
+func GetCallee(ctx context.Context) (string, bool) {
+	defer func() { recover() }()
+
+	ri := rpcinfo.GetRPCInfo(ctx)
+	if ri == nil {
+		return "", false
+	}
+	return ri.To().ServiceName(), true
+}
+
 // GetMethod is used to get the current RPC Method name.
 // Return false if failed to get the information.
 func GetMethod(ctx context.Context) (string, bool) {

pkg/utils/kitexutil/kitexutil_test.go

@@ -80,6 +80,33 @@ func TestGetCaller(t *testing.T) {
 	}
 }
 
+func TestGetCallee(t *testing.T) {
+	type args struct {
+		ctx context.Context
+	}
+	tests := []struct {
+		name  string
+		args  args
+		want  string
+		want1 bool
+	}{
+		{name: "Success", args: args{testCtx}, want: callee, want1: true},
+		{name: "Failure", args: args{context.Background()}, want: "", want1: false},
+		{name: "Panic recovered", args: args{panicCtx}, want: "", want1: false},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got, got1 := GetCallee(tt.args.ctx)
+			if got != tt.want {
+				t.Errorf("GetCallee() got = %v, want %v", got, tt.want)
+			}
+			if got1 != tt.want1 {
+				t.Errorf("GetCallee() got1 = %v, want %v", got1, tt.want1)
+			}
+		})
+	}
+}
+
 func TestGetCallerAddr(t *testing.T) {
 	type args struct {
 		ctx context.Context