uid.go

package uid

import (
	"database/sql"
	"errors"
	"log"
	"time"
)

type logger interface {
	Error(error)
}

// Logger Log&#25509;&#21475;&#65292;&#22914;&#26524;&#35774;&#32622;&#20102;Logger&#65292;&#23601;&#20351;&#29992;Logger&#25171;&#21360;&#26085;&#24535;&#65292;&#22914;&#26524;&#27809;&#26377;&#35774;&#32622;&#65292;&#23601;&#20351;&#29992;&#20869;&#32622;&#24211;log&#25171;&#21360;&#26085;&#24535;
var Logger logger

// ErrTimeOut &#33719;&#21462;uid&#36229;&#26102;&#38169;&#35823;
var ErrTimeOut = errors.New("get uid timeout")

type Uid struct {
	db         *sql.DB    // &#25968;&#25454;&#24211;&#36830;&#25509;
	businessId string     // &#19994;&#21153;id
	ch         chan int64 // id&#32531;&#20914;&#27744;
	min, max   int64      // id&#27573;&#26368;&#23567;&#20540;&#65292;&#26368;&#22823;&#20540;
}

// NewUid &#21019;&#24314;&#19968;&#20010;Uid;len&#65306;&#32531;&#20914;&#27744;&#22823;&#23567;()
// db:&#25968;&#25454;&#24211;&#36830;&#25509;
// businessId&#65306;&#19994;&#21153;id
// len&#65306;&#32531;&#20914;&#27744;&#22823;&#23567;(&#38271;&#24230;&#21487;&#25511;&#21046;&#32531;&#23384;&#20013;&#21097;&#19979;&#22810;&#23569;id&#26102;&#65292;&#21435;DB&#20013;&#21152;&#36733;)
func NewUid(db *sql.DB, businessId string, len int) (*Uid, error) {
	lid := Uid{
		db:         db,
		businessId: businessId,
		ch:         make(chan int64, len),
	}
	go lid.productId()
	return &lid, nil
}

// Get &#33719;&#21462;&#33258;&#22686;id,&#24403;&#21457;&#29983;&#36229;&#26102;&#65292;&#36820;&#22238;&#38169;&#35823;&#65292;&#36991;&#20813;&#22823;&#37327;&#35831;&#27714;&#38459;&#22622;&#65292;&#26381;&#21153;&#22120;&#23849;&#28291;
func (u *Uid) Get() (int64, error) {
	select {
	case <-time.After(1 * time.Second):
		return 0, ErrTimeOut
	case uid := <-u.ch:
		return uid, nil
	}
}

// productId &#29983;&#20135;id&#65292;&#24403;ch&#36798;&#21040;&#26368;&#22823;&#23481;&#37327;&#26102;&#65292;&#36825;&#20010;&#26041;&#27861;&#20250;&#38459;&#22622;&#65292;&#30452;&#21040;ch&#20013;&#30340;id&#34987;&#28040;&#36153;
func (u *Uid) productId() {
	u.reLoad()

	for {
		if u.min >= u.max {
			u.reLoad()
		}

		u.min++
		u.ch <- u.min
	}
}

// reLoad &#22312;&#25968;&#25454;&#24211;&#33719;&#21462;id&#27573;&#65292;&#22914;&#26524;&#22833;&#36133;&#65292;&#20250;&#27599;&#38548;&#19968;&#31186;&#23581;&#35797;&#19968;&#27425;
func (u *Uid) reLoad() error {
	var err error
	for {
		err = u.getFromDB()
		if err == nil {
			return nil
		}

		// &#25968;&#25454;&#24211;&#21457;&#29983;&#24322;&#24120;&#65292;&#31561;&#24453;&#19968;&#31186;&#20043;&#21518;&#20877;&#27425;&#36827;&#34892;&#23581;&#35797;
		if Logger != nil {
			Logger.Error(err)
		} else {
			log.Println(err)
		}
		time.Sleep(time.Second)
	}
}

// getFromDB &#20174;&#25968;&#25454;&#24211;&#33719;&#21462;id&#27573;
func (u *Uid) getFromDB() error {
	var (
		maxId int64
		step  int64
	)

	tx, err := u.db.Begin()
	if err != nil {
		return err
	}
	defer tx.Rollback()

	row := tx.QueryRow("SELECT max_id,step FROM uid WHERE business_id = ? FOR UPDATE", u.businessId)
	err = row.Scan(&maxId, &step)
	if err != nil {
		return err
	}

	_, err = tx.Exec("UPDATE uid SET max_id = ? WHERE business_id = ?", maxId+step, u.businessId)
	if err != nil {
		return err
	}
	err = tx.Commit()
	if err != nil {
		return err
	}

	u.min = maxId
	u.max = maxId + step
	return nil
}