add bookstore example

example/bookstore/rpc/check/check.go

@@ -0,0 +1,39 @@
+// Code generated by goctl. DO NOT EDIT!
+// Source: check.proto
+
+package main
+
+import (
+	"bookstore/rpc/check/internal/config"
+	"bookstore/rpc/check/internal/server"
+	"bookstore/rpc/check/internal/svc"
+	check "bookstore/rpc/check/pb"
+	"flag"
+	"fmt"
+	"log"
+
+	"github.com/tal-tech/go-zero/core/conf"
+	"github.com/tal-tech/go-zero/rpcx"
+	"google.golang.org/grpc"
+)
+
+var configFile = flag.String("f", "etc/check.yaml", "the config file")
+
+func main() {
+	flag.Parse()
+
+	var c config.Config
+	conf.MustLoad(*configFile, &c)
+	ctx := svc.NewServiceContext(c)
+	checkerSrv := server.NewCheckerServer(ctx)
+
+	s, err := rpcx.NewServer(c.RpcServerConf, func(grpcServer *grpc.Server) {
+		check.RegisterCheckerServer(grpcServer, checkerSrv)
+	})
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	fmt.Printf("Starting rpc server at %s...\n", c.ListenOn)
+	s.Start()
+}

example/bookstore/rpc/check/check.proto

@@ -0,0 +1,16 @@
+syntax = "proto3";
+
+package check;
+
+message checkReq {
+    string book = 1;
+}
+
+message checkResp {
+    bool found = 1;
+    int64 price = 2;
+}
+
+service checker {
+    rpc check(checkReq) returns(checkResp);
+}

example/bookstore/rpc/check/checker/checker.go

@@ -0,0 +1,62 @@
+// Code generated by goctl. DO NOT EDIT!
+// Source: check.proto
+
+//go:generate mockgen -destination ./checker_mock.go -package checker -source $GOFILE
+
+package checker
+
+import (
+	check "bookstore/rpc/check/pb"
+	"context"
+
+	"github.com/tal-tech/go-zero/core/jsonx"
+	"github.com/tal-tech/go-zero/rpcx"
+)
+
+type (
+	Checker interface {
+		Check(ctx context.Context, in *CheckReq) (*CheckResp, error)
+	}
+
+	defaultChecker struct {
+		cli rpcx.Client
+	}
+)
+
+func NewChecker(cli rpcx.Client) Checker {
+	return &defaultChecker{
+		cli: cli,
+	}
+}
+
+func (m *defaultChecker) Check(ctx context.Context, in *CheckReq) (*CheckResp, error) {
+	var request check.CheckReq
+	bts, err := jsonx.Marshal(in)
+	if err != nil {
+		return nil, errJsonConvert
+	}
+
+	err = jsonx.Unmarshal(bts, &request)
+	if err != nil {
+		return nil, errJsonConvert
+	}
+
+	client := check.NewCheckerClient(m.cli.Conn())
+	resp, err := client.Check(ctx, &request)
+	if err != nil {
+		return nil, err
+	}
+
+	var ret CheckResp
+	bts, err = jsonx.Marshal(resp)
+	if err != nil {
+		return nil, errJsonConvert
+	}
+
+	err = jsonx.Unmarshal(bts, &ret)
+	if err != nil {
+		return nil, errJsonConvert
+	}
+
+	return &ret, nil
+}

example/bookstore/rpc/check/checker/checker_mock.go

@@ -0,0 +1,49 @@
+// Code generated by MockGen. DO NOT EDIT.
+// Source: checker.go
+
+// Package checker is a generated GoMock package.
+package checker
+
+import (
+	context "context"
+	gomock "github.com/golang/mock/gomock"
+	reflect "reflect"
+)
+
+// MockChecker is a mock of Checker interface
+type MockChecker struct {
+	ctrl     *gomock.Controller
+	recorder *MockCheckerMockRecorder
+}
+
+// MockCheckerMockRecorder is the mock recorder for MockChecker
+type MockCheckerMockRecorder struct {
+	mock *MockChecker
+}
+
+// NewMockChecker creates a new mock instance
+func NewMockChecker(ctrl *gomock.Controller) *MockChecker {
+	mock := &MockChecker{ctrl: ctrl}
+	mock.recorder = &MockCheckerMockRecorder{mock}
+	return mock
+}
+
+// EXPECT returns an object that allows the caller to indicate expected use
+func (m *MockChecker) EXPECT() *MockCheckerMockRecorder {
+	return m.recorder
+}
+
+// Check mocks base method
+func (m *MockChecker) Check(ctx context.Context, in *CheckReq) (*CheckResp, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "Check", ctx, in)
+	ret0, _ := ret[0].(*CheckResp)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// Check indicates an expected call of Check
+func (mr *MockCheckerMockRecorder) Check(ctx, in interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Check", reflect.TypeOf((*MockChecker)(nil).Check), ctx, in)
+}

example/bookstore/rpc/check/checker/types.go

