中查询一个文件夹下文件数量_gomicroV2从零开始（二）手写第一个微服务

mkdir go-todolist && cd go-todolist
# 初始化go-todolist为go mod项目
go mod init go-todolist# 继续创建&＃96;task-srv&＃96;目录
mkdir task-srv && cd task-srv# 这部分可以参考上一章的官方demo程序
# 注意这里多出一个repository文件夹用于数据库相关操作
mkdir proto repository handler
cd proto
mkdir task

//声明proto本版
syntax &＃61; "proto3";
//服务名
package go.micro.service.task;
//生成go文件的包路径
option go_package &＃61; "proto/task";//定义task服务的接口&＃xff0c;主要是增删改查
//结构非常类似于go语言的interface定义&＃xff0c;只是返回值必须用括号包裹&＃xff0c;且不能使用基本类型作为参数或返回值
service TaskService {rpc Create(Task)returns (EditResponse){}rpc Delete(Task)returns (EditResponse){}rpc Modify(Task)returns (EditResponse){}rpc Finished(Task)returns (EditResponse){}rpc Search(SearchRequest)returns (SearchResponse){}
}//下面是消息体message的定义&＃xff0c;可以暂时理解为go中的struct&＃xff0c;其中的1&＃xff0c;2&＃xff0c;3...是每个变量唯一的编码
message Task {//每条任务的ID&＃xff0c;本项目中对应mongodb记录的"_id"字段//&＃64;inject_tag: bson:"_id"string id &＃61; 1;//任务主体文字//&＃64;inject_tag: bson:"body"string body &＃61; 2;//用户设定的任务开始时间戳//&＃64;inject_tag: bson:"startTime"int64 startTime &＃61; 3;//用户设定的任务截止时间戳//&＃64;inject_tag: bson:"endTime"int64 endTime &＃61; 4;//任务是否已完成//&＃64;inject_tag: bson:"isFinished"int32 isFinished &＃61; 5;//用户实际完成时间戳//&＃64;inject_tag: bson:"finishTime"int64 finishTime &＃61; 6;//任务创建时间//&＃64;inject_tag: bson:"createTime"int64 createTime &＃61; 7;//任务修改时间//&＃64;inject_tag: bson:"updateTime"int64 updateTime &＃61; 8;
}//增删改接口返回参数
message EditResponse {//操作返回的消息string msg &＃61; 1;
}
//查询接口的参数
message SearchRequest{//分页查询页码&＃xff0c;从第一页开始int64 pageSize &＃61; 1;//分页查询每页数量&＃xff0c;默认20int64 pageCode &＃61; 2;// 排序字段string sortBy &＃61; 3;// 顺序 -1降序 1升序int32 order&＃61;4;//关键字模糊查询任务body字段string keyword &＃61; 5;
}message SearchResponse{//分页查询页码&＃xff0c;从第一页开始int64 pageSize &＃61; 1;//分页查询每页数量&＃xff0c;默认20int64 pageCode &＃61; 2;// 排序字段string sortBy &＃61; 3;// 顺序 -1降序 1升序int32 order&＃61;4;//数据总数int64 total &＃61; 5;//具体数据&＃xff0c;这里repeated表示可以出现多条&＃xff0c;类似于go中的slicerepeated Task rows &＃61; 6;
}


go get -u github.com/favadi/protoc-go-inject-tag

protoc-go-inject-tag -input&＃61;proto/task/task.pb.go

GOPATH:&＃61;$(shell go env GOPATH)
MODIFY&＃61;Mproto/imports/api.proto&＃61;github.com/micro/go-micro/v2/api/proto.PHONY: proto
proto:protoc --proto_path&＃61;. --micro_out&＃61;${MODIFY}:. --go_out&＃61;${MODIFY}:. proto/task/task.proto# 注意这里我们添加了tag控件的命令protoc-go-inject-tag -input&＃61;proto/task/task.pb.go.PHONY: build
build: protogo build -o task-srv main.go.PHONY: test
test:go test -v ./... -cover.PHONY: docker
docker:docker build . -t task-srv:latest

