52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
package smithy
|
|
|
|
import (
|
|
"go_dreamfactory/comm"
|
|
"go_dreamfactory/pb"
|
|
"go_dreamfactory/utils"
|
|
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
// 3个顾客
|
|
var customerCount int32 = 3
|
|
|
|
func (this *apiComp) CustomerCheck(session comm.IUserSession, req *pb.SmithyCustomerReq) (code pb.ErrorCode) {
|
|
return
|
|
}
|
|
|
|
func (this *apiComp) Customer(session comm.IUserSession, req *pb.SmithyCustomerReq) (code pb.ErrorCode, data proto.Message) {
|
|
cus, err := this.module.modelTrade.getDBCustomer(session.GetUserId())
|
|
if err != nil {
|
|
if err == mongo.ErrNoDocuments {
|
|
c, err := this.module.modelTrade.addCustomer(session.GetUserId(), customerCount) //3个顾客
|
|
if err != nil {
|
|
code = pb.ErrorCode_DBError
|
|
return
|
|
}
|
|
cus = c
|
|
} else {
|
|
code = pb.ErrorCode_DBError
|
|
return
|
|
}
|
|
}
|
|
|
|
if utils.IsFirstTody(cus.LastRefreshTime) {
|
|
this.module.modelTrade.DelByUId(session.GetUserId())
|
|
c, err := this.module.modelTrade.addCustomer(session.GetUserId(), customerCount)
|
|
if err != nil {
|
|
code = pb.ErrorCode_DBError
|
|
return
|
|
}
|
|
cus = c
|
|
}
|
|
|
|
rsp := &pb.SmithyCustomerResp{
|
|
Customers: cus.Customers,
|
|
}
|
|
|
|
session.SendMsg(string(this.module.GetType()), "customer", rsp)
|
|
return
|
|
}
|