74 lines
1.6 KiB
Go
74 lines
1.6 KiB
Go
package service
|
|
|
|
import (
|
|
"blazing/cool"
|
|
"blazing/modules/config/model"
|
|
"context"
|
|
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/util/grand"
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
const charsetWithSymbol = "0123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghjkmnpqrstuvwxyz"
|
|
|
|
func Generate16CharSecure() string {
|
|
result := make([]byte, 16)
|
|
for i := 0; i < 16; i++ {
|
|
result[i] = charsetWithSymbol[grand.N(0, len(charsetWithSymbol)-1)]
|
|
}
|
|
return string(result)
|
|
}
|
|
|
|
type CdkService struct {
|
|
*cool.Service
|
|
}
|
|
|
|
func NewCdkService() *CdkService {
|
|
return &CdkService{
|
|
&cool.Service{
|
|
Model: model.NewCDKConfig(),
|
|
InsertParam: func(ctx context.Context) g.MapStrAny {
|
|
uuid.NewV7()
|
|
return g.MapStrAny{
|
|
"cdk_code": Generate16CharSecure(),
|
|
}
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func (s *CdkService) Get(id string) *model.CDKConfig {
|
|
var item *model.CDKConfig
|
|
dbm_notenable(s.Model).Where("cdk_code", id).WhereNot("exchange_remain_count", 0).Scan(&item)
|
|
return item
|
|
}
|
|
|
|
func (s *CdkService) GetByID(id uint32) *model.CDKConfig {
|
|
if id == 0 {
|
|
return nil
|
|
}
|
|
|
|
var item *model.CDKConfig
|
|
dbm_notenable(s.Model).Where("id", id).Scan(&item)
|
|
return item
|
|
}
|
|
|
|
func (s *CdkService) All() []model.CDKConfig {
|
|
var item []model.CDKConfig
|
|
dbm_notenable(s.Model).WhereLT("exchange_remain_count", 0).Scan(&item)
|
|
return item
|
|
}
|
|
|
|
func (s *CdkService) Set(id string) bool {
|
|
res, err := cool.DBM(s.Model).Where("cdk_code", id).WhereNot("exchange_remain_count", 0).Decrement("exchange_remain_count", 1)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
rows, _ := res.RowsAffected()
|
|
if rows == 0 {
|
|
return false
|
|
}
|
|
return true
|
|
}
|