add bookstore example

example/bookstore/rpc/add/add.go

@@ -0,0 +1,39 @@
+// Code generated by goctl. DO NOT EDIT!
+// Source: add.proto
+
+package main
+
+import (
+	"bookstore/rpc/add/internal/config"
+	"bookstore/rpc/add/internal/server"
+	"bookstore/rpc/add/internal/svc"
+	add "bookstore/rpc/add/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/add.yaml", "the config file")
+
+func main() {
+	flag.Parse()
+
+	var c config.Config
+	conf.MustLoad(*configFile, &c)
+	ctx := svc.NewServiceContext(c)
+	adderSrv := server.NewAdderServer(ctx)
+
+	s, err := rpcx.NewServer(c.RpcServerConf, func(grpcServer *grpc.Server) {
+		add.RegisterAdderServer(grpcServer, adderSrv)
+	})
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	fmt.Printf("Starting rpc server at %s...\n", c.ListenOn)
+	s.Start()
+}

example/bookstore/rpc/add/add.proto

@@ -0,0 +1,16 @@
+syntax = "proto3";
+
+package add;
+
+message addReq {
+    string book = 1;
+    int64 price = 2;
+}
+
+message addResp {
+    bool ok = 1;
+}
+
+service adder {
+    rpc add(addReq) returns(addResp);
+}

example/bookstore/rpc/add/adder/adder.go

