package models
|
|
import "errors"
|
|
type Oauth2Client struct {
|
ID string `gorm:"column:id;not null;default:''" json:"id"` //模块名称
|
Secret string `gorm:"column:secret;not null;default:''" json:"secret"` //中文名称
|
Domain string `gorm:"column:domain;not null;default:''" json:"domain"` //中文名称
|
Intro string `gorm:"column:intro;not null;default:''" json:"intro"` //说明
|
}
|
|
func (Oauth2Client) TableName() string {
|
return "t_oauth2_client"
|
}
|
|
func (a *Oauth2Client) FindAll() (rows []Oauth2Client, err error) {
|
if err := oauth2DB.Table(a.TableName()).Find(&rows).Error; err != nil {
|
return nil, err
|
}
|
|
return rows, nil
|
}
|
|
func (a *Oauth2Client) Insert() error {
|
count := 0
|
result := oauth2DB.Table(a.TableName()).Where("domain=?", a.Domain).Count(&count)
|
if count > 0 {
|
return errors.New("回调地址已存在!")
|
}
|
result = oauth2DB.Table(a.TableName()).Create(&a)
|
if result.Error != nil {
|
return result.Error
|
}
|
return nil
|
}
|
|
func (a *Oauth2Client) DeleteById(id string) error {
|
result := oauth2DB.Exec("delete from "+a.TableName()+" where id=?", id)
|
if result.Error != nil {
|
return result.Error
|
}
|
if result.RowsAffected > 0 {
|
return nil
|
}
|
return errors.New("删除失败")
|
}
|
