添加工作台充值和开票待确认的数量

db/finance.go

@@ -119,6 +119,30 @@ func GetWithdrawNums(ctx context.Context) (*http_model.WithdrawNums, error) {
 	return WithdrawNums, err
 }
 
+func GetInvoiceNums(ctx context.Context) (*http_model.InvoiceNums, error) {
+	var invoiceNums int64
+	db := GetReadDB(ctx)
+	err := db.Model(gorm_model.YounggeeInvoiceRecord{}).Where("status = 1").Count(&invoiceNums).Error
+	if err != nil {
+		return nil, err
+	}
+	InvoiceNums := new(http_model.InvoiceNums)
+	InvoiceNums.InvoiceNums = int(invoiceNums)
+	return InvoiceNums, err
+}
+
+func GetRechargeNums(ctx context.Context) (*http_model.RechargeNums, error) {
+	var rechargeNums int64
+	db := GetReadDB(ctx)
+	err := db.Model(gorm_model.YounggeeRechargeRecord{}).Where("status = 1").Count(&rechargeNums).Error
+	if err != nil {
+		return nil, err
+	}
+	RechargeNums := new(http_model.RechargeNums)
+	RechargeNums.RechargeNums = rechargeNums
+	return RechargeNums, err
+}
+
 func ConfirmWithdrawal(ctx context.Context, withdrawId string) error {
 	db := GetReadDB(ctx)
 	db2 := GetReadDB(ctx)

handler/getInvoiceNums.go

@@ -0,0 +1,55 @@
+package handler
+
+import (
+	"github.com/gin-gonic/gin"
+	"github.com/sirupsen/logrus"
+	"youngee_m_api/consts"
+	"youngee_m_api/db"
+	"youngee_m_api/model/http_model"
+	"youngee_m_api/util"
+)
+
+func WrapGetInvoiceNumsHandler(ctx *gin.Context) {
+	handler := newGetInvoiceNumsHandler(ctx)
+	BaseRun(handler)
+}
+
+type GetInvoiceNumsHandler struct {
+	ctx  *gin.Context
+	req  *http_model.GetInvoiceNumsRequest
+	resp *http_model.CommonResponse
+}
+
+func (g GetInvoiceNumsHandler) getContext() *gin.Context {
+	return g.ctx
+}
+
+func (g GetInvoiceNumsHandler) getResponse() interface{} {
+	return g.resp
+}
+
+func (g GetInvoiceNumsHandler) getRequest() interface{} {
+	return g.req
+}
+
+func (g GetInvoiceNumsHandler) run() {
+	data, err := db.GetInvoiceNums(g.ctx)
+	if err != nil {
+		logrus.WithContext(g.ctx).Errorf("[GetInvoiceNumsHandler] error GetInvoiceNums, err:%+v", err)
+		util.HandlerPackErrorResp(g.resp, consts.ErrorInternal, consts.DefaultToast)
+		return
+	}
+	g.resp.Data = data
+}
+
+func (g GetInvoiceNumsHandler) checkParam() error {
+	return nil
+}
+
+func newGetInvoiceNumsHandler(ctx *gin.Context) *GetInvoiceNumsHandler {
+	return &GetInvoiceNumsHandler{
+		ctx:  ctx,
+		req:  http_model.NewGetInvoiceNumsRequest(),
+		resp: http_model.NewGetInvoiceNumsResponse(),
+	}
+}

handler/getRechargeNums.go

@@ -0,0 +1,55 @@
+package handler
+
+import (
+	"github.com/gin-gonic/gin"
+	"github.com/sirupsen/logrus"
+	"youngee_m_api/consts"
+	"youngee_m_api/db"
+	"youngee_m_api/model/http_model"
+	"youngee_m_api/util"
+)
+
+func WrapGetRechargeNumsHandler(ctx *gin.Context) {
+	handler := newGetRechargeNumsHandler(ctx)
+	BaseRun(handler)
+}
+
+type GetRechargeNumsHandler struct {
+	ctx  *gin.Context
+	req  *http_model.GetRechargeNumsRequest
+	resp *http_model.CommonResponse
+}
+
+func (g GetRechargeNumsHandler) getContext() *gin.Context {
+	return g.ctx
+}
+
+func (g GetRechargeNumsHandler) getResponse() interface{} {
+	return g.resp
+}
+
+func (g GetRechargeNumsHandler) getRequest() interface{} {
+	return g.req
+}
+
+func (g GetRechargeNumsHandler) run() {
+	data, err := db.GetRechargeNums(g.ctx)
+	if err != nil {
+		logrus.WithContext(g.ctx).Errorf("[GetRechargeNumsHandler] error GetRechargeNums, err:%+v", err)
+		util.HandlerPackErrorResp(g.resp, consts.ErrorInternal, consts.DefaultToast)
+		return
+	}
+	g.resp.Data = data
+}
+
+func (g GetRechargeNumsHandler) checkParam() error {
+	return nil
+}
+
+func newGetRechargeNumsHandler(ctx *gin.Context) *GetRechargeNumsHandler {
+	return &GetRechargeNumsHandler{
+		ctx:  ctx,
+		req:  http_model.NewGetRechargeNumsRequest(),
+		resp: http_model.NewGetRechargeNumsResponse(),
+	}
+}

handler/getWithdrawNum.go

@@ -40,7 +40,6 @@ func (g GetWithdrawNumsHandler) run() {
 		return
 	}
 	g.resp.Data = data
-
 }
 
 func (g GetWithdrawNumsHandler) checkParam() error {

model/http_model/GetInvoiceNumsRequest.go

@@ -0,0 +1,18 @@
+package http_model
+
+type GetInvoiceNumsRequest struct {
+}
+
+type InvoiceNums struct {
+	InvoiceNums int `json:"invoice_nums"`
+}
+
+func NewGetInvoiceNumsRequest() *GetInvoiceNumsRequest {
+	return new(GetInvoiceNumsRequest)
+}
+
+func NewGetInvoiceNumsResponse() *CommonResponse {
+	resp := new(CommonResponse)
+	resp.Data = new(InvoiceNums)
+	return resp
+}

model/http_model/GetRechargeNumsRequest.go

@@ -0,0 +1,18 @@
+package http_model
+
+type GetRechargeNumsRequest struct {
+}
+
+type RechargeNums struct {
+	RechargeNums int64 `json:"recharge_nums"`
+}
+
+func NewGetRechargeNumsRequest() *GetRechargeNumsRequest {
+	return new(GetRechargeNumsRequest)
+}
+
+func NewGetRechargeNumsResponse() *CommonResponse {
+	resp := new(CommonResponse)
+	resp.Data = new(RechargeNums)
+	return resp
+}

route/init.go

@@ -138,6 +138,8 @@ func InitRoute(r *gin.Engine) {
 		f.POST("/withdrawalRecords", handler.WrapWithdrawalRecordsHandler)     // 搜索提现记录
 		f.POST("/getWithdrawalRecord", handler.WrapGetWithdrawalRecordHandler) // 查看提现记录
 		f.GET("/getWithdrawNums", handler.WrapGetWithdrawNumsHandler)          // 获取待提现的数量
+		f.GET("/getInvoiceNums", handler.WrapGetInvoiceNumsHandler)            // 获取待提现的数量
+		f.GET("/getRechargeNums", handler.WrapGetRechargeNumsHandler)          // 获取待开票的数量
 		f.POST("/ConfirmWithdrawal", handler.WrapConfirmWithdrawalHandler)     // 确认提现
 		f.POST("/getBankInfo", handler.WrapGetBankInfoHandler)                 // 获取银行开户地信息
 		f.POST("/getCodeUrl", handler.WrapGetCodeUrlHandler)                   // 获取微信支付链接