optimize: add GetCallerIP util method in kitexutil

pkg/utils/kitexutil/kitexutil.go

@@ -85,6 +85,26 @@ func GetCallerAddr(ctx context.Context) (net.Addr, bool) {
 	return ri.From().Address(), true
 }
 
+// GetCallerIP is used for the server to get the IP of the caller.
+// Return false if failed to get the information.
+func GetCallerIP(ctx context.Context) (string, bool) {
+	defer func() { recover() }()
+
+	ri := rpcinfo.GetRPCInfo(ctx)
+	if ri == nil {
+		return "", false
+	}
+	addrStr := ri.From().Address().String()
+	if len(addrStr) == 0 {
+		return "", false
+	}
+
+	if ip, _, err := net.SplitHostPort(addrStr); err == nil {
+		return ip, true
+	}
+	return addrStr, true
+}
+
 // GetTransportProtocol gets the transport protocol of the request.
 // Return false if failed to get the information.
 func GetTransportProtocol(ctx context.Context) (string, bool) {

pkg/utils/kitexutil/kitexutil_test.go

@@ -112,6 +112,33 @@ func TestGetCallerAddr(t *testing.T) {
 	}
 }
 
+func TestGetCallerIP(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: "127.0.0.1", 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 := GetCallerIP(tt.args.ctx)
+			if !reflect.DeepEqual(got, tt.want) {
+				t.Errorf("GetCallerAddr() got = %v, want %v", got, tt.want)
+			}
+			if got1 != tt.want1 {
+				t.Errorf("GetCallerAddr() got1 = %v, want %v", got1, tt.want1)
+			}
+		})
+	}
+}
+
 func TestGetMethod(t *testing.T) {
 	type args struct {
 		ctx context.Context