> protoc --proto_path&＃61;. --micro_out&＃61;. --go_out&＃61;. proto/task/task.proto
> protoc-go-inject-tag -input&＃61;proto/task/task.pb.go

...// Server API for TaskService servicetype TaskService interface {Create(ctx context.Context, in *Task, opts ...client.CallOption) (*EditResponse, error)Delete(ctx context.Context, in *Task, opts ...client.CallOption) (*EditResponse, error)Modify(ctx context.Context, in *Task, opts ...client.CallOption) (*EditResponse, error)Finished(ctx context.Context, in *Task, opts ...client.CallOption) (*EditResponse, error)Search(ctx context.Context, in *SearchRequest, opts ...client.CallOption) (*SearchResponse, error)
}...


package repositoryimport ("context""github.com/pkg/errors"pb "go-todolist/task-srv/proto/task""go.mongodb.org/mongo-driver/bson""go.mongodb.org/mongo-driver/bson/primitive""go.mongodb.org/mongo-driver/mongo""go.mongodb.org/mongo-driver/mongo/options""log""strings""time"
)const (// 默认数据库名DbName &＃61; "todolist"// 默认表名TaskCollection &＃61; "task"UnFinished &＃61; 0Finished &＃61; 1
)// 定义数据库基本的增删改查操作
type TaskRepository interface {InsertOne(ctx context.Context, task *pb.Task) errorDelete(ctx context.Context, id string) errorModify(ctx context.Context, task *pb.Task) errorFinished(ctx context.Context, task *pb.Task) errorCount(ctx context.Context, keyword string) (int64, error)Search(ctx context.Context, req *pb.SearchRequest) ([]*pb.Task, error)
}// 数据库操作实现类
type TaskRepositoryImpl struct {// 需要注入一个数据库连接客户端Conn *mongo.Client
}// 定义默认的操作表
func (repo *TaskRepositoryImpl) collection() *mongo.Collection {return repo.Conn.Database(DbName).Collection(TaskCollection)
}func (repo *TaskRepositoryImpl) InsertOne(ctx context.Context, task *pb.Task) error {_, err :&＃61; repo.collection().InsertOne(ctx, bson.M{"body": task.Body,"startTime": task.StartTime,"endTime": task.EndTime,"isFinished": UnFinished,"createTime": time.Now().Unix(),})return err
}func (repo *TaskRepositoryImpl) Delete(ctx context.Context, id string) error {oid, err :&＃61; primitive.ObjectIDFromHex(id)if err !&＃61; nil {return err}_, err &＃61; repo.collection().DeleteOne(ctx, bson.M{"_id": oid})return err
}func (repo *TaskRepositoryImpl) Modify(ctx context.Context, task *pb.Task) error {id, err :&＃61; primitive.ObjectIDFromHex(task.Id)if err !&＃61; nil {return err}_, err &＃61; repo.collection().UpdateOne(ctx, bson.M{"_id": id}, bson.M{"$set": bson.M{"body": task.Body,"startTime": task.StartTime,"endTime": task.EndTime,"updateTime": time.Now().Unix(),}})return err
}
func (repo *TaskRepositoryImpl) Finished(ctx context.Context, task *pb.Task) error {id, err :&＃61; primitive.ObjectIDFromHex(task.Id)if err !&＃61; nil {return err}now :&＃61; time.Now().Unix()update :&＃61; bson.M{"isFinished": int32(task.IsFinished),"updateTime": now,}if task.IsFinished &＃61;&＃61; Finished {update["finishTime"] &＃61; now}log.Print(task)log.Println(update)_, err &＃61; repo.collection().UpdateOne(ctx, bson.M{"_id": id}, bson.M{"$set": update})return err
}func (repo *TaskRepositoryImpl) Count(ctx context.Context, keyword string) (int64, error) {filter :&＃61; bson.M{}if keyword !&＃61; "" && strings.TrimSpace(keyword) !&＃61; "" {filter &＃61; bson.M{"body": bson.M{"$regex": keyword}}}count, err :&＃61; repo.collection().CountDocuments(ctx, filter)return count, err
}func (repo *TaskRepositoryImpl) Search(ctx context.Context, req *pb.SearchRequest) ([]*pb.Task, error) {filter :&＃61; bson.M{}if req.Keyword !&＃61; "" && strings.TrimSpace(req.Keyword) !&＃61; "" {filter &＃61; bson.M{"body": bson.M{"$regex": req.Keyword}}}cursor, err :&＃61; repo.collection().Find(ctx,filter,options.Find().SetSkip((req.PageCode-1)*req.PageSize),options.Find().SetLimit(req.PageSize),options.Find().SetSort(bson.M{req.SortBy: req.Order}))if err !&＃61; nil {return nil, errors.WithMessage(err, "search mongo")}var rows []*pb.Taskif err :&＃61; cursor.All(ctx, &rows); err !&＃61; nil {return nil, errors.WithMessage(err, "parse data")}return rows, nil
}


