HiPoweredDev
/
xorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

names with upper charactor on postgres will need quotes

This commit is contained in:
Lunny Xiao 2020-03-06 15:28:29 +08:00
parent 717aabf022
commit e86b5375a5
No known key found for this signature in database
GPG Key ID: C3B7C91B632F738A
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
dialects/postgres.go
View File

@ -788,6 +788,18 @@ func (db *postgres) Init(d *core.DB, uri *URI) error {
return nil return nil
} }
func (db *postgres) needQuote(name string) bool {
if db.IsReserved(name) {
return true
}
for _, c := range name {
if c >= 'A' && c <= 'Z' {
return true
}
}
return false
}
func (db *postgres) SetQuotePolicy(quotePolicy QuotePolicy) { func (db *postgres) SetQuotePolicy(quotePolicy QuotePolicy) {
switch quotePolicy { switch quotePolicy {
case QuotePolicyNone: case QuotePolicyNone:
@ -796,7 +808,7 @@ func (db *postgres) SetQuotePolicy(quotePolicy QuotePolicy) {
db.quoter = q db.quoter = q
case QuotePolicyReserved: case QuotePolicyReserved:
var q = postgresQuoter var q = postgresQuoter
q.IsReserved = db.IsReserved q.IsReserved = db.needQuote
db.quoter = q db.quoter = q
case QuotePolicyAlways: case QuotePolicyAlways:
fallthrough fallthrough