mirror of https://github.com/rocboss/paopao-ce
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
47 lines
1.2 KiB
47 lines
1.2 KiB
package jinzhu
|
|
|
|
import (
|
|
"github.com/rocboss/paopao-ce/internal/core"
|
|
"github.com/rocboss/paopao-ce/internal/dao/jinzhu/dbr"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type ShareKeyService struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func (s ShareKeyService) DeleteUserKey(UserName string, key string) (core.DeleteKeySuccessOrFail, error) {
|
|
var deleteKey core.DeleteKeySuccessOrFail
|
|
deleteKey.SuccessOrFail = false
|
|
//根据username和key删除相应的key
|
|
err := s.db.Table("p_share_key").Where("user_name = ? AND share_key = ?", UserName, key).Update("status", 1).Error
|
|
if err != nil {
|
|
return deleteKey, err
|
|
}
|
|
deleteKey.SuccessOrFail = true
|
|
return deleteKey, nil
|
|
}
|
|
|
|
func (s ShareKeyService) GetUserKeys(UserName string, offset, limit int) ([]*core.ShareKey, error) {
|
|
statement := &dbr.ShareKey{
|
|
UserName: UserName,
|
|
}
|
|
//fmt.Print(statement.List(s.db, &dbr.ConditionsT{}, offset, limit))
|
|
return statement.List(s.db, &dbr.ConditionsT{}, offset, limit)
|
|
}
|
|
|
|
func (s ShareKeyService) GetUserShareKeyCount(name string) (int64, error) {
|
|
var count int64
|
|
err := s.db.Table("p_share_key").Where("user_name = ? and status = 0", name).Count(&count).Error
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return count, nil
|
|
}
|
|
|
|
func NewShareKeyService(db *gorm.DB) *ShareKeyService {
|
|
return &ShareKeyService{
|
|
db: db,
|
|
}
|
|
}
|