41 lines
820 B
Go
41 lines
820 B
Go
package mysql
|
|
|
|
import "myschools.me/heritage/heritage-api/model"
|
|
|
|
func PermissionDefined(permissionCode *string) (bool, error) {
|
|
db, err := newDB()
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
var count int64
|
|
if err := db.Model(&model.Permission{}).Where("code = ?", *permissionCode).Count(&count).Error; err != nil {
|
|
return false, err
|
|
}
|
|
return count > 0, nil
|
|
}
|
|
|
|
func PermissionCount() (*int64, error) {
|
|
db, err := newDB()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var count int64
|
|
if err := db.Model(&model.Permission{}).Count(&count).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &count, nil
|
|
}
|
|
|
|
func PermissionCreate(obj *model.Permission) (*model.Permission, error) {
|
|
db, err := newDB()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err := db.Create(obj).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return obj, nil
|
|
}
|