@@ -0,0 +1,19 @@
+// Code generated by goctl. DO NOT EDIT!
+// Source: check.proto
+
+package checker
+
+import "errors"
+
+var errJsonConvert = errors.New("json convert error")
+
+type (
+	CheckReq struct {
+		Book string `json:"book,omitempty"`
+	}
+
+	CheckResp struct {
+		Found bool  `json:"found,omitempty"`
+		Price int64 `json:"price,omitempty"`
+	}
+)

example/bookstore/rpc/check/etc/check.yaml

@@ -0,0 +1,10 @@
+Name: check.rpc
+ListenOn: 127.0.0.1:8081
+Etcd:
+  Hosts:
+  - 127.0.0.1:2379
+  Key: check.rpc
+DataSource: root:@tcp(localhost:3306)/gozero
+Table: book
+Cache:
+  - Host: localhost:6379

example/bookstore/rpc/check/internal/config/config.go

@@ -0,0 +1,13 @@
+package config
+
+import (
+	"github.com/tal-tech/go-zero/core/stores/cache"
+	"github.com/tal-tech/go-zero/rpcx"
+)
+
+type Config struct {
+	rpcx.RpcServerConf
+	DataSource string
+	Table      string
+	Cache      cache.CacheConf
+}

example/bookstore/rpc/check/internal/logic/checklogic.go

