使用go-zero实现高性能的Web服务

go get -u github.com/tal-tech/go-zero

goctl api new -o greet -d greet

package services

import (
    "context"
    "greet/internal/model"
    "greet/internal/types"
)

type InventoryService struct{}

func NewInventoryService() *InventoryService {
    return &InventoryService{}
}

// 获取库存
func (s *InventoryService) GetInventory(ctx context.Context, req *types.InventoryRequest) (*types.InventoryResponse, error) {
    inventory := model.GetInventory(req.ProductId)
    if inventory == nil {
        return nil, ErrInventoryNotFound
    }
    return &types.InventoryResponse{
        Inventory: inventory.Count,
    }, nil
}

syntax = "proto3";
package greet;

message InventoryRequest {
  int64 product_id = 1;
}

message InventoryResponse {
  int32 inventory = 1;
}

service Greet {
  rpc GetInventory(InventoryRequest) returns (InventoryResponse);
}

goctl api rpc -api greet.proto -src . -dir .

package main

import (
    "flag"
    "github.com/tal-tech/go-zero/core/conf"
    "github.com/tal-tech/go-zero/core/service"
    "greet/internal/config"
    "greet/internal/server"
    "greet/internal/svc"
)

var configFile = flag.String("f", "etc/greet-api.yaml", "the config file")

func main() {
    flag.Parse()

    var c config.Config
    conf.MustLoad(*configFile, &c)

    ctx := svc.NewServiceContext(c)
    server := server.NewGreetServer(ctx)

    s := service.New(server, service.WithName(c.Name))
    s.Start()
}

go run main.go -f etc/greet-api.yaml

grpcurl -plaintext -proto rpc/greet.proto -d '{"product_id": 1}' localhost:8080 greet.Greet.GetInventory

输出结果如下：

{
    "inventory": 10
}