stu_uuos/util/snowflake.go

package util

import (
	"fmt"
	"github.com/bwmarrin/snowflake"
)

var node *snowflake.Node

func init() {
	n, err := snowflake.NewNode(1)
	if err != nil {
		fmt.Println(err)
		return
	}
	node = n
}

func GenCodeInt() int64 {
	return node.Generate().Int64()
}

func GenCodeStr() string {
	return node.Generate().String()
}