package dao

import (
	"context"
	"gorm.io/gorm"
)

type LoadBalance struct {
	ID                     int64  `json:"id" gorm:"primary_key"`
	ServiceID              int64  `json:"service_id" gorm:"column:service_id" description:"服务id	"`
	RoundType              int    `json:"round_type" gorm:"column:round_type" description:"轮询方式 round/weight_round/random/ip_hash"`
	ProxyName              string `json:"proxy_name" gorm:"column:proxy_name" description:"被代理服务名"`
	UpstreamConnectTimeout int    `json:"upstream_connect_timeout" gorm:"column:upstream_connect_timeout" description:"下游建立连接超时, 单位s"`
	UpstreamHeaderTimeout  int    `json:"upstream_header_timeout" gorm:"column:upstream_header_timeout" description:"下游获取header超时, 单位s	"`
	UpstreamIdleTimeout    int    `json:"upstream_idle_timeout" gorm:"column:upstream_idle_timeout" description:"下游链接最大空闲时间, 单位s	"`
	UpstreamMaxIdle        int    `json:"upstream_max_idle" gorm:"column:upstream_max_idle" description:"下游最大空闲链接数"`
}

func (t *LoadBalance) TableName() string {
	return "service_load_balance"
}

type LoadBalanceDao struct {
	*gorm.DB
}

func NewLoadBalanceDao(ctx context.Context, db ...*gorm.DB) *LoadBalanceDao {
	if db != nil {
		return &LoadBalanceDao{db[0]}
	} else {
		return &LoadBalanceDao{NewDBClient(ctx)}
	}
}

func (d *LoadBalanceDao) Find(search *LoadBalance) (*LoadBalance, error) {
	model := &LoadBalance{}
	err := d.Where(search).Find(model).Error
	return model, err
}

func (d *LoadBalanceDao) Save(loadBalance *LoadBalance) error {
	return d.DB.Save(loadBalance).Error
}

func (d *LoadBalanceDao) Delete(ID int64) error {
	return d.Where("service_id = ?", ID).Delete(&LoadBalance{}).Error
}