calculator2

server/service/staff_salary.go

@@ -7,6 +7,7 @@ import (
 	"enterprise/common/model"
 	"enterprise/server/api"
 	"enterprise/server/session"
+	CommonService "enterprise/service"
 	"fmt"
 	"github.com/gin-gonic/gin"
 	log "github.com/sirupsen/logrus"
@@ -62,6 +63,25 @@ func (s *StaffSalary) List(sess *session.AdminSession, req *api.ListSalaryReq) (
 }
 
 func (s *StaffSalary) Create(sess *session.AdminSession, req *api.CreateSalaryReq) {
+	users, _, err := dao.NewStaffUserDao().Query(1, -1, sess.GetCorpId(), 0, req.Username, "", "", "")
+	session.CheckDBError(err)
+
+	for _, user := range users {
+		salary, err := dao.NewStaffSalaryDao().GetBy(sess.GetCorpId(), user.Id, req.Month)
+		session.CheckDBError(err)
+		salaryServ := CommonService.NewStaffSalary(user)
+		salary, err = salaryServ.CalcSalary(salary, req.Month)
+		if err != nil {
+			log.Errorf("CalcSalary error :%s", err.Error())
+			continue
+		}
+		if salary.Id == 0 {
+			_, err = dao.NewStaffSalaryDao().Create(salary)
+		} else {
+			err = dao.NewStaffSalaryDao().Update(salary)
+		}
+		session.CheckDBError(err)
+	}
 	return
 }