@@ -0,0 +1,62 @@
+// Code generated by goctl. DO NOT EDIT!
+// Source: add.proto
+
+//go:generate mockgen -destination ./adder_mock.go -package adder -source $GOFILE
+
+package adder
+
+import (
+	add "bookstore/rpc/add/pb"
+	"context"
+
+	"github.com/tal-tech/go-zero/core/jsonx"
+	"github.com/tal-tech/go-zero/rpcx"
+)
+
+type (
+	Adder interface {
+		Add(ctx context.Context, in *AddReq) (*AddResp, error)
+	}
+
+	defaultAdder struct {
+		cli rpcx.Client
+	}
+)
+
+func NewAdder(cli rpcx.Client) Adder {
+	return &defaultAdder{
+		cli: cli,
+	}
+}
+
+func (m *defaultAdder) Add(ctx context.Context, in *AddReq) (*AddResp, error) {
+	var request add.AddReq
+	bts, err := jsonx.Marshal(in)
+	if err != nil {
+		return nil, errJsonConvert
+	}
+
+	err = jsonx.Unmarshal(bts, &request)
+	if err != nil {
+		return nil, errJsonConvert
+	}
+
+	client := add.NewAdderClient(m.cli.Conn())
+	resp, err := client.Add(ctx, &request)
+	if err != nil {
+		return nil, err
+	}
+
+	var ret AddResp
+	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/add/adder/adder_mock.go

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

example/bookstore/rpc/add/adder/types.go

@@ -0,0 +1,19 @@
+// Code generated by goctl. DO NOT EDIT!
+// Source: add.proto
+
+package adder
+
+import "errors"
+
+var errJsonConvert = errors.New("json convert error")
+
+type (
+	AddReq struct {
+		Book  string `json:"book,omitempty"`
+		Price int64  `json:"price,omitempty"`
+	}
+
+	AddResp struct {
+		Ok bool `json:"ok,omitempty"`
+	}
+)

example/bookstore/rpc/add/etc/add.yaml

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

example/bookstore/rpc/add/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/add/internal/logic/addlogic.go

@@ -0,0 +1,38 @@
+package logic
+
+import (
+	"bookstore/rpc/add/internal/svc"
+	add "bookstore/rpc/add/pb"
+	"bookstore/rpc/model"
+	"context"
+
+	"github.com/tal-tech/go-zero/core/logx"
+)
+
+type AddLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewAddLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AddLogic {
+	return &AddLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+func (l *AddLogic) Add(in *add.AddReq) (*add.AddResp, error) {
+	_, err := l.svcCtx.Model.Insert(model.Book{
+		Book:  in.Book,
+		Price: in.Price,
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	return &add.AddResp{
+		Ok: true,
+	}, nil
+}

example/bookstore/rpc/add/internal/server/adderserver.go

@@ -0,0 +1,26 @@
+// Code generated by goctl. DO NOT EDIT!
+// Source: add.proto
+
+package server
+
+import (
+	"bookstore/rpc/add/internal/logic"
+	"bookstore/rpc/add/internal/svc"
+	add "bookstore/rpc/add/pb"
+	"context"
+)
+
+type AdderServer struct {
+	svcCtx *svc.ServiceContext
+}
+
+func NewAdderServer(svcCtx *svc.ServiceContext) *AdderServer {
+	return &AdderServer{
+		svcCtx: svcCtx,
+	}
+}
+
+func (s *AdderServer) Add(ctx context.Context, in *add.AddReq) (*add.AddResp, error) {
+	l := logic.NewAddLogic(ctx, s.svcCtx)
+	return l.Add(in)
+}

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

@@ -0,0 +1,20 @@
+package svc
+
+import (
+	"bookstore/rpc/add/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/add/pb/add.pb.go

@@ -0,0 +1,167 @@
+// Code generated by protoc-gen-go.
+// source: add.proto
+// DO NOT EDIT!
+
+/*
+Package add is a generated protocol buffer package.
+
+It is generated from these files:
+	add.proto
+
+It has these top-level messages:
+	AddReq
+	AddResp
+*/
+package add
+
+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 AddReq struct {
+	Book  string `protobuf:"bytes,1,opt,name=book" json:"book,omitempty"`
+	Price int64  `protobuf:"varint,2,opt,name=price" json:"price,omitempty"`
+}
+
+func (m *AddReq) Reset()                    { *m = AddReq{} }
+func (m *AddReq) String() string            { return proto.CompactTextString(m) }
+func (*AddReq) ProtoMessage()               {}
+func (*AddReq) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{0} }
+
+func (m *AddReq) GetBook() string {
+	if m != nil {
+		return m.Book
+	}
+	return ""
+}
+
+func (m *AddReq) GetPrice() int64 {
+	if m != nil {
+		return m.Price
+	}
+	return 0
+}
+
+type AddResp struct {
+	Ok bool `protobuf:"varint,1,opt,name=ok" json:"ok,omitempty"`
+}
+
+func (m *AddResp) Reset()                    { *m = AddResp{} }
+func (m *AddResp) String() string            { return proto.CompactTextString(m) }
+func (*AddResp) ProtoMessage()               {}
+func (*AddResp) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{1} }
+
+func (m *AddResp) GetOk() bool {
+	if m != nil {
+		return m.Ok
+	}
+	return false
+}
+
+func init() {
+	proto.RegisterType((*AddReq)(nil), "add.addReq")
+	proto.RegisterType((*AddResp)(nil), "add.addResp")
+}
+
+// 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 Adder service
+
+type AdderClient interface {
+	Add(ctx context.Context, in *AddReq, opts ...grpc.CallOption) (*AddResp, error)
+}
+
+type adderClient struct {
+	cc *grpc.ClientConn
+}
+
+func NewAdderClient(cc *grpc.ClientConn) AdderClient {
+	return &adderClient{cc}
+}
+
+func (c *adderClient) Add(ctx context.Context, in *AddReq, opts ...grpc.CallOption) (*AddResp, error) {
+	out := new(AddResp)
+	err := grpc.Invoke(ctx, "/add.adder/add", in, out, c.cc, opts...)
+	if err != nil {
+		return nil, err
+	}
+	return out, nil
+}
+
+// Server API for Adder service
+
+type AdderServer interface {
+	Add(context.Context, *AddReq) (*AddResp, error)
+}
+
+func RegisterAdderServer(s *grpc.Server, srv AdderServer) {
+	s.RegisterService(&_Adder_serviceDesc, srv)
+}
+
+func _Adder_Add_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
+	in := new(AddReq)
+	if err := dec(in); err != nil {
+		return nil, err
+	}
+	if interceptor == nil {
+		return srv.(AdderServer).Add(ctx, in)
+	}
+	info := &grpc.UnaryServerInfo{
+		Server:     srv,
+		FullMethod: "/add.adder/Add",
+	}
+	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
+		return srv.(AdderServer).Add(ctx, req.(*AddReq))
+	}
+	return interceptor(ctx, in, info, handler)
+}
+
+var _Adder_serviceDesc = grpc.ServiceDesc{
+	ServiceName: "add.adder",
+	HandlerType: (*AdderServer)(nil),
+	Methods: []grpc.MethodDesc{
+		{
+			MethodName: "add",
+			Handler:    _Adder_Add_Handler,
+		},
+	},
+	Streams:  []grpc.StreamDesc{},
+	Metadata: "add.proto",
+}
+
+func init() { proto.RegisterFile("add.proto", fileDescriptor0) }
+
+var fileDescriptor0 = []byte{
+	// 136 bytes of a gzipped FileDescriptorProto
+	0x1f, 0x8b, 0x08, 0x00, 0x00, 0x09, 0x6e, 0x88, 0x02, 0xff, 0xe2, 0xe2, 0x4c, 0x4c, 0x49, 0xd1,
+	0x2b, 0x28, 0xca, 0x2f, 0xc9, 0x17, 0x62, 0x4e, 0x4c, 0x49, 0x51, 0x32, 0xe2, 0x62, 0x4b, 0x4c,
+	0x49, 0x09, 0x4a, 0x2d, 0x14, 0x12, 0xe2, 0x62, 0x49, 0xca, 0xcf, 0xcf, 0x96, 0x60, 0x54, 0x60,
+	0xd4, 0xe0, 0x0c, 0x02, 0xb3, 0x85, 0x44, 0xb8, 0x58, 0x0b, 0x8a, 0x32, 0x93, 0x53, 0x25, 0x98,
+	0x14, 0x18, 0x35, 0x98, 0x83, 0x20, 0x1c, 0x25, 0x49, 0x2e, 0x76, 0xb0, 0x9e, 0xe2, 0x02, 0x21,
+	0x3e, 0x2e, 0x26, 0xa8, 0x16, 0x8e, 0x20, 0xa6, 0xfc, 0x6c, 0x23, 0x4d, 0x2e, 0xd6, 0xc4, 0x94,
+	0x94, 0xd4, 0x22, 0x21, 0x05, 0x2e, 0x90, 0xf1, 0x42, 0xdc, 0x7a, 0x20, 0xfb, 0x20, 0x36, 0x48,
+	0xf1, 0x20, 0x38, 0xc5, 0x05, 0x49, 0x6c, 0x60, 0x57, 0x18, 0x03, 0x02, 0x00, 0x00, 0xff, 0xff,
+	0xe2, 0x6d, 0xb5, 0x91, 0x92, 0x00, 0x00, 0x00,
+}

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,
+}

