77 lines
1.6 KiB
Go
77 lines
1.6 KiB
Go
package mixrepoimpl
|
|
|
|
import (
|
|
"context"
|
|
"github.com/bwmarrin/snowflake"
|
|
"hash/fnv"
|
|
"math"
|
|
"os"
|
|
"voucher/internal/biz/mixrepos"
|
|
"voucher/internal/data"
|
|
"voucher/internal/pkg/uid"
|
|
)
|
|
|
|
type GenerateRepoImpl struct {
|
|
rdb *data.Rdb
|
|
|
|
node *snowflake.Node
|
|
}
|
|
|
|
func NewGenerateMixRepoImpl(rdb *data.Rdb) (mixrepos.GenerateMixRepo, error) {
|
|
|
|
g := &GenerateRepoImpl{rdb: rdb}
|
|
|
|
name, err := os.Hostname()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
serverId := g.hashMod(name)
|
|
|
|
node, err := snowflake.NewNode(int64(serverId))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
g.node = node
|
|
|
|
return g, nil
|
|
}
|
|
|
|
// GeneratorString 生成字符串
|
|
func (s *GenerateRepoImpl) GeneratorString(_ context.Context, workId int) (string, error) {
|
|
|
|
return s.node.Generate().String(), nil
|
|
|
|
//serverIdStr := uid.GetAppId(ctx)
|
|
//serverId := s.hashMod(serverIdStr)
|
|
//
|
|
//number, err := uid.NewSignGenerator(s.rdb.Rdb).SetWorkerID(workId).SetServerID(serverId).GetNumber()
|
|
//if err != nil {
|
|
// return "", err
|
|
//}
|
|
//
|
|
//return strconv.FormatUint(number, 10), nil
|
|
}
|
|
|
|
// GeneratorNumber 生成unit64
|
|
func (s *GenerateRepoImpl) GeneratorNumber(ctx context.Context, workId int) (uint64, error) {
|
|
serverIdStr := uid.GetAppId(ctx)
|
|
serverId := s.hashMod(serverIdStr)
|
|
|
|
number, err := uid.NewSignGenerator(s.rdb.Rdb).SetWorkerID(workId).SetServerID(serverId).GetNumber()
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
return number, nil
|
|
}
|
|
|
|
// hashMod hash mod
|
|
func (s *GenerateRepoImpl) hashMod(hashStr string) int {
|
|
hash := fnv.New32a()
|
|
_, _ = hash.Write([]byte(hashStr))
|
|
|
|
hashValue := hash.Sum32()
|
|
return int(math.Mod(float64(hashValue), 32))
|
|
}
|