LBRYCommunity/sqlboiler
1
0
Fork 0
Code Issues Pull requests1 Projects Releases Packages Wiki Activity

Add provision for enforced types, eg uuid

Browse source 
* Add enforced to column data
* Add enforced list to driver
* Fix bug in TitleCase (now uses regexp)
* Fix broken zero-value enforced type inserts by using RandomizeEnforcedStruct
This commit is contained in:
Patrick O'brien 2016-08-03 12:05:05 +10:00
parent 39fe91f2cb
commit c7c0fe5c0d
10 changed files with 69 additions and 20 deletions
strmangle
strmangle.go

19
strmangle/strmangle.go
View file

@ -15,8 +15,15 @@ import (
var (
idAlphabet = []byte("abcdefghijklmnopqrstuvwxyz")
uppercaseWords = []string{"id", "uid", "uuid", "guid", "ssn", "tz"}
smartQuoteRgx = regexp.MustCompile(`^(?i)"?[a-z_][_a-z0-9]*"?(\."?[_a-z][_a-z0-9]*"?)*$`)
uppercaseWords = []*regexp.Regexp{
regexp.MustCompile(`^id[0-9]*$`),
regexp.MustCompile(`^uid[0-9]*$`),
regexp.MustCompile(`^uuid[0-9]*$`),
regexp.MustCompile(`^guid[0-9]*$`),
regexp.MustCompile(`^ssn[0-9]*$`),
regexp.MustCompile(`^tz[0-9]*$`),
}
smartQuoteRgx = regexp.MustCompile(`^(?i)"?[a-z_][_a-z0-9]*"?(\."?[_a-z][_a-z0-9]*"?)*$`)
)
// IdentQuote attempts to quote simple identifiers in SQL tatements
@ -90,8 +97,8 @@ func TitleCase(name string) string {
for i, split := range splits {
found := false
for _, uc := range uppercaseWords {
if split == uc {
splits[i] = strings.ToUpper(uc)
if uc.MatchString(split) {
splits[i] = strings.ToUpper(split)
found = true
break
}
@ -122,8 +129,8 @@ func CamelCase(name string) string {
found := false
if i > 0 {
for _, uc := range uppercaseWords {
if split == uc {
splits[i] = strings.ToUpper(uc)
if uc.MatchString(split) {
splits[i] = strings.ToUpper(split)
found = true
break
}