package handlerimport ("context""github.com/pkg/errors"pb "go-todolist/task-srv/proto/task""go-todolist/task-srv/repository"
)// 要实现接口&＃xff0c;首先当然要定义一个结构体
type TaskHandler struct {TaskRepository repository.TaskRepository
}func (t *TaskHandler) Create(ctx context.Context, req *pb.Task, resp *pb.EditResponse) error {if req.Body &＃61;&＃61; "" || req.StartTime <&＃61; 0 || req.EndTime <&＃61; 0 {return errors.New("bad param")}if err :&＃61; t.TaskRepository.InsertOne(ctx, req); err !&＃61; nil {return err}resp.Msg &＃61; "success"return nil
}
func (t *TaskHandler) Delete(ctx context.Context, req *pb.Task, resp *pb.EditResponse) error {if req.Id &＃61;&＃61; "" {return errors.New("bad param")}if err :&＃61; t.TaskRepository.Delete(ctx, req.Id); err !&＃61; nil {return err}resp.Msg &＃61; req.Idreturn nil
}
func (t *TaskHandler) Modify(ctx context.Context, req *pb.Task, resp *pb.EditResponse) error {if req.Id &＃61;&＃61; "" || req.Body &＃61;&＃61; "" || req.StartTime <&＃61; 0 || req.EndTime <&＃61; 0 {return errors.New("bad param")}if err :&＃61; t.TaskRepository.Modify(ctx, req); err !&＃61; nil {return err}resp.Msg &＃61; "success"return nil
}
func (t *TaskHandler) Finished(ctx context.Context, req *pb.Task, resp *pb.EditResponse) error {if req.Id &＃61;&＃61; "" || req.IsFinished !&＃61; repository.UnFinished && req.IsFinished !&＃61; repository.Finished {return errors.New("bad param")}if err :&＃61; t.TaskRepository.Finished(ctx, req); err !&＃61; nil {return err}resp.Msg &＃61; "success"return nil
}
func (t *TaskHandler) Search(ctx context.Context, req *pb.SearchRequest, resp *pb.SearchResponse) error {count, err :&＃61; t.TaskRepository.Count(ctx, req.Keyword)if err !&＃61; nil {return errors.WithMessage(err, "count row number")}if req.PageCode <&＃61; 0 {req.PageCode &＃61; 1}if req.PageSize <&＃61; 0 {req.PageSize &＃61; 20}if req.SortBy &＃61;&＃61; "" {req.SortBy &＃61; "createTime"}if req.Order &＃61;&＃61; 0 {req.Order &＃61; -1}if req.PageSize*(req.PageCode-1) > count {return errors.New("There&＃39;s not that much data")}rows, err :&＃61; t.TaskRepository.Search(ctx, req)if err !&＃61; nil {return errors.WithMessage(err, "search data")}*resp &＃61; pb.SearchResponse{PageCode: req.PageCode,PageSize: req.PageSize,SortBy: req.SortBy,Order: req.Order,Rows: rows,}return nil
}


