ERP_storage/controllers/IOTNetworkCard.go

package controllers

import (
	"ERP_storage/Nats/NatsServer"
	"ERP_storage/conf"
	"ERP_storage/logs"
	"ERP_storage/models/Account"
	"ERP_storage/models/IOTNetworkCard"
	"encoding/json"
	"fmt"
	"github.com/beego/beego/v2/adapter/orm"
	beego "github.com/beego/beego/v2/server/web"
	"github.com/xuri/excelize/v2"
	userlibs "gogs.baozhida.cn/zoie/ERP_libs/User"
	"gogs.baozhida.cn/zoie/ERP_libs/lib"
	"math"
	"net/url"
	"os"
	"strconv"
	"time"
)

type IOTNetworkCardController struct {
	beego.Controller
	User userlibs.User
}

func (c *IOTNetworkCardController) Prepare() {
	c.User = *Account.User_r
}

// 财务管理列表
func (c *IOTNetworkCardController) IOTNetworkCard_List() {

	// 分页参数 初始化
	page, _ := c.GetInt("page")
	if page < 1 {
		page = 1
	}
	page_z, _ := c.GetInt("page_z")
	if page_z < 1 {
		page_z = conf.Page_size
	}

	// 查询
	T_name := c.GetString("T_name")
	T_company_name := c.GetString("T_company_name")
	T_state, _ := c.GetInt("T_state")

	userList, _ := NatsServer.Read_User_List_All()
	Account.Read_User_All_Map(userList)
	IOTNetworkCardDao := IOTNetworkCard.NewIOTNetworkCard(orm.NewOrm())
	R_List, R_cnt := IOTNetworkCardDao.Read_IOTNetworkCard_List(T_name, T_company_name, T_state, page, page_z)

	var r_jsons lib.R_JSONS
	r_jsons.Num = R_cnt
	r_jsons.Data = R_List
	r_jsons.Page = page
	r_jsons.Page_size = int(math.Ceil(float64(R_cnt) / float64(page_z)))

	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
	c.ServeJSON()
	return
}

func (c *IOTNetworkCardController) IOTNetworkCard_Get() {

	// 查询
	T_id, _ := c.GetInt("T_id")
	IOTNetworkCard_r, err := IOTNetworkCard.Read_IOTNetworkCard_ById(T_id)
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 203, Msg: "T_id Err!"}
		c.ServeJSON()
		return
	}

	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: IOTNetworkCard.IOTNetworkCardToIOTNetworkCard_R(IOTNetworkCard_r)}
	c.ServeJSON()
	return
}

func (c *IOTNetworkCardController) IOTNetworkCard_Add() {

	T_company_name := c.GetString("T_company_name")
	T_in_date := c.GetString("T_in_date")
	T_meal_type := c.GetString("T_meal_type")
	T_meal := c.GetString("T_meal")
	T_expire := c.GetString("T_expire")
	T_iccid := c.GetString("T_iccid")
	T_MSISDN := c.GetString("T_MSISDN")
	T_remark := c.GetString("T_remark")
	T_State, _ := c.GetInt("T_State")

	var_ := IOTNetworkCard.IOTNetworkCard{
		T_company_name: T_company_name,
		T_in_date:      T_in_date,
		T_meal_type:    T_meal_type,
		T_meal:         T_meal,
		T_expire:       T_expire,
		T_iccid:        T_iccid,
		T_MSISDN:       T_MSISDN,
		T_State:        T_State,
		T_remark:       T_remark,
	}
	// 添加入库记录
	var record []IOTNetworkCard.IOTNetworkCardRecord
	record = append(record, IOTNetworkCard.IOTNetworkCardRecord{
		T_data:  time.Now().Format("2006-01-02"),
		T_State: T_State,
	})
	recordB, _ := json.Marshal(&record)
	var_.T_record = string(recordB)

	card, err := IOTNetworkCard.Read_IOTNetworkCard_ByT_iccid(T_iccid)
	if err != nil && err.Error() != orm.ErrNoRows.Error() {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "添加失败"}
		c.ServeJSON()
		return
	}
	if card.Id > 0 {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "此卡已入库"}
		c.ServeJSON()
		return
	}
	o := orm.NewOrm()

	IOTNetworkCardDao := IOTNetworkCard.NewIOTNetworkCard(o)
	Id, err := IOTNetworkCardDao.Add_IOTNetworkCard(var_)
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 203, Msg: "添加失败"}
		c.ServeJSON()
		return
	}

	NatsServer.AddUserLogs(c.User.T_uuid, "物联网卡", "添加", var_)

	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: Id}
	c.ServeJSON()
	return
}

