youngee_b_api/handler/enterprise_balance.go

package handler

import (
	"youngee_b_api/consts"
	"youngee_b_api/db"
	"youngee_b_api/middleware"
	"youngee_b_api/model/http_model"
	"youngee_b_api/util"

	"github.com/sirupsen/logrus"

	"github.com/gin-gonic/gin"
)

func WrapEnterpriseBalanceHandler(ctx *gin.Context) {
	handler := newEnterpriseBalanceHandler(ctx)
	baseRun(handler)
}

func newEnterpriseBalanceHandler(ctx *gin.Context) *EnterpriseBalanceHandler {
	return &EnterpriseBalanceHandler{
		req:  http_model.NewEnterpriseBalanceRequest(),
		resp: http_model.NewEnterpriseBalanceResponse(),
		ctx:  ctx,
	}
}

type EnterpriseBalanceHandler struct {
	req  *http_model.EnterpriseBalanceRequest
	resp *http_model.CommonResponse
	ctx  *gin.Context
}

func (e *EnterpriseBalanceHandler) getContext() *gin.Context {
	return e.ctx
}

func (e *EnterpriseBalanceHandler) getResponse() interface{} {
	return e.resp
}

func (e *EnterpriseBalanceHandler) getRequest() interface{} {
	return e.req
}

func (e *EnterpriseBalanceHandler) run() {
	enterpriseID := middleware.GetSessionAuth(e.ctx).EnterpriseID
	logrus.Println(enterpriseID)
	data, err := db.GetEnterpriseBalance(e.ctx, enterpriseID)
	if err != nil {
		logrus.WithContext(e.ctx).Errorf("[FullProjectListHandler] error GetFullProjectList, err:%+v", err)
		util.HandlerPackErrorResp(e.resp, consts.ErrorInternal, consts.DefaultToast)
		return
	}
	e.resp.Data = data
}

func (e *EnterpriseBalanceHandler) checkParam() error {
	return nil
}