package controllers

import (
	"basic.com/pubsub/esutil.git"
	"basic.com/valib/logger.git"
	"github.com/gin-gonic/gin"
	"strconv"
	"webserver/cache"
	"webserver/extend/code"
	"webserver/extend/config"
	"webserver/extend/util"
)

type ClearDataController struct{}

type clearParams struct {
	StartTime string `json:"startTime"`
	EndTime   string `json:"endTime"`
}

func (cdc *ClearDataController) ClearEsData(c *gin.Context) {
	var cp clearParams
	localConf, err := cache.GetServerInfo()
	if err != nil || localConf.AlarmIp == "" || localConf.ServerId == "" {
		logger.Debug("localConfig is wrong!!!")
		util.ResponseFormat(c, code.ComError, "es config err")
		return
	}
	ip := localConf.AlarmIp
	port := strconv.Itoa(int(localConf.AlarmPort))
	analyServerId := localConf.ServerId
	c.BindJSON(&cp)
	startTime := cp.StartTime
	endTime := cp.EndTime
	queryRes, queryErr := esutil.QueryAnalyServerData(ip, port, config.EsInfo.EsIndex.AiOcean.IndexName, startTime, endTime, analyServerId)
	if queryErr != nil {
		util.ResponseFormat(c, code.InvalidRequest, "校验数据失败")
		return
	}
	if queryRes != true {
		util.ResponseFormat(c, code.InvalidRequest, "数据已删除或该时间段无数据")
		return
	}
	addRes, addErr := esutil.AddDelTask(ip, port, config.BasicFS.IndexName, startTime, endTime, analyServerId)
	if addErr != nil {
		util.ResponseFormat(c, code.InvalidRequest, "追加任务失败")
		return
	}
	if addRes != true {
		util.ResponseFormat(c, code.InvalidRequest, "追加任务失败")
		return
	}
	deleteRes, deleteErr := esutil.DeleteAnalyServerData(ip, port, config.BasicFS.IndexName, startTime, endTime, analyServerId)
	if deleteErr != nil {
		util.ResponseFormat(c, code.InvalidRequest, "数据删除失败")
		return
	}
	if deleteRes != true {
		util.ResponseFormat(c, code.InvalidRequest, "数据删除失败")
		return
	}
	util.ResponseFormat(c, code.Success, deleteRes)
	return
}