idgenerator

idgenerator is an id generator based on redis

dgenerator是基于redis的id生成器

Installation

获取

go get github.com/lbfatcgf/idgenerator

Quick start

package main

import (
	"fmt"
	"net/http"
	"os"
	"os/signal"
	"syscall"

	"github.com/go-redis/redis"
	"github.com/lbfatcgf/idgenerator"
	"github.com/lbfatcgf/idgenerator/recorder"
)

func main() {
	c := redis.NewClient(&redis.Options{
		Addr:     "192.168.122.208:6379",
		Password: "",
		DB:       0,
	})
	r := recorder.NewRedisRecorder(c, fmt.Sprintf("idg_%d", 0xff), 0)

	conf := idgenerator.NewIDConfig()

	// idg, err :=idgenerator.NewIdGenerator(conf, 0xff, r)
	idg, err := idgenerator.NewCachedIdGenerator(conf, 0xff, r)

	if err != nil {
		fmt.Println(err)
		return
	}

	go func() {
		c := make(chan os.Signal)
		signal.Notify(c, syscall.SIGHUP, syscall.SIGINT, syscall.SIGQUIT, syscall.SIGTSTP)

		 <-c
		idg.IDRecorder.(recorder.Recorder).Close()
		os.Exit(0)
	}()
	
	http.HandleFunc("/", func(rw http.ResponseWriter, r *http.Request) {
		id, _ := idg.GetID()
		rw.Header().Set("content-type","text/json")
		rw.WriteHeader(http.StatusOK)
		rw.Write([]byte(fmt.Sprintf("{\"id\":%d}",id)))
	})
	http.ListenAndServe(":8080",nil)
}

The record customization in id can be realized through the Recorder interface

可以通过Recorder接口实现在id的记录自定义

//Recorder 记录器接口
type Recorder interface {
	Save(data uint64)      //Save id to recorder.   保存id到记录器
	Load() (uint64, error) //Load id from recorder. 从记录器加载id
	IsClose() bool         //Get closed status.     获取关闭状态
	Close()                //Turn off the recorder. 关闭记录器
}