您好,登录后才能下订单哦!
怎么启动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进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。