怎么启动http服务和rpc服务?这篇文章运用了实例代码展示,代码非常详细,可供感兴趣的小伙伴们参考借鉴,希望对大家有所帮助。
老规矩,先添加配置,顺便把rpc的配置也写进去,config.json:
"http_config": { "addr": ":8080" }, "rpc_config": { "addr": ":8081" }
config.go:
type HttpConfig struct { Addr string `json:"addr"` } type RpcConfig struct { Addr string `json:"addr"` }
接着,来看看process/http/http.go的内容:
package http import ( "github.com/gin-gonic/gin" "go.uber.org/zap" "os" ) var engine *gin.Engine //启动http服务 func StartHttpServer(addr string) { engine = gin.Default() Route() if err := engine.Run(addr); err != nil { zap.Error(err) os.Exit(1) } } //路由 func Route() { engine.GET("/server_time", GetServerTime) }
我用了gin来提供http服务,具体路由分发的方法写在了dispatch.go里面,这里写了一个简单的示例接口,获取服务器系统时间:
type Request struct { } type Response struct { Code int `json:"code"` Msg string `json:"msg"` Data interface{} `json:"data"` } func GetServerTime(ctx *gin.Context) { resp := Response{} resp.Data, resp.Code, resp.Msg = controller.GetServerTime() ctx.JSON(resp.Code, resp) }
我定义了一个请求结构体和返回结构体来统一一下,请求和返回的数据格式。GetServerTime调用了controller.GetServerTime()方法。我希望有个数据处理层来隔离接口。
rpc服务
我现在想对外提供rpc服务,和http服务提供一样的数据。我用的是grpc。
从 https://github.com/google/protobuf/releases下载对应的安装包,然后解压,把protoc的执行文件放到PATH里;
安装 golang protobuf
go get -u github.com/golang/protobuf/proto // golang protobuf 库
go get -u github.com/golang/protobuf/protoc-gen-go //protoc 转换go工具
我在process/rpc/server目录下,创建了一个server.proto文件来定义接口协议:
syntax = "proto3"; //语法声明 package server; //包名 service Server { rpc GetServerTime (ServerTimeRequest) returns (ServerTimeResponse); } message ServerTimeRequest { } message ServerTimeResponse { uint32 code = 1; string msg = 2; ServerTimeResponseData data = 3; } message ServerTimeResponseData { uint64 server_time = 1; }
接着,进到process/rpc/server目录,并执行生成go文件的命令,会生成一个go文件:
$ cd process/rpc/server $ protoc --go_out=plugins=grpc:. server.proto
然后,我们写一下我们的rpc服务端:
package rpc import ( "context" "github.com/TomatoMr/awesomeframework/process/controller" "github.com/TomatoMr/awesomeframework/process/rpc/server" "net" ) import "google.golang.org/grpc" type Server struct { } //启动rpc服务 func StartRpcServer(addr string) { lis, err := net.Listen("tcp", addr) if err != nil { } s := grpc.NewServer() server.RegisterServerServer(s, &Server{}) if err := s.Serve(lis); err != nil { } } func (rp *Server) GetServerTime(ctx context.Context, request *server.ServerTimeRequest) (*server.ServerTimeResponse, error) { data, code, msg := controller.GetServerTime() resp := &server.ServerTimeResponse{} respData := &server.ServerTimeResponseData{} resp.Msg = msg resp.Code = uint32(code) respData.ServerTime = uint64(data.ServerTime) resp.Data = respData return resp, nil }
main.go
//启动http服务 go http.StartHttpServer(config.GetConfig().HttpConfig.Addr) //启动rpc服务 go rpc.StartRpcServer(config.GetConfig().RpcConfig.Addr) logger.GetLogger().Info("Init success.") select {}
main.go添加了这几行代码,以go routine启动了http服务和rpc服务,因此,在最后用一个select来阻塞程序退出。
写一个rpc客户端的test如下:
package main import ( "context" "github.com/TomatoMr/awesomeframework/process/rpc/server" "google.golang.org/grpc" "log" "time" ) const ( Addr = "127.0.0.1:8081" ) func main() { conn, err := grpc.Dial(Addr, grpc.WithInsecure()) if err != nil { log.Fatalf("did not connect: %v", err) } defer conn.Close() c := server.NewServerClient(conn) ctx, cancel := context.WithTimeout(context.Background(), time.Second) defer cancel() r, _ := c.GetServerTime(ctx, &server.ServerTimeRequest{}) log.Printf("Data: %v", r.Data.ServerTime) }
编译:
$ go build
运行:
$ awesomeframework --config=./config/config.json
测试:
$ curl localhost:8080/server_time % Total % Received % Xferd Average Speed Time Time Time Current Dload Upload Total Spent Left Speed 100 56 100 56 0 0 18666 0 --:--:-- --:--:-- --:--:-- 28000 {"code":200,"msg":"","data":{"server_time":1579597891}}
测试rpc接口的文件我放在了process/rpc/test/request.go
$ go run request.go 2020/01/21 17:18:08 Data: 1579598288
关于就分享到这里了,希望以上内容可以对大家有一定的帮助,可以学到更多知识。如果觉得文章不错,可以把它分享出去让更多的人看到。
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。