jypkg/public/ordercode.go

package public

import (
	qutil "app.yhyue.com/moapp/jybase/common"
	"fmt"
	"sync"
	"time"
)

var (
	VarOrderCode *orderCode
)

type orderCode struct {
	pool chan string
	all  *sync.Map
	lock *sync.Mutex
}

func (o *orderCode) gc() {
	VarOrderCode.all.Range(func(key, value interface{}) bool {
		if time.Now().Day() != value.(int) {
			VarOrderCode.all.Delete(key)
		}
		return true
	})
	time.AfterFunc(24*time.Hour, o.gc)
}
func init() {
	VarOrderCode = &orderCode{
		pool: make(chan string, 20),
		all:  &sync.Map{},
		lock: &sync.Mutex{},
	}
	VarOrderCode.gc()
	for i := 0; i < 10; i++ {
		go func() {
			VarOrderCode.lock.Lock()
			defer VarOrderCode.lock.Unlock()
			for {
				o := fmt.Sprintf("%d%s", time.Now().Unix()+900000000, qutil.GetRandom(2))
				if _, ok := VarOrderCode.all.Load(o); ok {
					continue
				}
				VarOrderCode.all.Store(o, time.Now().Day())
				VarOrderCode.pool <- o
			}
		}()
	}
}