super-graph/serv/utils.go

144 lines
2.3 KiB
Go
Raw Normal View History

2019-04-19 07:55:03 +02:00
package serv
2019-07-29 07:13:33 +02:00
import (
"bytes"
"crypto/sha1"
"encoding/hex"
2019-09-05 06:09:56 +02:00
"io"
"sort"
"strings"
2019-07-29 07:13:33 +02:00
"github.com/cespare/xxhash/v2"
2019-09-05 06:09:56 +02:00
"github.com/dosco/super-graph/jsn"
2019-07-29 07:13:33 +02:00
)
2019-04-19 07:55:03 +02:00
// nolint: errcheck
2019-05-13 01:27:26 +02:00
func mkkey(h *xxhash.Digest, k1 string, k2 string) uint64 {
h.WriteString(k1)
h.WriteString(k2)
v := h.Sum64()
h.Reset()
2019-04-19 07:55:03 +02:00
2019-05-13 01:27:26 +02:00
return v
2019-04-19 07:55:03 +02:00
}
2019-07-29 07:13:33 +02:00
// nolint: errcheck
func gqlHash(b string, vars []byte, role string) string {
2019-09-05 06:09:56 +02:00
b = strings.TrimSpace(b)
2019-07-29 07:13:33 +02:00
h := sha1.New()
query := "query"
2019-07-30 07:38:05 +02:00
2019-07-29 07:13:33 +02:00
s, e := 0, 0
2019-07-30 07:38:05 +02:00
space := []byte{' '}
starting := true
2019-07-30 07:38:05 +02:00
var b0, b1 byte
2019-07-29 07:13:33 +02:00
if len(b) == 0 {
return ""
}
2019-07-29 07:13:33 +02:00
for {
if starting && b[e] == 'q' {
n := 0
se := e
for e < len(b) && n < len(query) && b[e] == query[n] {
n++
e++
}
if n != len(query) {
io.WriteString(h, strings.ToLower(b[se:e]))
}
}
if e >= len(b) {
break
}
2019-07-30 07:38:05 +02:00
if ws(b[e]) {
for e < len(b) && ws(b[e]) {
2019-07-29 07:13:33 +02:00
e++
}
2019-07-30 07:38:05 +02:00
if e < len(b) {
b1 = b[e]
}
if al(b0) && al(b1) {
h.Write(space)
}
2019-07-29 07:13:33 +02:00
} else {
starting = false
2019-07-30 07:38:05 +02:00
s = e
for e < len(b) && !ws(b[e]) {
2019-07-30 07:38:05 +02:00
e++
}
if e != 0 {
b0 = b[(e - 1)]
}
2019-09-05 06:09:56 +02:00
io.WriteString(h, strings.ToLower(b[s:e]))
2019-07-30 07:38:05 +02:00
}
if e >= len(b) {
break
2019-07-29 07:13:33 +02:00
}
}
if len(role) != 0 {
io.WriteString(h, role)
}
if len(vars) == 0 {
2019-09-05 06:09:56 +02:00
return hex.EncodeToString(h.Sum(nil))
}
fields := jsn.Keys([]byte(vars))
sort.Slice(fields, func(i, j int) bool {
return bytes.Compare(fields[i], fields[j]) == -1
})
for i := range fields {
h.Write(fields[i])
}
2019-07-29 07:13:33 +02:00
return hex.EncodeToString(h.Sum(nil))
}
func ws(b byte) bool {
2019-07-30 07:38:05 +02:00
return b == ' ' || b == '\n' || b == '\t' || b == ','
}
func al(b byte) bool {
return (b >= 'a' && b <= 'z') || (b >= 'A' && b <= 'Z') || (b >= '0' && b <= '9')
2019-07-29 07:13:33 +02:00
}
func gqlName(b string) string {
state, s := 0, 0
for i := 0; i < len(b); i++ {
switch {
case state == 2 && b[i] == '{':
return b[s:i]
case state == 2 && b[i] == ' ':
return b[s:i]
case state == 1 && b[i] == '{':
return ""
case state == 1 && b[i] != ' ':
s = i
state = 2
case state == 1 && b[i] == ' ':
continue
case i != 0 && b[i] == ' ' && (b[i-1] == 'n' || b[i-1] == 'y'):
state = 1
}
}
return ""
}
func findStmt(role string, stmts []stmt) *stmt {
for i := range stmts {
if stmts[i].role.Name != role {
continue
}
return &stmts[i]
}
return nil
}