package mainimport ("context""github.com/micro/go-micro/v2""github.com/pkg/errors""go-todolist/task-srv/handler"pb "go-todolist/task-srv/proto/task""go-todolist/task-srv/repository""go.mongodb.org/mongo-driver/mongo""go.mongodb.org/mongo-driver/mongo/options""log""time"
)// 这里是我内网的mongo地址&＃xff0c;请根据你得实际情况配置&＃xff0c;推荐使用dockers部署
const MONGO_URI &＃61; "mongodb://172.18.0.58:27017"func main() {// 在日志中打印文件路径&＃xff0c;便于调试代码log.SetFlags(log.Llongfile)conn, err :&＃61; connectMongo(MONGO_URI, time.Second)if err !&＃61; nil {log.Fatal(err)}defer conn.Disconnect(context.Background())// New Serviceservice :&＃61; micro.NewService(micro.Name("go.micro.service.task"),micro.Version("latest"),)// Initialise serviceservice.Init()// Register HandlertaskHandler :&＃61; &handler.TaskHandler{TaskRepository: &repository.TaskRepositoryImpl{Conn: conn,},}if err :&＃61; pb.RegisterTaskServiceHandler(service.Server(), taskHandler); err !&＃61; nil {log.Fatal(errors.WithMessage(err, "register server"))}// Run serviceif err :&＃61; service.Run(); err !&＃61; nil {log.Fatal(errors.WithMessage(err, "run server"))}
}// 连接到MongoDB
func connectMongo(uri string, timeout time.Duration) (*mongo.Client, error) {ctx, cancel :&＃61; context.WithTimeout(context.Background(), timeout)defer cancel()client, err :&＃61; mongo.Connect(ctx, options.Client().ApplyURI(uri))if err !&＃61; nil {return nil, errors.WithMessage(err, "create mongo connection session")}return client, nil
}


// This can be removed once etcd becomes go gettable, version 3.4 and 3.5 is not,
// see https://github.com/etcd-io/etcd/issues/11154 and https://github.com/etcd-io/etcd/issues/11931.
replace google.golang.org/grpc &＃61;> google.golang.org/grpc v1.26.0

replace google.golang.org/grpc &＃61;> google.golang.org/grpc v1.26.0

> go run main.go
2020-09-10 17:42:46 file&＃61;v2&＃64;v2.9.1/service.go:200 level&＃61;info Starting [service] go.micro.service.task
2020-09-10 17:42:46 file&＃61;grpc/grpc.go:864 level&＃61;info Server [grpc] Listening on [::]:63776
2020-09-10 17:42:46 file&＃61;grpc/grpc.go:697 level&＃61;info Registry [mdns] Registering node: go.micro.service.task-816e865d-7d43-4d02-94b4-be11cd475193

