You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
31 lines
736 B
Go
31 lines
736 B
Go
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
|
|
}
|