package dao

import (
	"github.com/druidcaesa/gotool"
	"ulink-admin/frame"
	"ulink-admin/modules/system/models/model"
	"ulink-admin/modules/system/models/req"
	"ulink-admin/pkg/base"
	"ulink-admin/pkg/page"
)

type SysCompanyDao struct {
	base.BaseDao
}

// Page 查询公司管理分页数据
func (this SysCompanyDao) Page(query *req.SysCompanyQuery) (*[]model.SysCompany, int64) {
	sysCompany := make([]model.SysCompany, 0)
	session := this.GetSession().Table(model.SysCompany{}.TableName())
	if query.Id > 0 {
		session.And("id = ?", query.Id)
	}
	if !gotool.StrUtils.HasEmpty(query.Label) {
		session.And("label = ?", query.Label)
	}
	if query.UserId > 0 {
		session.And("user_id = ?", query.UserId)
	}
	if query.IsTip > 0 {
		session.And("is_tip = ?", query.IsTip)
	}
	if query.IsPay > 0 {
		session.And("is_pay = ?", query.IsPay)
	}
	if query.Status > 0 {
		session.And("status = ?", query.Status)
	}
	if !gotool.StrUtils.HasEmpty(query.CreateBy) {
		session.And("create_by = ?", query.CreateBy)
	}
	if !gotool.StrUtils.HasEmpty(query.UpdateBy) {
		session.And("update_by = ?", query.UpdateBy)
	}
	if !gotool.StrUtils.HasEmpty(query.BeginTime) {
		session.And("date_format(u.create_time,'%y%m%d') >= date_format(?,'%y%m%d')", query.BeginTime)
	}
	if !gotool.StrUtils.HasEmpty(query.EndTime) {
		session.And("date_format(u.create_time,'%y%m%d') <= date_format(?,'%y%m%d')", query.EndTime)
	}

	total, err := session.Limit(query.PageSize, page.StartSize(query.PageNum, query.PageSize)).FindAndCount(&sysCompany)
	if err != nil {
		frame.Throw(frame.SQL_CODE, "数据查询错误"+err.Error())
	}
	return &sysCompany, total
}

// List 查询公司管理分页数据
func (this SysCompanyDao) List(query *req.SysCompanyQuery, list interface{}) {
	session := this.GetSession().Table(model.SysCompany{}.TableName())

	if query.Id > 0 {
		session.And("id = ?", query.Id)
	}
	if !gotool.StrUtils.HasEmpty(query.Label) {
		session.And("label = ?", query.Label)
	}
	if query.UserId > 0 {
		session.And("user_id = ?", query.UserId)
	}
	if query.IsTip > 0 {
		session.And("is_tip = ?", query.IsTip)
	}
	if query.IsPay > 0 {
		session.And("is_pay = ?", query.IsPay)
	}
	if query.Status > 0 {
		session.And("status = ?", query.Status)
	}
	if !gotool.StrUtils.HasEmpty(query.CreateBy) {
		session.And("create_by = ?", query.CreateBy)
	}
	if !gotool.StrUtils.HasEmpty(query.UpdateBy) {
		session.And("update_by = ?", query.UpdateBy)
	}
	if !gotool.StrUtils.HasEmpty(query.BeginTime) {
		session.And("date_format(u.create_time,'%y%m%d') >= date_format(?,'%y%m%d')", query.BeginTime)
	}
	if !gotool.StrUtils.HasEmpty(query.EndTime) {
		session.And("date_format(u.create_time,'%y%m%d') <= date_format(?,'%y%m%d')", query.EndTime)
	}

	err := session.Find(list)
	if err != nil {
		frame.Throw(frame.SQL_CODE, "数据查询错误"+err.Error())
	}
}