package model

import "gorm.io/gorm"

type AgentAdmin struct {
	AgentID int64  `json:"agent_id" gorm:"comment:'用户ID'"`
	AdminID string `json:"admin_id" gorm:"comment:'管理员ID'"`
}

func (AgentAdmin) TableName() string {
	return "sg_agent_admin"
}

func GetAgentByAgentId(db *gorm.DB, agent_id int64) (AgentAdmin, error) {
	r := AgentAdmin{}
	err := db.Where("agent_id = ? ", agent_id).First(&r).Error
	if err != nil && err != gorm.ErrRecordNotFound {
		return r, err
	}
	return r, nil
}

func GetAgentByAdminId(db *gorm.DB, admin_id string) (AgentAdmin, error) {
	r := AgentAdmin{}
	err := db.Where("admin_id = ? ", admin_id).First(&r).Error
	if err != nil && err != gorm.ErrRecordNotFound {
		return r, err
	}
	return r, nil
}