package report

import (
	"encoding/json"
	"io/ioutil"
	"kingdee-dbapi/logger"

	"kingdee-dbapi/cache"
	"kingdee-dbapi/config"
	"kingdee-dbapi/kingdee"
	"kingdee-dbapi/models"
	"kingdee-dbapi/nsqclient"
)

const orderLocalStore = "order.tmp"
const inventoryLocalStore = "inventory.tmp"

func SendOrder() {
	var list []kingdee.SEOrder

	if config.Options.Debug {
		data, err := ioutil.ReadFile(orderLocalStore)
		if err != nil {
			logger.Error("文件读取失败, %s", err.Error())
			return
		}
		err = json.Unmarshal(data, &list)
		if err != nil {
			logger.Error("文件内容解析失败, %s", err.Error())
			return
		}
	} else {
		list = kingdee.SeOrderList()
		logger.Debug("查询到%d条订单信息", len(list))
	}

	var completedOrderNo = make(map[string]struct{})

	for i := 0; i < len(list); {
		if cache.Exists(list[i].FBillNo) {
			list = append(list[:i], list[i+1:]...)
		} else {
			completedOrderNo[list[i].FBillNo] = struct{}{}
			i++
		}
	}

	b, _ := json.Marshal(list)

	if !config.Options.Debug {
		ioutil.WriteFile(orderLocalStore, b, 0644)
	}

	// http协议上报, 已修改为TCP
	//ok := nsqclient.HttpPost(config.Options.OrderTopic, b)

	if len(list) == 0 {
		logger.Debug("没有新的订单需要上报")
		return
	}

	// TCP协议上报
	ok := nsqclient.Produce(config.Options.OrderTopic, b)
	if ok {
		// 写入数据库, 标记已经上报过了,避免重复上报
		for orderNo, _ := range completedOrderNo {
			cursor := models.Order{
				OrderNo: orderNo,
			}

			cursor.Insert()
			cache.WriteCache(orderNo)
		}

		logger.Debug("已上报%d个订单信息", len(list))
	} else {
		logger.Warn("订单数据上报失败")
	}
}

func SendInventory() {
	var list []kingdee.Inventory

	if config.Options.Debug {
		data, err := ioutil.ReadFile(inventoryLocalStore)
		if err != nil {
			logger.Error("文件读取失败, %s", err.Error())
			return
		}

		err = json.Unmarshal(data, &list)
		if err != nil {
			logger.Error("文件内容解析失败, %s", err.Error())
			return
		}
	} else {
		list = kingdee.ICInventory()
		logger.Debug("查询到%d条库存数据", len(list))
	}

	// 每次发 100 条
	successCnt := 0
	for i := 0; i < len(list); i += 1000 {
		end := i + 1000
		if end > len(list) {
			end = len(list)
		}

		b, _ := json.Marshal(list[i:end])

		if !config.Options.Debug {
			ioutil.WriteFile(inventoryLocalStore, b, 0644)
		}

		// HTTP协议上报,已修改为TCP
		//nsqclient.HttpPost(config.Options.InventoryTopic, b)

		// TCP协议上报
		ok := nsqclient.Produce(config.Options.InventoryTopic, b)
		if !ok {
			logger.Warn("库存数据上报失败")
		} else {
			successCnt += end
		}
	}
	logger.Debug("已上报%d条库存数据", successCnt)
}