4cf7956ff5
This PR changes the generated SQL so that it's also compatible with CockroachDB.
Notable changes:
* use `SELECT to_jsonb("__sr_0".*)` instead of `SELECT to_jsonb("__sr_0")`
* don't use `json_populate_record`, use the `CAST` and `->>` instead. For example:
instead of: `SELECT "t"."full_name", "t"."email" FROM "_sg_input" i, json_populate_record(NULL::users, i.j) t`
do: `CAST( i.j ->>'full_name' AS character varying), CAST( i.j ->>'email' AS character varying) FROM "_sg_input" i`
This PR also adds some integration tests against an actual database instance. If you have the cockroachdb binary installed on your PATH,
the test suite will startup a temporary cockroachdb instance on a random port to test against. It is stopped and the tmp data files are deleted once the test ends. It will also run the integration tests against database
pointed at by your `SG_POSTGRESQL_TEST_URL` environment variable if it’s set.
Also includes some small formatting changes introduced by `gofmt -w .`
73 lines
1.4 KiB
Go
73 lines
1.4 KiB
Go
package core
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/base64"
|
|
|
|
"github.com/dosco/super-graph/core/internal/crypto"
|
|
"github.com/dosco/super-graph/core/internal/qcode"
|
|
"github.com/dosco/super-graph/jsn"
|
|
)
|
|
|
|
func (sg *SuperGraph) encryptCursor(qc *qcode.QCode, data []byte) ([]byte, error) {
|
|
var keys [][]byte
|
|
|
|
for _, s := range qc.Selects {
|
|
if s.Paging.Type != qcode.PtOffset {
|
|
var buf bytes.Buffer
|
|
|
|
buf.WriteString(s.FieldName)
|
|
buf.WriteString("_cursor")
|
|
keys = append(keys, buf.Bytes())
|
|
}
|
|
}
|
|
|
|
if len(keys) == 0 {
|
|
return data, nil
|
|
}
|
|
|
|
from := jsn.Get(data, keys)
|
|
to := make([]jsn.Field, len(from))
|
|
|
|
for i, f := range from {
|
|
to[i].Key = f.Key
|
|
|
|
if f.Value[0] != '"' || f.Value[len(f.Value)-1] != '"' {
|
|
continue
|
|
}
|
|
|
|
var buf bytes.Buffer
|
|
|
|
if len(f.Value) > 2 {
|
|
v, err := crypto.Encrypt(f.Value[1:len(f.Value)-1], &sg.encKey)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
buf.WriteByte('"')
|
|
buf.WriteString(base64.StdEncoding.EncodeToString(v))
|
|
buf.WriteByte('"')
|
|
} else {
|
|
buf.WriteString(`null`)
|
|
}
|
|
|
|
to[i].Value = buf.Bytes()
|
|
}
|
|
|
|
var buf bytes.Buffer
|
|
|
|
if err := jsn.Replace(&buf, data, from, to); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return buf.Bytes(), nil
|
|
}
|
|
|
|
func (sg *SuperGraph) decrypt(data string) ([]byte, error) {
|
|
v, err := base64.StdEncoding.DecodeString(data)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return crypto.Decrypt(v, &sg.encKey)
|
|
}
|