func (c *IOTNetworkCardController) IOTNetworkCard_Edit() {
	T_id, _ := c.GetInt("T_id")
	T_in_date := c.GetString("T_in_date")
	T_company_name := c.GetString("T_company_name")
	T_meal_type := c.GetString("T_meal_type")
	T_meal := c.GetString("T_meal")
	T_expire := c.GetString("T_expire")
	T_iccid := c.GetString("T_iccid")
	T_MSISDN := c.GetString("T_MSISDN")
	T_out_date := c.GetString("T_out_date")       // 出库日期
	T_out_project := c.GetString("T_out_project") // 出库项目
	T_receive := c.GetString("T_receive")         // 领用人
	T_State, _ := c.GetInt("T_State")             // 状态
	T_remark := c.GetString("T_remark")           // 状态

	IOTNetworkCard_r, err := IOTNetworkCard.Read_IOTNetworkCard_ById(T_id)
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 203, Msg: "T_id Err!"}
		c.ServeJSON()
		return
	}
	if len(T_in_date) > 0 {
		IOTNetworkCard_r.T_in_date = T_in_date
	}
	if len(T_company_name) > 0 {
		IOTNetworkCard_r.T_company_name = T_company_name
	}
	if len(T_meal_type) > 0 {
		IOTNetworkCard_r.T_meal_type = T_meal_type
	}
	if len(T_meal) > 0 {
		IOTNetworkCard_r.T_meal = T_meal
	}
	if len(T_expire) > 0 {
		IOTNetworkCard_r.T_expire = T_expire
	}
	if len(T_iccid) > 0 {
		IOTNetworkCard_r.T_iccid = T_iccid
	}
	if len(T_MSISDN) > 0 {
		IOTNetworkCard_r.T_MSISDN = T_MSISDN
	}
	if len(T_out_date) > 0 {
		IOTNetworkCard_r.T_out_date = T_out_date
	}
	if len(T_out_project) > 0 {
		IOTNetworkCard_r.T_out_project = T_out_project
	}
	if len(T_receive) > 0 {
		IOTNetworkCard_r.T_receive = T_receive
	}
	if len(T_remark) > 0 {
		IOTNetworkCard_r.T_remark = T_remark
	}
	if T_State > 0 && T_State != IOTNetworkCard_r.T_State {
		IOTNetworkCard_r.T_State = T_State

		// 添加入库记录
		var record []IOTNetworkCard.IOTNetworkCardRecord
		err = json.Unmarshal([]byte(IOTNetworkCard_r.T_record), &record)
		if err != nil {
			logs.Error("JSON 反序列化失败:", err)
			return
		}
		record = append(record, IOTNetworkCard.IOTNetworkCardRecord{
			T_data:        time.Now().Format("2006-01-02"),
			T_State:       T_State,
			T_out_project: T_out_project,
			T_remark:      T_remark,
		})
		recordB, _ := json.Marshal(&record)
		IOTNetworkCard_r.T_record = string(recordB)

	}
	if T_State == IOTNetworkCard.IOTNetworkCardReturnIn {
		IOTNetworkCard_r.T_out_date = ""
	}
	o := orm.NewOrm()
	IOTNetworkCardDao := IOTNetworkCard.NewIOTNetworkCard(o)
	if err = IOTNetworkCardDao.Update_IOTNetworkCard(IOTNetworkCard_r,
		"T_in_date", "T_meal_type", "T_meal", "T_expire", "T_iccid", "T_MSISDN", "T_out_date", "T_out_project", "T_receive", "T_State", "T_record", "T_remark"); err != nil {
		c.Data["json"] = lib.JSONS{Code: 203, Msg: "修改失败"}
		c.ServeJSON()
		return
	}

	NatsServer.AddUserLogs(c.User.T_uuid, "物联网卡", "修改", IOTNetworkCard_r)

	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!"}
	c.ServeJSON()
	return
}
func (c *IOTNetworkCardController) IOTNetworkCard_Del() {

	T_id, _ := c.GetInt("T_id")
	IOTNetworkCard_r, err := IOTNetworkCard.Read_IOTNetworkCard_ById(T_id)
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 203, Msg: "T_id Err!"}
		c.ServeJSON()
		return
	}
	o := orm.NewOrm()
	IOTNetworkCardDao := IOTNetworkCard.NewIOTNetworkCard(o)
	err = IOTNetworkCardDao.Delete_IOTNetworkCard(IOTNetworkCard_r)
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "删除失败!"}
		c.ServeJSON()
		return
	}

	NatsServer.AddUserLogs(c.User.T_uuid, "物联网卡", "删除", strconv.Itoa(T_id))

	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: T_id}
	c.ServeJSON()
	return
}

