feat: v1版本(随意函数传入)

Author: Yziyan

go.mod

@@ -1,3 +1,11 @@
 module github.com/Go-To-Byte/grpc-goroutine
 
 go 1.19
+
+require github.com/stretchr/testify v1.8.2
+
+require (
+github.com/davecgh/go-spew v1.1.1 // indirect
+github.com/pmezard/go-difflib v1.0.0 // indirect
+gopkg.in/yaml.v3 v3.0.1 // indirect
+)

go.sum

@@ -0,0 +1,17 @@
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
+github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
+github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
+github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8=
+github.com/stretchr/testify v1.8.2/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

grpcrun/grpcrun_test.go

@@ -0,0 +1,46 @@
+// @Author: Ciusyan 2023/2/26
+package grpcrun_test
+
+import (
+"context"
+"fmt"
+"github.com/stretchr/testify/assert"
+"testing"
+"time"
+
+"github.com/Go-To-Byte/grpc-goroutine/grpcrun"
+)
+
+// 作为GRPC请求的参数
+type loginReq struct {
+Username string `json:"username"`
+Password string `json:"password"`
+}
+
+// 作为GRPC响应的参数
+type loginResp struct {
+UserId int `json:"user_id"`
+Token string `json:"token"`
+}
+
+// GRPC请求的方法参数
+func Login(ctx context.Context, req *loginReq) (*loginResp, error) {
+if req.Username != "ciusyan" && req.Password != "222" {
+return nil, fmt.Errorf("登录失败")
+}
+fmt.Println("登录成功")
+return &loginResp{UserId: 21, Token: "testgrpccallsuccess"}, nil
+}
+
+func TestGrpcTask(t *testing.T) {
+
+should := assert.New(t)
+req := &loginReq{Username: "ciusyan", Password: "222"}
+ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
+defer cancel()
+resp, err := grpcrun.GrpcTask(Login, &ctx, req)
+
+if should.NoError(err) {
+t.Log(resp.(*loginResp))
+}
+}

grpcrun/task.go

@@ -0,0 +1,33 @@
+// Package grpcrun @Author: Ciusyan 2023/2/26
+package grpcrun
+
+import (
+"context"
+"fmt"
+"reflect"
+)
+
+// GrpcTask ：去调用GRPC的方法
+func GrpcTask(f any, ctx *context.Context, req any) (any, error) {
+
+v := reflect.ValueOf(f)
+if v.Kind() != reflect.Func {
+return nil, fmt.Errorf("参数错误")
+}
+
+// 调用参数
+argv := make([]reflect.Value, 2)
+argv[0] = reflect.ValueOf(*ctx)
+argv[1] = reflect.ValueOf(req)
+
+// 具体调用
+res := v.Call(argv)
+
+// 返回错误
+errV := res[1].Interface()
+if errV != nil {
+return res[0].Interface(), errV.(error)
+}
+
+return res[0].Interface(), nil
+}