2015-03-24 11:25:26 +00:00
|
|
|
package stringid
|
2015-02-21 04:48:23 +00:00
|
|
|
|
|
|
|
import (
|
|
|
|
"crypto/rand"
|
|
|
|
"encoding/hex"
|
|
|
|
"io"
|
2015-04-10 17:09:08 +00:00
|
|
|
"regexp"
|
2015-02-21 04:48:23 +00:00
|
|
|
"strconv"
|
|
|
|
)
|
|
|
|
|
2015-04-10 17:09:08 +00:00
|
|
|
const shortLen = 12
|
|
|
|
|
2015-05-13 18:48:48 +00:00
|
|
|
var validShortID = regexp.MustCompile("^[a-z0-9]{12}$")
|
|
|
|
|
2015-04-10 17:09:08 +00:00
|
|
|
// Determine if an arbitrary string *looks like* a short ID.
|
|
|
|
func IsShortID(id string) bool {
|
2015-05-13 18:48:48 +00:00
|
|
|
return validShortID.MatchString(id)
|
2015-04-10 17:09:08 +00:00
|
|
|
}
|
|
|
|
|
2015-02-21 04:48:23 +00:00
|
|
|
// TruncateID returns a shorthand version of a string identifier for convenience.
|
|
|
|
// A collision with other shorthands is very unlikely, but possible.
|
|
|
|
// In case of a collision a lookup with TruncIndex.Get() will fail, and the caller
|
|
|
|
// will need to use a langer prefix, or the full-length Id.
|
|
|
|
func TruncateID(id string) string {
|
2015-04-10 17:09:08 +00:00
|
|
|
trimTo := shortLen
|
2015-02-21 04:48:23 +00:00
|
|
|
if len(id) < shortLen {
|
2015-04-10 17:09:08 +00:00
|
|
|
trimTo = len(id)
|
2015-02-21 04:48:23 +00:00
|
|
|
}
|
2015-04-10 17:09:08 +00:00
|
|
|
return id[:trimTo]
|
2015-02-21 04:48:23 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// GenerateRandomID returns an unique id
|
|
|
|
func GenerateRandomID() string {
|
|
|
|
for {
|
|
|
|
id := make([]byte, 32)
|
|
|
|
if _, err := io.ReadFull(rand.Reader, id); err != nil {
|
|
|
|
panic(err) // This shouldn't happen
|
|
|
|
}
|
|
|
|
value := hex.EncodeToString(id)
|
|
|
|
// if we try to parse the truncated for as an int and we don't have
|
|
|
|
// an error then the value is all numberic and causes issues when
|
|
|
|
// used as a hostname. ref #3869
|
|
|
|
if _, err := strconv.ParseInt(TruncateID(value), 10, 64); err == nil {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
return value
|
|
|
|
}
|
|
|
|
}
|