gocryptotrader/exchanges/nonce/nonce.go

package nonce

import (
	"strconv"
	"sync"
	"time"
)

// Nonce struct holds the nonce value
type Nonce struct {
	n   int64
	mtx sync.Mutex
}

// Inc increments the nonce value
func (n *Nonce) Inc() {
	n.mtx.Lock()
	n.n++
	n.mtx.Unlock()
}

// Get retrives the nonce value
func (n *Nonce) Get() int64 {
	n.mtx.Lock()
	defer n.mtx.Unlock()
	return n.n
}

// GetInc increments and returns the value of the nonce
func (n *Nonce) GetInc() int64 {
	n.mtx.Lock()
	defer n.mtx.Unlock()
	n.n++
	return n.n
}

// Set sets the nonce value
func (n *Nonce) Set(val int64) {
	n.mtx.Lock()
	n.n = val
	n.mtx.Unlock()
}

// Returns a string version of the nonce
func (n *Nonce) String() string {
	n.mtx.Lock()
	result := strconv.FormatInt(n.n, 10)
	n.mtx.Unlock()
	return result
}

// Evaluate returns a nonce while evaluating in a single locked call
func (n *Nonce) Evaluate() string {
	n.mtx.Lock()
	defer n.mtx.Unlock()
	if n.n == 0 {
		n.n = time.Now().Unix()
		return strconv.FormatInt(n.n, 10)
	}
	n.n = n.n + 1
	return strconv.FormatInt(n.n, 10)
}