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