怎么啟動http服務和rpc服務

怎么啟動http服務和rpc服務？這篇文章運用了實例代碼展示，代碼非常詳細，可供感興趣的小伙伴們參考借鑒，希望對大家有所幫助。

開啟http服務

老規矩，先添加配置，順便把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。

使用grpc

1. 從 https://github.com/google/protobuf/releases下載對應的安裝包，然后解壓，把protoc的執行文件放到PATH里；

2. 安裝 golang protobuf
   go get -u github.com/golang/protobuf/proto // golang protobuf 庫
   go get -u github.com/golang/protobuf/protoc-gen-go //protoc 轉換go工具

3. 安裝 gRPC-go
   go get google.golang.org/grpc

編寫protobuf文件

我在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

關于就分享到這里了，希望以上內容可以對大家有一定的幫助，可以學到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。