package main

import (
	"data_msg_push_server/config"
	"data_msg_push_server/service"

	"fmt"
	"time"

	"basic.com/valib/logger.git"
)

func init() {
	config.Init()
}

func main() {
	done := make(chan bool)
	// 日志初始化
	var logFile = config.LogConf.Path + "esSyncClient.log"
	logger.InitLogger(logFile, config.LogConf.Level, config.LogConf.MaxSize, config.LogConf.MaxBackups, config.LogConf.MaxAge)

	go runEvery(done)
	<-done
	logger.Debug("程序已退出")
}

func runEvery(done chan<- bool) {
	ticker := time.NewTicker(time.Duration(config.Options.Interval) * time.Second)
	defer ticker.Stop()

	for {
		select {
		case <-ticker.C:
			logger.Debug("开始执行推送任务")
			doServer()
			logger.Debug("本次执行完毕, 当前所有数据已发送！！！")
			time.Sleep(10 * time.Second)
		}
	}
}

func doServer() {
	total := 0
	retryCount := 0
	for {
		t, err := service.GetTotal()
		if err != nil {
			logger.Error(err.Error())
			return
		}
		total = t
		logger.Debugf("未发送的数据总量:%d, 重试次数:%d", total, retryCount)
		if total == 0 || retryCount > 10 {
			break
		} else {
			for {
				connectStatus := service.ConnectControl()
				logger.Debug("服务端连接状态connectStatus: ", connectStatus)
				if connectStatus == true {
					url := config.Options.ServerUrl
					result, err1 := service.GetData()
					if err1 != nil {
						logger.Warn("获取es数据失败,", err1.Error())
						return
					}

					id, err := service.SendData(result, url)
					logger.Debug("推送的数据 id: ", id)
					if err == nil {
						markStatus := service.MarkData(id)
						logger.Debug("记录推送状态 markStatus: ", markStatus)
					} else {
						logger.Debugf("推送的数据失败 id: %s, err:%s", id, err.Error())
						time.Sleep(5 * time.Second)
						retryCount += 1
					}
					break
				} else {
					logger.Debug("服务端连接失败, 5秒后尝试重新连接。。。")
					time.Sleep(5 * time.Second)
				}
			}
		}
	}
}