32 lines
645 B
Go
32 lines
645 B
Go
package service
|
||
|
||
import (
|
||
"ego/internal/conf"
|
||
"strings"
|
||
"sync"
|
||
)
|
||
|
||
type SysSequenceService struct {
|
||
mutex sync.Mutex
|
||
tableName string
|
||
}
|
||
|
||
// SysSequenceServiceBuilder 创建一个SysSequenceService
|
||
func SysSequenceServiceBuilder(tableName string) *SysSequenceService {
|
||
return &SysSequenceService{
|
||
tableName: tableName,
|
||
}
|
||
}
|
||
|
||
// GenerateId 根据表名生成ID（使用序列）
|
||
func (s *SysSequenceService) GenerateId() (string, error) {
|
||
s.mutex.Lock()
|
||
defer s.mutex.Unlock()
|
||
var id string
|
||
err := conf.Db.Raw("SELECT NEXTVAL(?)", strings.ToLower(s.tableName)).Scan(&id).Error
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
return id, nil
|
||
}
|