package mainimport ("context""github.com/micro/go-micro/v2""log"pb "task-srv/proto/task""task-srv/repository""time"
)func main() {// 在日志中打印文件路径&＃xff0c;便于调试代码log.SetFlags(log.Llongfile)// 客户端也注册为服务server :&＃61; micro.NewService(micro.Name("go.micro.client.task"))server.Init()taskService :&＃61; pb.NewTaskService("go.micro.service.task", server.Client())// 调用服务生成三条任务now :&＃61; time.Now()insertTask(taskService, "完成学习笔记&＃xff08;一&＃xff09;", now.Unix(), now.Add(time.Hour*24).Unix())insertTask(taskService, "完成学习笔记&＃xff08;二&＃xff09;", now.Add(time.Hour*24).Unix(), now.Add(time.Hour*48).Unix())insertTask(taskService, "完成学习笔记&＃xff08;三&＃xff09;", now.Add(time.Hour*48).Unix(), now.Add(time.Hour*72).Unix())// 分页查询任务列表page, err :&＃61; taskService.Search(context.Background(), &pb.SearchRequest{PageCode: 1,PageSize: 20,})if err !&＃61; nil {log.Fatal("search1", err)}log.Println(page)// 更新第一条记录为完成row :&＃61; page.Rows[0]if _, err &＃61; taskService.Finished(context.Background(), &pb.Task{Id: row.Id,IsFinished: repository.Finished,}); err !&＃61; nil {log.Fatal("finished", row.Id, err)}// 修改查询到的第二条数据,延长截至日期row &＃61; page.Rows[1]if _, err &＃61; taskService.Modify(context.Background(), &pb.Task{Id: row.Id,Body: row.Body,StartTime: row.StartTime,EndTime: now.Add(time.Hour * 72).Unix(),}); err !&＃61; nil {log.Fatal("modify", row.Id, err)}// 删除第三条记录row &＃61; page.Rows[2]if _, err &＃61; taskService.Delete(context.Background(), &pb.Task{Id: row.Id,}); err !&＃61; nil {log.Fatal("delete", row.Id, err)}// 再次分页查询&＃xff0c;校验修改结果page, err &＃61; taskService.Search(context.Background(), &pb.SearchRequest{})if err !&＃61; nil {log.Fatal("search2", err)}log.Println(page)
}
func insertTask(taskService pb.TaskService, body string, start, end int64) {_, err :&＃61; taskService.Create(context.Background(), &pb.Task{Body: body,StartTime: start,EndTime: end,})if err !&＃61; nil {log.Fatal("create", err)}log.Println("create task success! ")
}


>go run task-cli.go
2020-09-11 02:46:17.731409 I | E:/go-workspace/go-micro-study-notes/go-todolist1/task-srv/task-cli.go:79: create task success!
2020-09-11 02:46:17.732439 I | E:/go-workspace/go-micro-study-notes/go-todolist1/task-srv/task-cli.go:79: create task success!
2020-09-11 02:46:17.733434 I | E:/go-workspace/go-micro-study-notes/go-todolist1/task-srv/task-cli.go:79: create task success!
2020-09-11 02:46:17.734431 I | E:/go-workspace/go-micro-study-notes/go-todolist1/task-srv/task-cli.go:33: pageSize:20 pageCode:1 sortBy:"CreateTime" order:-1 rows:{id:"5f5a74791f01617808fef8b6" body:"完成学习笔记&＃xff08;一&＃xff09;
" startTime:1599763577 endTime:1599849977 createTime:1599763577} rows:{id:"5f5a74791f01617808fef8b7" body:"完成学习笔记&＃xff08;二&＃xff09;" startTime:1599849977 endTime:1599936377 createTime:1599763577} rows:{id:"5f5a74791f0161
7808fef8b8" body:"完成学习笔记&＃xff08;三&＃xff09;" startTime:1599936377 endTime:1600022777 createTime:1599763577}
2020-09-11 02:46:17.738421 I | E:/go-workspace/go-micro-study-notes/go-todolist1/task-srv/task-cli.go:68: pageSize:20 pageCode:1 sortBy:"CreateTime" order:-1 rows:{id:"5f5a74791f01617808fef8b6" body:"完成学习笔记&＃xff08;一&＃xff09;
" startTime:1599763577 endTime:1599849977 isFinished:1 finishTime:1599763577 createTime:1599763577 updateTime:1599763577} rows:{id:"5f5a74791f01617808fef8b7" body:"完成学习笔记&＃xff08;二&＃xff09;" startTime:1599849977 endTime:
1600022777 createTime:1599763577 updateTime:1599763577}