尺幅

services/sim.go

@@ -8,6 +8,7 @@ import (
 	"git.rosy.net.cn/jx-print/model"
 	putils "git.rosy.net.cn/jx-print/putils"
 	"git.rosy.net.cn/jx-print/services/api"
+	"github.com/jmoiron/sqlx"
 
 	"time"
 )
@@ -81,8 +82,7 @@ func SimFlowDaySettle() (err error) {
 //每月流量卡流量划转
 func SimFlowMonthSettle() {
 	var (
-		db  = globals.GetDB()
-		now = time.Now()
+		db = globals.GetDB()
 	)
 	//..每月1号划转
 	if time.Now().Day() != 1 {
@@ -92,16 +92,24 @@ func SimFlowMonthSettle() {
 	printers, _ := dao.GetPrinters(db, 0, "", 0, 0, 0)
 	for _, v := range printers {
 		if v.IccID != "" {
-			flowIncome := &model.SimFlowIncome{
-				IccID:        v.IccID,
-				CreatedAt:    &now,
-				UpdatedAt:    &now,
-				LastOperator: "jxadmin",
-				IncomeType:   model.FlowIncomeTypeJX,
-				Flow:         30,
-				FlowUnit:     "MB",
-			}
-			dao.Insert(db, flowIncome)
+			FlowIncome(db, v.IccID, 30, "MB")
 		}
 	}
 }
+
+func FlowIncome(db *sqlx.DB, iccID string, flow float64, unit string) (err error) {
+	var (
+		now = time.Now()
+	)
+	flowIncome := &model.SimFlowIncome{
+		IccID:        iccID,
+		CreatedAt:    &now,
+		UpdatedAt:    &now,
+		LastOperator: "jxadmin",
+		IncomeType:   model.FlowIncomeTypeJX,
+		Flow:         30,
+		FlowUnit:     "MB",
+	}
+	dao.Insert(db, flowIncome)
+	return err
+}