example/bookstore/rpc/model/book.sql

@@ -0,0 +1,6 @@
+CREATE TABLE `book`
+(
+  `book` varchar(255) NOT NULL COMMENT 'book name',
+  `price` int NOT NULL COMMENT 'book price',
+  PRIMARY KEY(`book`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;

example/bookstore/rpc/model/bookmodel.go

@@ -0,0 +1,82 @@
+package model
+
+import (
+	"database/sql"
+	"fmt"
+	"strings"
+
+	"github.com/tal-tech/go-zero/core/stores/cache"
+	"github.com/tal-tech/go-zero/core/stores/sqlc"
+	"github.com/tal-tech/go-zero/core/stores/sqlx"
+	"github.com/tal-tech/go-zero/core/stringx"
+	"github.com/tal-tech/go-zero/tools/goctl/model/sql/builderx"
+)
+
+var (
+	bookFieldNames          = builderx.FieldNames(&Book{})
+	bookRows                = strings.Join(bookFieldNames, ",")
+	bookRowsExpectAutoSet   = strings.Join(stringx.Remove(bookFieldNames, "create_time", "update_time"), ",")
+	bookRowsWithPlaceHolder = strings.Join(stringx.Remove(bookFieldNames, "book", "create_time", "update_time"), "=?,") + "=?"
+
+	cacheBookBookPrefix = "cache#Book#book#"
+)
+
+type (
+	BookModel struct {
+		sqlc.CachedConn
+		table string
+	}
+
+	Book struct {
+		Book  string `db:"book"`  // book name
+		Price int64  `db:"price"` // book price
+	}
+)
+
+func NewBookModel(conn sqlx.SqlConn, c cache.CacheConf, table string) *BookModel {
+	return &BookModel{
+		CachedConn: sqlc.NewConn(conn, c),
+		table:      table,
+	}
+}
+
+func (m *BookModel) Insert(data Book) (sql.Result, error) {
+	query := `insert into ` + m.table + ` (` + bookRowsExpectAutoSet + `) values (?, ?)`
+	return m.ExecNoCache(query, data.Book, data.Price)
+}
+
+func (m *BookModel) FindOne(book string) (*Book, error) {
+	bookBookKey := fmt.Sprintf("%s%v", cacheBookBookPrefix, book)
+	var resp Book
+	err := m.QueryRow(&resp, bookBookKey, func(conn sqlx.SqlConn, v interface{}) error {
+		query := `select ` + bookRows + ` from ` + m.table + ` where book = ? limit 1`
+		return conn.QueryRow(v, query, book)
+	})
+	switch err {
+	case nil:
+		return &resp, nil
+	case sqlc.ErrNotFound:
+		return nil, ErrNotFound
+	default:
+		return nil, err
+	}
+}
+
+func (m *BookModel) Update(data Book) error {
+	bookBookKey := fmt.Sprintf("%s%v", cacheBookBookPrefix, data.Book)
+	_, err := m.Exec(func(conn sqlx.SqlConn) (result sql.Result, err error) {
+		query := `update ` + m.table + ` set ` + bookRowsWithPlaceHolder + ` where book = ?`
+		return conn.Exec(query, data.Price, data.Book)
+	}, bookBookKey)
+	return err
+}
+
+func (m *BookModel) Delete(book string) error {
+
+	bookBookKey := fmt.Sprintf("%s%v", cacheBookBookPrefix, book)
+	_, err := m.Exec(func(conn sqlx.SqlConn) (result sql.Result, err error) {
+		query := `delete from ` + m.table + ` where book = ?`
+		return conn.Exec(query, book)
+	}, bookBookKey)
+	return err
+}

example/bookstore/rpc/model/vars.go

@@ -0,0 +1,5 @@
+package model
+
+import "github.com/tal-tech/go-zero/core/stores/sqlx"
+
+var ErrNotFound = sqlx.ErrNotFound