// 导入
func (c *IOTNetworkCardController) IOTNetworkCard_Import() {
	// 获取上传的文件
	f, _, err := c.GetFile("file")
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 1201, Msg: "读取文件失败!", Data: nil}
		c.ServeJSON()
		return
	}
	defer f.Close()

	// 读取 Excel 文件
	reader, err := excelize.OpenReader(f)
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 1201, Msg: "打开文件失败!", Data: nil}
		c.ServeJSON()
		return
	}
	defer func() {
		if err = reader.Close(); err != nil {
			c.Data["json"] = lib.JSONS{Code: 1201, Msg: "关闭文件失败!", Data: nil}
			c.ServeJSON()
			return
		}
	}()
	var IOTNetworkCardList []IOTNetworkCard.IOTNetworkCard
	sheetMap := reader.GetSheetMap()
	for _, sheetName := range sheetMap {
		//读取工作表中的数据
		rows, errs := reader.GetRows(sheetName)
		if errs != nil {
			c.Data["json"] = lib.JSONS{Code: 1201, Msg: "读取工作表失败!", Data: nil}
			c.ServeJSON()
			return
		}
		for i, row := range rows {
			if i < 2 {
				continue // 忽略第一行和第二行
			}
			if len(row) >= 6 {
				// 解析时间
				t, errT := time.Parse("01-02-06", row[1])
				if errT != nil {
					c.Data["json"] = lib.JSONS{Code: 203, Msg: "解析入库时间失败"}
					c.ServeJSON()
					return
				}
				T_MSISDN := row[6]
				//_, errGet := IOTNetworkCard.Read_IOTNetworkCard_ByT_iccid(T_MSISDN)
				//if errGet != nil && errGet.Error() != orm.ErrNoRows.Error() {
				//	continue
				//}
				var_ := IOTNetworkCard.IOTNetworkCard{
					T_in_date:      t.Format("2006-01-02"),
					T_company_name: row[0],
					T_meal_type:    row[2],
					T_meal:         row[3],
					T_expire:       row[4],
					T_iccid:        row[5],
					T_MSISDN:       T_MSISDN,
					T_State:        IOTNetworkCard.IOTNetworkCardNewCardIn,
				}
				// 添加入库记录
				var record []IOTNetworkCard.IOTNetworkCardRecord
				record = append(record, IOTNetworkCard.IOTNetworkCardRecord{
					T_data:  time.Now().Format("2006-01-02"),
					T_State: IOTNetworkCard.IOTNetworkCardNewCardIn,
				})

				if len(row) >= 8 && len(row[7]) > 0 {
					var outTime string
					outT, errOutT := time.Parse("01-02-06", row[7])
					if errOutT != nil {
						c.Data["json"] = lib.JSONS{Code: 203, Msg: "解析出库时间失败"}
						c.ServeJSON()
						return
					}
					outTime = outT.Format("2006-01-02")
					var_.T_State = IOTNetworkCard.IOTNetworkCardOut
					var_.T_out_date = outTime
					var_.T_out_project = row[8]
					record = append(record, IOTNetworkCard.IOTNetworkCardRecord{
						T_data:        time.Now().Format("2006-01-02"),
						T_State:       IOTNetworkCard.IOTNetworkCardOut,
						T_out_project: row[8],
					})
				}

				recordB, _ := json.Marshal(&record)
				var_.T_record = string(recordB)
				IOTNetworkCardList = append(IOTNetworkCardList, var_)

			}
		}
	}
	o := orm.NewOrm()
	o.Begin()
	IOTNetworkCardDao := IOTNetworkCard.NewIOTNetworkCard(o)

	// 定义批次大小
	batchSize := 1000
	total := len(IOTNetworkCardList)
	for i := 0; i < total; i += batchSize {
		end := i + batchSize
		if end > total {
			end = total
		}
		batch := IOTNetworkCardList[i:end]

		// 执行数据库批量插入
		if _, err := IOTNetworkCardDao.Add_IOTNetworkCardMulti(batch); err != nil {
			o.Rollback()
			c.Data["json"] = lib.JSONS{Code: 203, Msg: "导入失败"}
			c.ServeJSON()
			return // 中断处理流程
		}
	}
	o.Commit()
	// 去重 保留最小值
	IOTNetworkCardDao.DeleteDeduplicate()

	c.Data["json"] = lib.JSONS{Code: 200, Msg: "成功!", Data: nil}
	c.ServeJSON()
}

