package IdGenerator

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

type idGenerator interface {
	Next() int64
}

type SnowflakeIdGenerator struct {
	node *snowflake.Node
}

func (s *SnowflakeIdGenerator) Next() int64 {
	return s.node.Generate().Int64()
}

var IdGenerator idGenerator

func init() {
	node, err := snowflake.NewNode(1)
	if err != nil {
		panic(err)
	}
	IdGenerator = &SnowflakeIdGenerator{
		node: node,
	}
}