package models

import (
	"basic.com/pubsub/protomsg.git"
	"github.com/pkg/errors"
	"strconv"
	"vamicro/config"
)

type Dbtables struct {
	BaseEntity
	TableName     string `gorm:"column:tableName" json:"tableName" example:"底库123"`
	TableDesc     string `gorm:"column:tableDesc" json:"tableDesc" example:"底库描述"`
	TableType     string `gorm:"column:tableType" json:"tableType" example:"person,car"`
	BwType        string `gorm:"column:bwType" json:"bwType" example:"黑名单:1,白名单:0"`
	StartTime     string `gorm:"column:startTime" json:"startTime" example:"2019-01-12 12:14:56"`
	EndTime       string `gorm:"column:endTime" json:"endTime" example:"2019-10-12 12:14:56"`
	IsSync        string `gorm:"column:isSync" json:"isSync" example:"1:同步到管理平台，2：不同步"`
	AnalyServerId string `gorm:"column:analyServerId" json:"analyServerId" example:"本地库所属的分析设备id，如果是同步库就不需要记录"`
}

func (dt *Dbtables) Add() (bool, error) {
	if dt.AnalyServerId != "" { //本地库
		db.LogMode(false)
		defer db.LogMode(true)
	}
	result := db.Table("dbtables").Create(&dt)
	if result.Error != nil {
		return false, result.Error
	}
	if result.RowsAffected > 0 {
		go SyncSdkCompareCache(dt.AnalyServerId, protomsg.EsCacheChanged_T_DbTable,
			dt.Id, "", "", protomsg.DbAction_Insert, int32(dt.Enable), "")
		return true, nil
	}
	return false, errors.New("dbtable create err")
}

func (dt *Dbtables) Update() (bool, error) {
	if dt.AnalyServerId != "" { //本地库不同步
		db.LogMode(false)
		defer db.LogMode(true)
	}
	sql := "update dbtables set tableName=?,tableDesc=?,tableType=?,bwType=?,startTime=?,endTime=?,isSync=?,updateTime=?,enable=? where id=?"
	result := db.Exec(sql, dt.TableName, dt.TableDesc, dt.TableType, dt.BwType, dt.StartTime, dt.EndTime, dt.IsSync, dt.UpdateTime, dt.Enable, dt.Id)
	//result := db.Table("dbtables").Where("id=?",dt.Id).Update(&dt)
	if result.Error != nil {
		return false, result.Error
	}
	if result.RowsAffected > 0 {
		go SyncSdkCompareCache(dt.AnalyServerId, protomsg.EsCacheChanged_T_DbTable,
			dt.Id, "", "", protomsg.DbAction_Update, int32(dt.Enable), "")

		return true, nil
	}
	return false, errors.New("dbtable update err")
}

func (dt *Dbtables) FindByName(name string) (list []Dbtables, err error) {
	err = db.Table("dbtables").Where("tableName=? and isDelete=0", name).Scan(&list).Error
	return list, err
}

func (dt *Dbtables) UpdateEnableStatus(id string, enable int) (bool, error) {
	var dtTmp Dbtables
	rows, _ := dtTmp.SelectById(id)
	if rows > 0 {
		if dtTmp.AnalyServerId != "" { //本地库
			db.LogMode(false)
			defer db.LogMode(true)
		}
		sql := ""

		/*
			// 不明白之前的逻辑为什么关闭的时候要修改到期时间. 暂时先取消 20221117
			if enable ==0 {//切换为关闭
				//如果截止日期为当前时间以后，则强制将截止日期改为过去时间
				var flag = false
				if dtTmp.EndTime != "" {
					ct := time.Now()
					et, _ := time.ParseInLocation("2006-01-02 15:04:05", dtTmp.EndTime, time.Local)
					if et.After(ct) {
						flag = true
					}
				} else {
					flag = true
				}
				if flag {
					yesTime := time.Now().AddDate(0,0,-1).Format("2006-01-02")+" 23:59:59"
					sql = "update dbtables set enable="+strconv.Itoa(enable)+",endTime='"+yesTime+"' where id='"+id+"'"
				} else {
					sql = "update dbtables set enable="+strconv.Itoa(enable)+" where id='"+id+"'"
				}

			} else {
				sql = "update dbtables set enable="+strconv.Itoa(enable)+" where id='"+id+"'"
			}
		*/

		sql = "update dbtables set enable=" + strconv.Itoa(enable) + " where id='" + id + "'"

		result := db.Exec(sql)
		if result.Error != nil {
			return false, result.Error
		}
		if result.RowsAffected > 0 {
			go SyncSdkCompareCache(dtTmp.AnalyServerId, protomsg.EsCacheChanged_T_DbTable,
				id, "", "", protomsg.DbAction_Update, int32(enable), "")

			return true, nil
		}
	}

	return false, errors.New("dbtable not exist")
}