@@ -0,0 +1,35 @@
+package logic
+
+import (
+	"bookstore/rpc/check/internal/svc"
+	check "bookstore/rpc/check/pb"
+	"context"
+
+	"github.com/tal-tech/go-zero/core/logx"
+)
+
+type CheckLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewCheckLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CheckLogic {
+	return &CheckLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+func (l *CheckLogic) Check(in *check.CheckReq) (*check.CheckResp, error) {
+	resp, err := l.svcCtx.Model.FindOne(in.Book)
+	if err != nil {
+		return nil, err
+	}
+
+	return &check.CheckResp{
+		Found: true,
+		Price: resp.Price,
+	}, nil
+}

example/bookstore/rpc/check/internal/server/checkerserver.go

@@ -0,0 +1,26 @@
+// Code generated by goctl. DO NOT EDIT!
+// Source: check.proto
+
+package server
+
+import (
+	"bookstore/rpc/check/internal/logic"
+	"bookstore/rpc/check/internal/svc"
+	check "bookstore/rpc/check/pb"
+	"context"
+)
+
+type CheckerServer struct {
+	svcCtx *svc.ServiceContext
+}
+
+func NewCheckerServer(svcCtx *svc.ServiceContext) *CheckerServer {
+	return &CheckerServer{
+		svcCtx: svcCtx,
+	}
+}
+
+func (s *CheckerServer) Check(ctx context.Context, in *check.CheckReq) (*check.CheckResp, error) {
+	l := logic.NewCheckLogic(ctx, s.svcCtx)
+	return l.Check(in)
+}

example/bookstore/rpc/check/internal/svc/servicecontext.go

@@ -0,0 +1,20 @@
+package svc
+
+import (
+	"bookstore/rpc/check/internal/config"
+	"bookstore/rpc/model"
+
+	"github.com/tal-tech/go-zero/core/stores/sqlx"
+)
+
+type ServiceContext struct {
+	c     config.Config
+	Model *model.BookModel
+}
+
+func NewServiceContext(c config.Config) *ServiceContext {
+	return &ServiceContext{
+		c:             c,
+		Model: model.NewBookModel(sqlx.NewMysql(c.DataSource), c.Cache, c.Table),
+	}
+}

example/bookstore/rpc/check/pb/check.pb.go

@@ -0,0 +1,167 @@
+// Code generated by protoc-gen-go.
+// source: check.proto
+// DO NOT EDIT!
+
+/*
+Package check is a generated protocol buffer package.
+
+It is generated from these files:
+	check.proto
+
+It has these top-level messages:
+	CheckReq
+	CheckResp
+*/
+package check
+
+import proto "github.com/golang/protobuf/proto"
+import fmt "fmt"
+import math "math"
+
+import (
+	context "golang.org/x/net/context"
+	grpc "google.golang.org/grpc"
+)
+
+// Reference imports to suppress errors if they are not otherwise used.
+var _ = proto.Marshal
+var _ = fmt.Errorf
+var _ = math.Inf
+
+// This is a compile-time assertion to ensure that this generated file
+// is compatible with the proto package it is being compiled against.
+// A compilation error at this line likely means your copy of the
+// proto package needs to be updated.
+const _ = proto.ProtoPackageIsVersion2 // please upgrade the proto package
+
+type CheckReq struct {
+	Book string `protobuf:"bytes,1,opt,name=book" json:"book,omitempty"`
+}
+
+func (m *CheckReq) Reset()                    { *m = CheckReq{} }
+func (m *CheckReq) String() string            { return proto.CompactTextString(m) }
+func (*CheckReq) ProtoMessage()               {}
+func (*CheckReq) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{0} }
+
+func (m *CheckReq) GetBook() string {
+	if m != nil {
+		return m.Book
+	}
+	return ""
+}
+
+type CheckResp struct {
+	Found bool  `protobuf:"varint,1,opt,name=found" json:"found,omitempty"`
+	Price int64 `protobuf:"varint,2,opt,name=price" json:"price,omitempty"`
+}
+
+func (m *CheckResp) Reset()                    { *m = CheckResp{} }
+func (m *CheckResp) String() string            { return proto.CompactTextString(m) }
+func (*CheckResp) ProtoMessage()               {}
+func (*CheckResp) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{1} }
+
+func (m *CheckResp) GetFound() bool {
+	if m != nil {
+		return m.Found
+	}
+	return false
+}
+
+func (m *CheckResp) GetPrice() int64 {
+	if m != nil {
+		return m.Price
+	}
+	return 0
+}
+
+func init() {
+	proto.RegisterType((*CheckReq)(nil), "check.checkReq")
+	proto.RegisterType((*CheckResp)(nil), "check.checkResp")
+}
+
+// Reference imports to suppress errors if they are not otherwise used.
+var _ context.Context
+var _ grpc.ClientConn
+
+// This is a compile-time assertion to ensure that this generated file
+// is compatible with the grpc package it is being compiled against.
+const _ = grpc.SupportPackageIsVersion4
+
+// Client API for Checker service
+
+type CheckerClient interface {
+	Check(ctx context.Context, in *CheckReq, opts ...grpc.CallOption) (*CheckResp, error)
+}
+
+type checkerClient struct {
+	cc *grpc.ClientConn
+}
+
+func NewCheckerClient(cc *grpc.ClientConn) CheckerClient {
+	return &checkerClient{cc}
+}
+
+func (c *checkerClient) Check(ctx context.Context, in *CheckReq, opts ...grpc.CallOption) (*CheckResp, error) {
+	out := new(CheckResp)
+	err := grpc.Invoke(ctx, "/check.checker/check", in, out, c.cc, opts...)
+	if err != nil {
+		return nil, err
+	}
+	return out, nil
+}
+
+// Server API for Checker service
+
+type CheckerServer interface {
+	Check(context.Context, *CheckReq) (*CheckResp, error)
+}
+
+func RegisterCheckerServer(s *grpc.Server, srv CheckerServer) {
+	s.RegisterService(&_Checker_serviceDesc, srv)
+}
+
+func _Checker_Check_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
+	in := new(CheckReq)
+	if err := dec(in); err != nil {
+		return nil, err
+	}
+	if interceptor == nil {
+		return srv.(CheckerServer).Check(ctx, in)
+	}
+	info := &grpc.UnaryServerInfo{
+		Server:     srv,
+		FullMethod: "/check.checker/Check",
+	}
+	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
+		return srv.(CheckerServer).Check(ctx, req.(*CheckReq))
+	}
+	return interceptor(ctx, in, info, handler)
+}
+
+var _Checker_serviceDesc = grpc.ServiceDesc{
+	ServiceName: "check.checker",
+	HandlerType: (*CheckerServer)(nil),
+	Methods: []grpc.MethodDesc{
+		{
+			MethodName: "check",
+			Handler:    _Checker_Check_Handler,
+		},
+	},
+	Streams:  []grpc.StreamDesc{},
+	Metadata: "check.proto",
+}
+
+func init() { proto.RegisterFile("check.proto", fileDescriptor0) }
+
+var fileDescriptor0 = []byte{
+	// 136 bytes of a gzipped FileDescriptorProto
+	0x1f, 0x8b, 0x08, 0x00, 0x00, 0x09, 0x6e, 0x88, 0x02, 0xff, 0xe2, 0xe2, 0x4e, 0xce, 0x48, 0x4d,
+	0xce, 0xd6, 0x2b, 0x28, 0xca, 0x2f, 0xc9, 0x17, 0x62, 0x05, 0x73, 0x94, 0xe4, 0xb8, 0x38, 0xc0,
+	0x8c, 0xa0, 0xd4, 0x42, 0x21, 0x21, 0x2e, 0x96, 0xa4, 0xfc, 0xfc, 0x6c, 0x09, 0x46, 0x05, 0x46,
+	0x0d, 0xce, 0x20, 0x30, 0x5b, 0xc9, 0x9c, 0x8b, 0x13, 0x2a, 0x5f, 0x5c, 0x20, 0x24, 0xc2, 0xc5,
+	0x9a, 0x96, 0x5f, 0x9a, 0x97, 0x02, 0x56, 0xc1, 0x11, 0x04, 0xe1, 0x80, 0x44, 0x0b, 0x8a, 0x32,
+	0x93, 0x53, 0x25, 0x98, 0x14, 0x18, 0x35, 0x98, 0x83, 0x20, 0x1c, 0x23, 0x53, 0x2e, 0x76, 0xb0,
+	0xc6, 0xd4, 0x22, 0x21, 0x2d, 0x2e, 0x88, 0x65, 0x42, 0xfc, 0x7a, 0x10, 0x17, 0xc0, 0x6c, 0x94,
+	0x12, 0x40, 0x15, 0x28, 0x2e, 0x48, 0x62, 0x03, 0xbb, 0xce, 0x18, 0x10, 0x00, 0x00, 0xff, 0xff,
+	0x6e, 0x6f, 0xa7, 0x1d, 0xac, 0x00, 0x00, 0x00,
+}