// 修改移动端设备编号
func (c *IOTNetworkCardController) IOTNetworkCard_Edit_Device_Number() {

	T_id, _ := c.GetInt("T_id")
	T_device_number := c.GetString("T_device_number")
	IOTNetworkCard_r, err := IOTNetworkCard.Read_IOTNetworkCard_ById(T_id)
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
		c.ServeJSON()
		return
	}
	IOTNetworkCard_r.T_device_number = T_device_number
	o := orm.NewOrm()
	IOTNetworkCardDao := IOTNetworkCard.NewIOTNetworkCard(o)
	if err = IOTNetworkCardDao.Update_IOTNetworkCard(IOTNetworkCard_r, "T_device_number"); err != nil {
		c.Data["json"] = lib.JSONS{Code: 203, Msg: "修改失败"}
		c.ServeJSON()
		return
	}

	NatsServer.AddUserLogs(c.User.T_uuid, "物联网卡", "修改移动端设备编号", IOTNetworkCard_r)

	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!"}
	c.ServeJSON()
	return
}

// 导出
func (c *IOTNetworkCardController) IOTNetworkCard_Export() {

	// 分页参数 初始化

	// 查询
	T_name := c.GetString("T_name")
	T_company_name := c.GetString("T_company_name")
	T_state, _ := c.GetInt("T_state")

	userList, _ := NatsServer.Read_User_List_All()
	Account.Read_User_All_Map(userList)
	IOTNetworkCardDao := IOTNetworkCard.NewIOTNetworkCard(orm.NewOrm())
	R_List, _ := IOTNetworkCardDao.Read_IOTNetworkCard_List(T_name, T_company_name, T_state, 0, 9999)
	f := excelize.NewFile() //设置单元格值
	// 这里设置表头ÒÒ
	Style1, _ := f.NewStyle(
		&excelize.Style{
			Font:      &excelize.Font{Size: 20, Family: "宋体"},
			Alignment: &excelize.Alignment{Horizontal: "center", Vertical: "center"},
		})
	Style2, _ := f.NewStyle(
		&excelize.Style{
			Font:      &excelize.Font{Size: 15, Family: "宋体"},
			Alignment: &excelize.Alignment{Horizontal: "center", Vertical: "center", WrapText: true},
			Border: []excelize.Border{
				{Type: "left", Color: "000000", Style: 1},
				{Type: "top", Color: "000000", Style: 1},
				{Type: "bottom", Color: "000000", Style: 1},
				{Type: "right", Color: "000000", Style: 1},
			},
			Fill: excelize.Fill{Type: "pattern", Color: []string{"D9D9D9"}, Pattern: 1},
		})
	Style3, _ := f.NewStyle(
		&excelize.Style{
			Font:      &excelize.Font{Size: 15, Family: "宋体"},
			Alignment: &excelize.Alignment{Horizontal: "center", Vertical: "center", WrapText: true},
			Border: []excelize.Border{
				{Type: "left", Color: "000000", Style: 1},
				{Type: "top", Color: "000000", Style: 1},
				{Type: "bottom", Color: "000000", Style: 1},
				{Type: "right", Color: "000000", Style: 1},
			},
		})

	f.MergeCell("Sheet1", "A1", "M1")
	f.SetCellValue("Sheet1", "A1", "物联网卡")
	f.SetCellStyle("Sheet1", "A1", "M1", Style1)

	// 写入表头
	headers := []string{"序号", "公司名称", "入库日期", "套餐类型", "套餐", "到期&欠费", "状态", "串码", "移动端设备编号", "MSISDN", "出库日期", "出库项目", "领用人"}
	for col, header := range headers {
		cell := fmt.Sprintf("%c%d", rune('A'+col), 2)
		f.SetCellValue("Sheet1", cell, header)
	}
	// 这里设置表头
	f.SetCellStyle("Sheet1", "A2", "M2", Style2)
	f.SetRowHeight("Sheet1", 2, 25)

	colsWidth := []float64{6, 40, 15, 12, 12, 25, 12, 30, 17, 17, 15, 30, 15}

	for col, width := range colsWidth {
		cell := fmt.Sprintf("%c", rune('A'+col))
		f.SetColWidth("Sheet1", cell, cell, width)
	}
	line := 2
	for i, v := range R_List {
		// 写入数据

		line++
		f.SetCellValue("Sheet1", fmt.Sprintf("A%d", line), i+1)
		f.SetCellValue("Sheet1", fmt.Sprintf("B%d", line), v.T_company_name)
		f.SetCellValue("Sheet1", fmt.Sprintf("C%d", line), v.T_in_date)
		f.SetCellValue("Sheet1", fmt.Sprintf("D%d", line), v.T_meal_type)
		f.SetCellValue("Sheet1", fmt.Sprintf("E%d", line), v.T_meal)
		f.SetCellValue("Sheet1", fmt.Sprintf("F%d", line), v.T_expire)
		var t_state string
		switch v.T_State {
		case 1:
			t_state = "新卡入库"
		case 2:
			t_state = "退还入库"
		case 3:
			t_state = "已出库"
		case 4:
			t_state = "损坏退还"
		case 5:
			t_state = "待注销"
		case 6:
			t_state = "已注销"
		}

		f.SetCellValue("Sheet1", fmt.Sprintf("G%d", line), t_state)
		f.SetCellValue("Sheet1", fmt.Sprintf("H%d", line), v.T_iccid)
		f.SetCellValue("Sheet1", fmt.Sprintf("I%d", line), v.T_device_number)
		f.SetCellValue("Sheet1", fmt.Sprintf("J%d", line), v.T_MSISDN)
		f.SetCellValue("Sheet1", fmt.Sprintf("K%d", line), v.T_out_date)
		f.SetCellValue("Sheet1", fmt.Sprintf("L%d", line), v.T_out_project)
		f.SetCellValue("Sheet1", fmt.Sprintf("M%d", line), v.T_receive_name)
	}

	f.SetCellStyle("Sheet1", "A3", fmt.Sprintf("M%d", line), Style3)

	timeStr := time.Now().Format("20060102150405")
	fileName := fmt.Sprintf("物联网卡%v.xlsx", timeStr)
	filePath := "ofile/" + fileName
	// 保存文件
	if err := f.SaveAs(filePath); err != nil {
		logs.Error("保存文件失败:", err)
	}
	defer func() {
		os.Remove(filePath)
	}()
	c.Ctx.Output.Header("Content-Type", "application/vnd.ms-excel;charset=utf8")
	c.Ctx.Output.Header("Content-Disposition", "attachment; filename="+url.PathEscape(fileName))
	c.Ctx.Output.Header("Content-Transfer-Encoding", "binary")
	c.Ctx.Output.Download(filePath, fileName)
}