func (dt *Dbtables) SelectById(id string) (rows int64, err error) {
	dbselect := db.Table("dbtables").Where("id = ?", id).First(&dt)
	if dbselect.Error != nil || dbselect.RowsAffected == 0 {
		return 0, err
	}
	return dbselect.RowsAffected, nil
}

func (dt *Dbtables) DeleteById(id string) (b bool, err error) {
	var dtTmp Dbtables
	rows, _ := dtTmp.SelectById(id)
	if rows > 0 {
		if dtTmp.AnalyServerId != "" { //本地库
			db.LogMode(false)
			defer db.LogMode(true)
		}
		tx := db.Begin()
		defer func() {
			if tx != nil && err != nil {
				tx.Rollback()
			}
		}()

		result := tx.Exec("update dbtables set isDelete=1 where id=?", id)
		err = result.Error
		if err != nil {
			return false, err
		}
		//底库被删除则底库下的人都置为删除状态
		err = tx.Exec("update dbtablepersons set isDelete=1 where tableId=?", id).Error
		if err != nil {
			return false, err
		}
		tx.Commit()
		if result.RowsAffected > 0 {
			go SyncSdkCompareCache(dtTmp.AnalyServerId, protomsg.EsCacheChanged_T_DbTable,
				id, "", "", protomsg.DbAction_Delete, 0, "")
			return true, nil
		}
	}

	return false, errors.New("dbtable not exist")
}

//查同步库或者本地库
//isSync:1表示查同步库,analyServerId为空，isSync:2表示查本地库，analyServerId为当前服务器的serverId
func (dt *Dbtables) FindByIsSync(tableName string, isSync string) (arr []Dbtables, err error) {
	if config.Server.AnalyServerId == "" {
		return nil, errors.New("localConfig select err")
	}
	sql := "select * from dbtables where isDelete=0"
	if tableName != "" {
		sql += " and tableName like '%" + tableName + "%'"
	}
	if isSync == "1" { //同步库
		sql += " and (analyServerId ='' or analyServerId is NULL)"
	} else if isSync == "2" { //本地库
		sql += " and analyServerId ='" + config.Server.AnalyServerId + "'"
	}

	err = db.Raw(sql).Scan(&arr).Error
	if err != nil {
		return nil, err
	}
	return
}

func (dt *Dbtables) FindByDbtIds(ids []string) (arr []Dbtables, err error) {
	str := ""
	for _, id := range ids {
		str += `'` + id + `',`
	}
	str = str[:len(str)-1]
	sql := "select * from dbtables where id in (" + str + ")"
	err = db.Raw(sql).Scan(&arr).Error
	if err != nil {
		return nil, err
	}
	return
}

func (dt *Dbtables) FindAllDbTablesByCurServer(isDelete int, typ string) (arr []Dbtables, err error) {
	if config.Server.AnalyServerId == "" {
		return nil, errors.New("localConfig select err")
	}
	sql := "select * from dbtables dbt where (analyServerId='" + config.Server.AnalyServerId + "' or (analyServerId='' or analyServerId is NULL))" //查本地库
	if isDelete == 0 || isDelete == 1 {
		sql += " and isDelete=" + strconv.Itoa(isDelete) + ""
	}
	if typ != "" {
		sql += " and tableType='" + typ + "'"
	}
	err = db.Raw(sql).Find(&arr).Error
	if err != nil {
		return nil, err
	}
	return
}

func (dt *Dbtables) FindAllDbTables() (arr []Dbtables, err error) {
	err = db.Raw("select * from dbtables").Scan(&arr).Error
	if err != nil {
		return nil, err
	}
	return arr, nil
}

func (dt *Dbtables) FindAllLocalDbTables(isDelete string) (arr []Dbtables, err error) {
	if config.Server.AnalyServerId == "" {
		return nil, errors.New("server config err")
	}
	sql := "select * from dbtables where analyServerId='" + config.Server.AnalyServerId + "'"
	if isDelete == "0" || isDelete == "1" {
		sql = sql + " and isDelete=" + isDelete + ""
	}

	err = db.Raw(sql).Scan(&arr).Error
	if err != nil {
		return nil, err
	}
	return
}