scriptoriumadmin/featurebase
1
0
Fork 0
mirror of https://github.com/featurebasedb/featurebase.git synced 2026-09-07 17:15:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

wip

Browse source
This commit is contained in:
Todd Gruben 2021-09-23 11:05:05 -05:00
parent d6dd1e025b
commit 5bc1364cdb
1 changed files with 15 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
pg/protocol.go
View file

@ -302,9 +302,16 @@ func (p *Portal) Bind() {
}
func (p *Portal) Parse(data []byte) {
p.queryStart = time.Now()
b := bytes.Trim(data, "\x00")
vprint.VV("PARSE RAW: (%v) (%d)", string(b), len(b))
if strings.Contains(string(b), "EXTRACT") {
queryStr := string(bytes.Trim(data, "\x00"))
vprint.VV("PARSE RAW: (%v) (%d)", queryStr, len(queryStr))
if strings.HasPrefix(queryStr, "[") {
p.sql = queryStr
p.Name = "PQL"
p.pgspecial = pgPassOn
p.Add(message.ParseOK)
return
}
if strings.Contains(queryStr, "EXTRACT") {
// had to add this hack because the vitis parser doesn't handle...
/*
SELECT pid as id,
@ -315,14 +322,14 @@ func (p *Portal) Parse(data []byte) {
*/
p.pgspecial = pgQueryTime
p.Name = "SELECT"
p.sql = string(b)
p.sql = queryStr
p.Add(message.ParseOK)
return
}
if len(b) > 2 {
if len(queryStr) > 2 {
query, err := p.mapper.MapSQL(string(b))
query, err := p.mapper.MapSQL(queryStr)
if err != nil {
vprint.VV("Parse Err: '%v'", err)
} else {
@ -393,14 +400,14 @@ func (p *Portal) Parse(data []byte) {
}
}
}
p.sql = string(b)
p.sql = queryStr
case sql.SQLTypeBegin:
// Ignore BEGIN
p.pgspecial = pgBegin
case sql.SQLTypeShow:
p.Name = "SHOW"
p.pgspecial = pgPassOn
p.sql = string(b)
p.sql = queryStr
}
}
}