golang/ent
1
0
Fork 0
mirror of https://github.com/ent/ent.git synced 2026-05-24 09:31:56 +03:00
Code Issues Packages Projects Releases Wiki Activity

dialect/sql: skip formatting qualified columns (#3469)

Browse Source
This commit is contained in:
Ariel Mashraki
2023-04-11 13:16:40 +03:00
committed by GitHub
parent da69615bd0
commit 36cd6a8a8b
2 changed files with 38 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
dialect/sql/builder.go
View File

@@ -2612,6 +2612,10 @@ func (s *Selector) Prefix(queries ...Querier) *Selector {
// C returns a formatted string for a selected column from this statement.
func (s *Selector) C(column string) string {
// Skip formatting qualified columns.
if s.isQualified(column) {
return column
}
if s.as != "" {
b := &Builder{dialect: s.dialect}
b.Ident(s.as)
@@ -3723,6 +3727,14 @@ func (b *Builder) isIdent(s string) bool {
}
}
// isIdent reports if the given string is a qualified identifier.
func (b *Builder) isQualified(s string) bool {
ident, pg := b.isIdent(s), b.postgres()
return !ident && len(s) > 2 && strings.ContainsRune(s[1:len(s)-1], '.') || // <qualifier>.<column>
ident && pg && strings.Contains(s, `"."`) || // "qualifier"."column"
ident && !pg && strings.Contains(s, "`.`") // `qualifier`.`column`
}
// state wraps the all methods for setting and getting
// update state between all queries in the query tree.
type state interface {