35 lines
609 B
Go
35 lines
609 B
Go
package tools
|
||
|
||
import (
|
||
"sync"
|
||
)
|
||
|
||
type ProductCounter struct {
|
||
mu sync.Mutex
|
||
counters map[string]int
|
||
}
|
||
|
||
func NewProductCounter() *ProductCounter {
|
||
return &ProductCounter{
|
||
counters: make(map[string]int),
|
||
}
|
||
}
|
||
|
||
func (pc *ProductCounter) GetNextProductNumber(categoryID string) int {
|
||
pc.mu.Lock()
|
||
defer pc.mu.Unlock()
|
||
|
||
// 如果之前没有这个分类的计数器，初始化为1
|
||
if _, ok := pc.counters[categoryID]; !ok {
|
||
pc.counters[categoryID] = 1
|
||
}
|
||
|
||
// 生成商品编号
|
||
productNumber := pc.counters[categoryID]
|
||
|
||
// 增加计数器
|
||
pc.counters[categoryID]++
|
||
|
||
return productNumber
|
||
}
|