Compare commits
8 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 0bb7f2cd85 | |||
| b61bf52df9 | |||
| f1dd467c95 | |||
| 3136d71032 | |||
| 8680e139e7 | |||
| 8789b85d92 | |||
| fefcba5901 | |||
| 5ad4ab2e23 |
@ -178,6 +178,6 @@ var results = store.query(ctx, "myCollection", {
|
|||||||
limit: 10,
|
limit: 10,
|
||||||
offset: 5,
|
offset: 5,
|
||||||
orderBy: "foo",
|
orderBy: "foo",
|
||||||
orderDirection: store.ASC,
|
orderDirection: store.DIRECTION_ASC,
|
||||||
});
|
});
|
||||||
```
|
```
|
||||||
|
|||||||
@ -4,7 +4,7 @@
|
|||||||
<meta charset="utf-8" />
|
<meta charset="utf-8" />
|
||||||
<title>Client SDK Test suite</title>
|
<title>Client SDK Test suite</title>
|
||||||
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
|
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
|
||||||
<link rel="stylesheet" href="vendor/mocha.css" />
|
<link rel="stylesheet" href="/vendor/mocha.css" />
|
||||||
<style>
|
<style>
|
||||||
body {
|
body {
|
||||||
background-color: white;
|
background-color: white;
|
||||||
@ -13,15 +13,18 @@
|
|||||||
</head>
|
</head>
|
||||||
<body>
|
<body>
|
||||||
<div id="mocha"></div>
|
<div id="mocha"></div>
|
||||||
<script src="vendor/chai.js"></script>
|
<script src="/vendor/chai.js"></script>
|
||||||
<script src="vendor/mocha.js"></script>
|
<script src="/vendor/mocha.js"></script>
|
||||||
<script class="mocha-init">
|
<script class="mocha-init">
|
||||||
mocha.setup('bdd');
|
mocha.setup('bdd');
|
||||||
mocha.checkLeaks();
|
mocha.checkLeaks();
|
||||||
</script>
|
</script>
|
||||||
<script src="/edge/sdk/client.js"></script>
|
<script src="/edge/sdk/client.js"></script>
|
||||||
<script src="test/client-sdk.js"></script>
|
<script src="/test/client-sdk.js"></script>
|
||||||
<script src="test/auth-module.js"></script>
|
<script src="/test/auth-module.js"></script>
|
||||||
|
<script src="/test/net-module.js"></script>
|
||||||
|
<script src="/test/rpc-module.js"></script>
|
||||||
|
<script src="/test/file-module.js"></script>
|
||||||
<script class="mocha-exec">
|
<script class="mocha-exec">
|
||||||
mocha.run();
|
mocha.run();
|
||||||
</script>
|
</script>
|
||||||
|
|||||||
@ -21,128 +21,5 @@ describe('Edge', function() {
|
|||||||
chai.assert.isNull(Edge._conn);
|
chai.assert.isNull(Edge._conn);
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|
||||||
describe('#send()', function() {
|
|
||||||
this.timeout(5000);
|
|
||||||
|
|
||||||
before(() => {
|
|
||||||
return Edge.connect();
|
|
||||||
});
|
|
||||||
|
|
||||||
after(() => {
|
|
||||||
Edge.disconnect();
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should send a message to the server and echo back', function(done) {
|
|
||||||
const now = new Date();
|
|
||||||
const handler = evt => {
|
|
||||||
chai.assert.equal(evt.detail.now, now.toJSON());
|
|
||||||
Edge.removeEventListener('message', handler);
|
|
||||||
done();
|
|
||||||
}
|
|
||||||
|
|
||||||
// Server should echo back message
|
|
||||||
Edge.addEventListener('message', handler);
|
|
||||||
|
|
||||||
// Send message to server
|
|
||||||
Edge.send({ now });
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
});
|
});
|
||||||
|
|
||||||
describe('Remote Procedure Call', function() {
|
|
||||||
|
|
||||||
before(() => {
|
|
||||||
return Edge.connect();
|
|
||||||
});
|
|
||||||
|
|
||||||
after(() => {
|
|
||||||
Edge.disconnect();
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should call the remote echo() method and resolve the returned value', function() {
|
|
||||||
const foo = "bar";
|
|
||||||
|
|
||||||
return Edge.rpc('echo', { foo })
|
|
||||||
.then(result => {
|
|
||||||
chai.assert.equal(result.foo, foo);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should call the remote throwErrorFromClient() method and reject with an error', function() {
|
|
||||||
return Edge.rpc('throwErrorFromClient')
|
|
||||||
.catch(err => {
|
|
||||||
// Assert that it's an "internal" error
|
|
||||||
// See https://www.jsonrpc.org/specification#error_object
|
|
||||||
chai.assert.equal(err.code, -32603);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should call an unregistered method and reject with an error', function() {
|
|
||||||
return Edge.rpc('unregisteredMethod')
|
|
||||||
.catch(err => {
|
|
||||||
// Assert that it's an "method not found" error
|
|
||||||
// See https://www.jsonrpc.org/specification#error_object
|
|
||||||
chai.assert.equal(err.code, -32601);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
|
|
||||||
it('should call the add() method repetitively and keep count of the sent values', function() {
|
|
||||||
this.timeout(10000);
|
|
||||||
|
|
||||||
const values = [];
|
|
||||||
for(let i = 0; i <= 1000; i++) {
|
|
||||||
values.push((Math.random() * 1000 | 0));
|
|
||||||
}
|
|
||||||
return Edge.rpc('reset')
|
|
||||||
.then(() => {
|
|
||||||
return Promise.all(values.map(v => Edge.rpc("add", {value: v})));
|
|
||||||
})
|
|
||||||
.then(() => Edge.rpc('total'))
|
|
||||||
.then(remoteTotal => {
|
|
||||||
const localTotal = values.reduce((t, v) => t+v);
|
|
||||||
console.log("Remote total:", remoteTotal, "Local total:", localTotal);
|
|
||||||
chai.assert.equal(remoteTotal, localTotal)
|
|
||||||
})
|
|
||||||
});
|
|
||||||
|
|
||||||
});
|
|
||||||
|
|
||||||
describe('File Module', function() {
|
|
||||||
|
|
||||||
before(() => {
|
|
||||||
return Edge.connect();
|
|
||||||
});
|
|
||||||
|
|
||||||
after(() => {
|
|
||||||
Edge.disconnect();
|
|
||||||
});
|
|
||||||
|
|
||||||
it('should upload then download a blob', function() {
|
|
||||||
const content = JSON.stringify({"date": new Date()});
|
|
||||||
const blob = new Blob([content], {type: "application/json"});
|
|
||||||
|
|
||||||
return Edge.upload(blob)
|
|
||||||
.then(upload => upload.result())
|
|
||||||
.then(result => {
|
|
||||||
|
|
||||||
chai.assert.isNotEmpty(result.blobId);
|
|
||||||
chai.assert.isNotEmpty(result.bucket);
|
|
||||||
|
|
||||||
const blobUrl = Edge.blobUrl(result.bucket, result.blobId);
|
|
||||||
chai.assert.isNotEmpty(blobUrl);
|
|
||||||
|
|
||||||
return fetch(blobUrl)
|
|
||||||
.then(res => res.text())
|
|
||||||
.then(blobContent => {
|
|
||||||
chai.assert.equal(content, blobContent);
|
|
||||||
});
|
|
||||||
})
|
|
||||||
.catch(err => {
|
|
||||||
chai.assert.fail(err);
|
|
||||||
})
|
|
||||||
});
|
|
||||||
|
|
||||||
});
|
|
||||||
|
|||||||
36
misc/client-sdk-testsuite/src/public/test/file-module.js
Normal file
36
misc/client-sdk-testsuite/src/public/test/file-module.js
Normal file
@ -0,0 +1,36 @@
|
|||||||
|
describe('File Module', function () {
|
||||||
|
|
||||||
|
before(() => {
|
||||||
|
return Edge.connect();
|
||||||
|
});
|
||||||
|
|
||||||
|
after(() => {
|
||||||
|
Edge.disconnect();
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should upload then download a blob', function () {
|
||||||
|
const content = JSON.stringify({ "date": new Date() });
|
||||||
|
const blob = new Blob([content], { type: "application/json" });
|
||||||
|
|
||||||
|
return Edge.upload(blob)
|
||||||
|
.then(upload => upload.result())
|
||||||
|
.then(result => {
|
||||||
|
|
||||||
|
chai.assert.isNotEmpty(result.blobId);
|
||||||
|
chai.assert.isNotEmpty(result.bucket);
|
||||||
|
|
||||||
|
const blobUrl = Edge.blobUrl(result.bucket, result.blobId);
|
||||||
|
chai.assert.isNotEmpty(blobUrl);
|
||||||
|
|
||||||
|
return fetch(blobUrl)
|
||||||
|
.then(res => res.text())
|
||||||
|
.then(blobContent => {
|
||||||
|
chai.assert.equal(content, blobContent);
|
||||||
|
});
|
||||||
|
})
|
||||||
|
.catch(err => {
|
||||||
|
chai.assert.fail(err);
|
||||||
|
})
|
||||||
|
});
|
||||||
|
|
||||||
|
});
|
||||||
49
misc/client-sdk-testsuite/src/public/test/net-module.js
Normal file
49
misc/client-sdk-testsuite/src/public/test/net-module.js
Normal file
@ -0,0 +1,49 @@
|
|||||||
|
describe('Net Module', function () {
|
||||||
|
this.timeout(5000);
|
||||||
|
|
||||||
|
before(() => {
|
||||||
|
return Edge.connect();
|
||||||
|
});
|
||||||
|
|
||||||
|
after(() => {
|
||||||
|
Edge.disconnect();
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should broadcast a message from server', function (done) {
|
||||||
|
const message = { test: 'broadcast', now: Date.now() };
|
||||||
|
|
||||||
|
const handler = (evt) => {
|
||||||
|
const receivedMessage = evt.detail;
|
||||||
|
if (receivedMessage.test !== 'broadcast') return;
|
||||||
|
|
||||||
|
chai.assert.deepEqual(message, evt.detail);
|
||||||
|
|
||||||
|
Edge.removeEventListener('message', handler);
|
||||||
|
done();
|
||||||
|
};
|
||||||
|
|
||||||
|
Edge.addEventListener("message", handler);
|
||||||
|
Edge.send(message);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should send a message to the server and echo back', function(done) {
|
||||||
|
const now = new Date();
|
||||||
|
|
||||||
|
const handler = evt => {
|
||||||
|
const receivedMessage = evt.detail;
|
||||||
|
if (receivedMessage.test !== 'echo') return;
|
||||||
|
|
||||||
|
chai.assert.equal(receivedMessage.now, now.toJSON());
|
||||||
|
|
||||||
|
Edge.removeEventListener('message', handler);
|
||||||
|
done();
|
||||||
|
}
|
||||||
|
|
||||||
|
// Server should echo back message
|
||||||
|
Edge.addEventListener('message', handler);
|
||||||
|
|
||||||
|
// Send message to server
|
||||||
|
Edge.send({ test: 'echo', now });
|
||||||
|
});
|
||||||
|
|
||||||
|
});
|
||||||
59
misc/client-sdk-testsuite/src/public/test/rpc-module.js
Normal file
59
misc/client-sdk-testsuite/src/public/test/rpc-module.js
Normal file
@ -0,0 +1,59 @@
|
|||||||
|
describe('Remote Procedure Call', function () {
|
||||||
|
|
||||||
|
before(() => {
|
||||||
|
return Edge.connect();
|
||||||
|
});
|
||||||
|
|
||||||
|
after(() => {
|
||||||
|
Edge.disconnect();
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should call the remote echo() method and resolve the returned value', function () {
|
||||||
|
const foo = "bar";
|
||||||
|
|
||||||
|
return Edge.rpc('echo', { foo })
|
||||||
|
.then(result => {
|
||||||
|
console.log(result);
|
||||||
|
chai.assert.equal(result.foo, foo);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should call the remote throwErrorFromClient() method and reject with an error', function () {
|
||||||
|
return Edge.rpc('throwErrorFromClient')
|
||||||
|
.catch(err => {
|
||||||
|
// Assert that it's an "internal" error
|
||||||
|
// See https://www.jsonrpc.org/specification#error_object
|
||||||
|
chai.assert.equal(err.code, -32603);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should call an unregistered method and reject with an error', function () {
|
||||||
|
return Edge.rpc('unregisteredMethod')
|
||||||
|
.catch(err => {
|
||||||
|
// Assert that it's an "method not found" error
|
||||||
|
// See https://www.jsonrpc.org/specification#error_object
|
||||||
|
chai.assert.equal(err.code, -32601);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
|
||||||
|
it('should call the add() method repetitively and keep count of the sent values', function () {
|
||||||
|
this.timeout(10000);
|
||||||
|
|
||||||
|
const values = [];
|
||||||
|
for (let i = 0; i <= 1000; i++) {
|
||||||
|
values.push((Math.random() * 1000 | 0));
|
||||||
|
}
|
||||||
|
return Edge.rpc('reset')
|
||||||
|
.then(() => {
|
||||||
|
return Promise.all(values.map(v => Edge.rpc("add", { value: v })));
|
||||||
|
})
|
||||||
|
.then(() => Edge.rpc('total'))
|
||||||
|
.then(remoteTotal => {
|
||||||
|
const localTotal = values.reduce((t, v) => t + v);
|
||||||
|
console.log("Remote total:", remoteTotal, "Local total:", localTotal);
|
||||||
|
chai.assert.equal(remoteTotal, localTotal)
|
||||||
|
})
|
||||||
|
});
|
||||||
|
|
||||||
|
});
|
||||||
@ -14,9 +14,16 @@ function onInit() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Called for each client message
|
// Called for each client message
|
||||||
function onClientMessage(ctx, data) {
|
function onClientMessage(ctx, message) {
|
||||||
console.log("onClientMessage", data.now);
|
console.log("onClientMessage", message);
|
||||||
net.send(ctx, { now: data.now });
|
|
||||||
|
switch (message.test) {
|
||||||
|
case "broadcast":
|
||||||
|
net.broadcast(message);
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
net.send(ctx, message);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Called for each blob upload request
|
// Called for each blob upload request
|
||||||
|
|||||||
@ -1,6 +1,6 @@
|
|||||||
**/*.go
|
**/*.go
|
||||||
pkg/app/sdk/client/src/**/*.js
|
pkg/sdk/client/src/**/*.js
|
||||||
pkg/app/sdk/client/src/**/*.ts
|
pkg/sdk/client/src/**/*.ts
|
||||||
misc/client-sdk-testsuite/src/**/*
|
misc/client-sdk-testsuite/src/**/*
|
||||||
modd.conf
|
modd.conf
|
||||||
{
|
{
|
||||||
|
|||||||
@ -1,15 +1,20 @@
|
|||||||
package app
|
package app
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
"math/rand"
|
"math/rand"
|
||||||
"sync"
|
"sync"
|
||||||
|
|
||||||
"github.com/dop251/goja"
|
"github.com/dop251/goja"
|
||||||
"github.com/dop251/goja_nodejs/eventloop"
|
"github.com/dop251/goja_nodejs/eventloop"
|
||||||
"github.com/pkg/errors"
|
"github.com/pkg/errors"
|
||||||
|
"gitlab.com/wpetit/goweb/logger"
|
||||||
)
|
)
|
||||||
|
|
||||||
var ErrFuncDoesNotExist = errors.New("function does not exist")
|
var (
|
||||||
|
ErrFuncDoesNotExist = errors.New("function does not exist")
|
||||||
|
ErUnknownError = errors.New("unknown error")
|
||||||
|
)
|
||||||
|
|
||||||
type Server struct {
|
type Server struct {
|
||||||
runtime *goja.Runtime
|
runtime *goja.Runtime
|
||||||
@ -26,16 +31,18 @@ func (s *Server) Load(name string, src string) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Server) ExecFuncByName(funcName string, args ...interface{}) (goja.Value, error) {
|
func (s *Server) ExecFuncByName(ctx context.Context, funcName string, args ...interface{}) (goja.Value, error) {
|
||||||
|
ctx = logger.With(ctx, logger.F("function", funcName), logger.F("args", args))
|
||||||
|
|
||||||
callable, ok := goja.AssertFunction(s.runtime.Get(funcName))
|
callable, ok := goja.AssertFunction(s.runtime.Get(funcName))
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, errors.WithStack(ErrFuncDoesNotExist)
|
return nil, errors.WithStack(ErrFuncDoesNotExist)
|
||||||
}
|
}
|
||||||
|
|
||||||
return s.Exec(callable, args...)
|
return s.Exec(ctx, callable, args...)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Server) Exec(callable goja.Callable, args ...interface{}) (goja.Value, error) {
|
func (s *Server) Exec(ctx context.Context, callable goja.Callable, args ...interface{}) (goja.Value, error) {
|
||||||
var (
|
var (
|
||||||
wg sync.WaitGroup
|
wg sync.WaitGroup
|
||||||
value goja.Value
|
value goja.Value
|
||||||
@ -45,6 +52,25 @@ func (s *Server) Exec(callable goja.Callable, args ...interface{}) (goja.Value,
|
|||||||
wg.Add(1)
|
wg.Add(1)
|
||||||
|
|
||||||
s.loop.RunOnLoop(func(vm *goja.Runtime) {
|
s.loop.RunOnLoop(func(vm *goja.Runtime) {
|
||||||
|
logger.Debug(ctx, "executing callable")
|
||||||
|
|
||||||
|
defer wg.Done()
|
||||||
|
|
||||||
|
defer func() {
|
||||||
|
if recovered := recover(); recovered != nil {
|
||||||
|
revoveredErr, ok := recovered.(error)
|
||||||
|
if ok {
|
||||||
|
logger.Error(ctx, "recovered runtime error", logger.E(errors.WithStack(revoveredErr)))
|
||||||
|
|
||||||
|
err = errors.WithStack(ErUnknownError)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
panic(recovered)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
jsArgs := make([]goja.Value, 0, len(args))
|
jsArgs := make([]goja.Value, 0, len(args))
|
||||||
for _, a := range args {
|
for _, a := range args {
|
||||||
jsArgs = append(jsArgs, vm.ToValue(a))
|
jsArgs = append(jsArgs, vm.ToValue(a))
|
||||||
@ -54,8 +80,6 @@ func (s *Server) Exec(callable goja.Callable, args ...interface{}) (goja.Value,
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
err = errors.WithStack(err)
|
err = errors.WithStack(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
wg.Done()
|
|
||||||
})
|
})
|
||||||
|
|
||||||
wg.Wait()
|
wg.Wait()
|
||||||
|
|||||||
@ -99,3 +99,5 @@ func (f *File) Readdir(count int) ([]os.FileInfo, error) {
|
|||||||
func (f *File) Stat() (os.FileInfo, error) {
|
func (f *File) Stat() (os.FileInfo, error) {
|
||||||
return f.fi, nil
|
return f.fi, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var _ http.FileSystem = &FileSystem{}
|
||||||
|
|||||||
@ -59,7 +59,7 @@ func (h *Handler) Load(bdle bundle.Bundle) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fs := bundle.NewFileSystem("public", bdle)
|
fs := bundle.NewFileSystem("public", bdle)
|
||||||
public := http.FileServer(fs)
|
public := HTML5Fileserver(fs)
|
||||||
sockjs := sockjs.NewHandler(sockJSPathPrefix, h.sockjsOpts, h.handleSockJSSession)
|
sockjs := sockjs.NewHandler(sockJSPathPrefix, h.sockjsOpts, h.handleSockJSSession)
|
||||||
|
|
||||||
if h.server != nil {
|
if h.server != nil {
|
||||||
|
|||||||
54
pkg/http/html5_fileserver.go
Normal file
54
pkg/http/html5_fileserver.go
Normal file
@ -0,0 +1,54 @@
|
|||||||
|
package http
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"path"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/pkg/errors"
|
||||||
|
"gitlab.com/wpetit/goweb/logger"
|
||||||
|
)
|
||||||
|
|
||||||
|
func HTML5Fileserver(fs http.FileSystem) http.Handler {
|
||||||
|
handler := http.FileServer(fs)
|
||||||
|
|
||||||
|
fn := func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
urlPath := r.URL.Path
|
||||||
|
if !strings.HasPrefix(urlPath, "/") {
|
||||||
|
urlPath = "/" + urlPath
|
||||||
|
r.URL.Path = urlPath
|
||||||
|
}
|
||||||
|
urlPath = path.Clean(urlPath)
|
||||||
|
|
||||||
|
file, err := fs.Open(urlPath)
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, os.ErrNotExist) {
|
||||||
|
r.URL.Path = "/"
|
||||||
|
|
||||||
|
handler.ServeHTTP(w, r)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
logger.Error(r.Context(), "could not open bundle file", logger.E(err))
|
||||||
|
|
||||||
|
http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
defer func() {
|
||||||
|
if err := file.Close(); err != nil {
|
||||||
|
logger.Error(r.Context(), "could not close file", logger.E(err))
|
||||||
|
|
||||||
|
http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
handler.ServeHTTP(w, r)
|
||||||
|
}
|
||||||
|
|
||||||
|
return http.HandlerFunc(fn)
|
||||||
|
}
|
||||||
@ -65,7 +65,7 @@ func TestAuthModule(t *testing.T) {
|
|||||||
|
|
||||||
ctx := context.WithValue(context.Background(), edgeHTTP.ContextKeyOriginRequest, req)
|
ctx := context.WithValue(context.Background(), edgeHTTP.ContextKeyOriginRequest, req)
|
||||||
|
|
||||||
if _, err := server.ExecFuncByName("testAuth", ctx); err != nil {
|
if _, err := server.ExecFuncByName(ctx, "testAuth", ctx); err != nil {
|
||||||
t.Fatalf("%+v", errors.WithStack(err))
|
t.Fatalf("%+v", errors.WithStack(err))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -104,7 +104,7 @@ func TestAuthAnonymousModule(t *testing.T) {
|
|||||||
|
|
||||||
ctx := context.WithValue(context.Background(), edgeHTTP.ContextKeyOriginRequest, req)
|
ctx := context.WithValue(context.Background(), edgeHTTP.ContextKeyOriginRequest, req)
|
||||||
|
|
||||||
if _, err := server.ExecFuncByName("testAuth", ctx); err != nil {
|
if _, err := server.ExecFuncByName(ctx, "testAuth", ctx); err != nil {
|
||||||
t.Fatalf("%+v", errors.WithStack(err))
|
t.Fatalf("%+v", errors.WithStack(err))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@ -88,7 +88,7 @@ func (m *BlobModule) handleUploadRequest(req *MessageUploadRequest) (*MessageUpl
|
|||||||
"contentType": req.FileHeader.Header.Get("Content-Type"),
|
"contentType": req.FileHeader.Header.Get("Content-Type"),
|
||||||
}
|
}
|
||||||
|
|
||||||
rawResult, err := m.server.ExecFuncByName("onBlobUpload", ctx, blobID, blobInfo, req.Metadata)
|
rawResult, err := m.server.ExecFuncByName(ctx, "onBlobUpload", ctx, blobID, blobInfo, req.Metadata)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if errors.Is(err, app.ErrFuncDoesNotExist) {
|
if errors.Is(err, app.ErrFuncDoesNotExist) {
|
||||||
res.Allow = false
|
res.Allow = false
|
||||||
@ -193,7 +193,7 @@ func (m *BlobModule) saveBlob(ctx context.Context, bucketName string, blobID sto
|
|||||||
func (m *BlobModule) handleDownloadRequest(req *MessageDownloadRequest) (*MessageDownloadResponse, error) {
|
func (m *BlobModule) handleDownloadRequest(req *MessageDownloadRequest) (*MessageDownloadResponse, error) {
|
||||||
res := NewMessageDownloadResponse(req.RequestID)
|
res := NewMessageDownloadResponse(req.RequestID)
|
||||||
|
|
||||||
rawResult, err := m.server.ExecFuncByName("onBlobDownload", req.Context, req.Bucket, req.BlobID)
|
rawResult, err := m.server.ExecFuncByName(req.Context, "onBlobDownload", req.Context, req.Bucket, req.BlobID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if errors.Is(err, app.ErrFuncDoesNotExist) {
|
if errors.Is(err, app.ErrFuncDoesNotExist) {
|
||||||
res.Allow = false
|
res.Allow = false
|
||||||
|
|||||||
@ -1,6 +1,7 @@
|
|||||||
package cast
|
package cast
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"os"
|
"os"
|
||||||
"testing"
|
"testing"
|
||||||
@ -79,7 +80,7 @@ func TestCastModuleRefreshDevices(t *testing.T) {
|
|||||||
|
|
||||||
defer server.Stop()
|
defer server.Stop()
|
||||||
|
|
||||||
result, err := server.ExecFuncByName("refreshDevices")
|
result, err := server.ExecFuncByName(context.Background(), "refreshDevices")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Error(errors.WithStack(err))
|
t.Error(errors.WithStack(err))
|
||||||
}
|
}
|
||||||
|
|||||||
@ -21,7 +21,7 @@ func (m *LifecycleModule) Export(export *goja.Object) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (m *LifecycleModule) OnInit() error {
|
func (m *LifecycleModule) OnInit() error {
|
||||||
if _, err := m.server.ExecFuncByName("onInit"); err != nil {
|
if _, err := m.server.ExecFuncByName(context.Background(), "onInit"); err != nil {
|
||||||
if errors.Is(err, app.ErrFuncDoesNotExist) {
|
if errors.Is(err, app.ErrFuncDoesNotExist) {
|
||||||
logger.Warn(context.Background(), "could not find onInit() function", logger.E(errors.WithStack(err)))
|
logger.Warn(context.Background(), "could not find onInit() function", logger.E(errors.WithStack(err)))
|
||||||
|
|
||||||
|
|||||||
@ -38,9 +38,10 @@ func (m *Module) broadcast(call goja.FunctionCall, rt *goja.Runtime) goja.Value
|
|||||||
}
|
}
|
||||||
|
|
||||||
data := call.Argument(0).Export()
|
data := call.Argument(0).Export()
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
msg := module.NewServerMessage(nil, data)
|
msg := module.NewServerMessage(ctx, data)
|
||||||
if err := m.bus.Publish(context.Background(), msg); err != nil {
|
if err := m.bus.Publish(ctx, msg); err != nil {
|
||||||
panic(rt.ToValue(errors.WithStack(err)))
|
panic(rt.ToValue(errors.WithStack(err)))
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -129,7 +130,7 @@ func (m *Module) handleClientMessages() {
|
|||||||
logger.F("message", clientMessage),
|
logger.F("message", clientMessage),
|
||||||
)
|
)
|
||||||
|
|
||||||
if _, err := m.server.ExecFuncByName("onClientMessage", clientMessage.Context, clientMessage.Data); err != nil {
|
if _, err := m.server.ExecFuncByName(clientMessage.Context, "onClientMessage", clientMessage.Context, clientMessage.Data); err != nil {
|
||||||
if errors.Is(err, app.ErrFuncDoesNotExist) {
|
if errors.Is(err, app.ErrFuncDoesNotExist) {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|||||||
@ -161,7 +161,7 @@ func (m *RPCModule) handleMessages() {
|
|||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
result, err := m.server.Exec(callable, clientMessage.Context, req.Params)
|
result, err := m.server.Exec(clientMessage.Context, callable, clientMessage.Context, req.Params)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Error(
|
logger.Error(
|
||||||
ctx, "rpc call error",
|
ctx, "rpc call error",
|
||||||
|
|||||||
@ -102,7 +102,7 @@ func (m *Module) query(call goja.FunctionCall, rt *goja.Runtime) goja.Value {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if queryOptions.Offset != nil {
|
if queryOptions.Offset != nil {
|
||||||
queryOptionsFuncs = append(queryOptionsFuncs, storage.WithOffset(*queryOptions.Limit))
|
queryOptionsFuncs = append(queryOptionsFuncs, storage.WithOffset(*queryOptions.Offset))
|
||||||
}
|
}
|
||||||
|
|
||||||
if queryOptions.OrderDirection != nil {
|
if queryOptions.OrderDirection != nil {
|
||||||
|
|||||||
@ -1,6 +1,7 @@
|
|||||||
package store
|
package store
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
@ -34,7 +35,7 @@ func TestStoreModule(t *testing.T) {
|
|||||||
t.Fatalf("%+v", errors.WithStack(err))
|
t.Fatalf("%+v", errors.WithStack(err))
|
||||||
}
|
}
|
||||||
|
|
||||||
if _, err := server.ExecFuncByName("testStore"); err != nil {
|
if _, err := server.ExecFuncByName(context.Background(), "testStore"); err != nil {
|
||||||
t.Fatalf("%+v", errors.WithStack(err))
|
t.Fatalf("%+v", errors.WithStack(err))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@ -93,6 +93,7 @@ export class Client extends EventTarget {
|
|||||||
const { jsonrpc, id, error, result } = evt.detail;
|
const { jsonrpc, id, error, result } = evt.detail;
|
||||||
|
|
||||||
if (jsonrpc !== '2.0' || id === undefined) return;
|
if (jsonrpc !== '2.0' || id === undefined) return;
|
||||||
|
if (!evt.detail.hasOwnProperty("error") && !evt.detail.hasOwnProperty("result")) return;
|
||||||
|
|
||||||
// Prevent additional handlers to catch this event
|
// Prevent additional handlers to catch this event
|
||||||
evt.stopImmediatePropagation();
|
evt.stopImmediatePropagation();
|
||||||
|
|||||||
@ -7,35 +7,35 @@ const (
|
|||||||
OrderDirectionDesc OrderDirection = "DESC"
|
OrderDirectionDesc OrderDirection = "DESC"
|
||||||
)
|
)
|
||||||
|
|
||||||
type QueryOption struct {
|
type QueryOptions struct {
|
||||||
Limit *int
|
Limit *int
|
||||||
Offset *int
|
Offset *int
|
||||||
OrderBy *string
|
OrderBy *string
|
||||||
OrderDirection *OrderDirection
|
OrderDirection *OrderDirection
|
||||||
}
|
}
|
||||||
|
|
||||||
type QueryOptionFunc func(o *QueryOption)
|
type QueryOptionFunc func(o *QueryOptions)
|
||||||
|
|
||||||
func WithLimit(limit int) QueryOptionFunc {
|
func WithLimit(limit int) QueryOptionFunc {
|
||||||
return func(o *QueryOption) {
|
return func(o *QueryOptions) {
|
||||||
o.Limit = &limit
|
o.Limit = &limit
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func WithOffset(offset int) QueryOptionFunc {
|
func WithOffset(offset int) QueryOptionFunc {
|
||||||
return func(o *QueryOption) {
|
return func(o *QueryOptions) {
|
||||||
o.Offset = &offset
|
o.Offset = &offset
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func WithOrderBy(orderBy string) QueryOptionFunc {
|
func WithOrderBy(orderBy string) QueryOptionFunc {
|
||||||
return func(o *QueryOption) {
|
return func(o *QueryOptions) {
|
||||||
o.OrderBy = &orderBy
|
o.OrderBy = &orderBy
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func WithOrderDirection(direction OrderDirection) QueryOptionFunc {
|
func WithOrderDirection(direction OrderDirection) QueryOptionFunc {
|
||||||
return func(o *QueryOption) {
|
return func(o *QueryOptions) {
|
||||||
o.OrderDirection = &direction
|
o.OrderDirection = &direction
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -4,6 +4,7 @@ import (
|
|||||||
"context"
|
"context"
|
||||||
"database/sql"
|
"database/sql"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"math"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@ -90,6 +91,11 @@ func (s *DocumentStore) Get(ctx context.Context, collection string, id storage.D
|
|||||||
|
|
||||||
// Query implements storage.DocumentStore
|
// Query implements storage.DocumentStore
|
||||||
func (s *DocumentStore) Query(ctx context.Context, collection string, filter *filter.Filter, funcs ...storage.QueryOptionFunc) ([]storage.Document, error) {
|
func (s *DocumentStore) Query(ctx context.Context, collection string, filter *filter.Filter, funcs ...storage.QueryOptionFunc) ([]storage.Document, error) {
|
||||||
|
opts := &storage.QueryOptions{}
|
||||||
|
for _, fn := range funcs {
|
||||||
|
fn(opts)
|
||||||
|
}
|
||||||
|
|
||||||
var documents []storage.Document
|
var documents []storage.Document
|
||||||
|
|
||||||
err := s.withTx(ctx, func(tx *sql.Tx) error {
|
err := s.withTx(ctx, func(tx *sql.Tx) error {
|
||||||
@ -120,6 +126,29 @@ func (s *DocumentStore) Query(ctx context.Context, collection string, filter *fi
|
|||||||
|
|
||||||
args = append([]interface{}{collection}, args...)
|
args = append([]interface{}{collection}, args...)
|
||||||
|
|
||||||
|
if opts.OrderBy != nil {
|
||||||
|
direction := storage.OrderDirectionAsc
|
||||||
|
if opts.OrderDirection != nil {
|
||||||
|
direction = *opts.OrderDirection
|
||||||
|
}
|
||||||
|
|
||||||
|
query, args = withOrderByClause(query, args, *opts.OrderBy, direction)
|
||||||
|
}
|
||||||
|
|
||||||
|
if opts.Offset != nil || opts.Limit != nil {
|
||||||
|
offset := 0
|
||||||
|
if opts.Offset != nil {
|
||||||
|
offset = *opts.Offset
|
||||||
|
}
|
||||||
|
|
||||||
|
limit := math.MaxInt
|
||||||
|
if opts.Limit != nil {
|
||||||
|
limit = *opts.Limit
|
||||||
|
}
|
||||||
|
|
||||||
|
query, args = withLimitOffsetClause(query, args, limit, offset)
|
||||||
|
}
|
||||||
|
|
||||||
logger.Debug(
|
logger.Debug(
|
||||||
ctx, "executing query",
|
ctx, "executing query",
|
||||||
logger.F("query", query),
|
logger.F("query", query),
|
||||||
@ -331,6 +360,41 @@ func (s *DocumentStore) ensureTables(ctx context.Context, db *sql.DB) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func withOrderByClause(query string, args []any, orderBy string, orderDirection storage.OrderDirection) (string, []any) {
|
||||||
|
direction := "ASC"
|
||||||
|
if orderDirection == storage.OrderDirectionDesc {
|
||||||
|
direction = "DESC"
|
||||||
|
}
|
||||||
|
|
||||||
|
var column string
|
||||||
|
|
||||||
|
switch orderBy {
|
||||||
|
case storage.DocumentAttrID:
|
||||||
|
column = "id"
|
||||||
|
|
||||||
|
case storage.DocumentAttrCreatedAt:
|
||||||
|
column = "created_at"
|
||||||
|
|
||||||
|
case storage.DocumentAttrUpdatedAt:
|
||||||
|
column = "updated_at"
|
||||||
|
|
||||||
|
default:
|
||||||
|
column = fmt.Sprintf("json_extract(data, '$.' || $%d)", len(args)+1)
|
||||||
|
args = append(args, orderBy)
|
||||||
|
}
|
||||||
|
|
||||||
|
query += fmt.Sprintf(` ORDER BY %s %s`, column, direction)
|
||||||
|
|
||||||
|
return query, args
|
||||||
|
}
|
||||||
|
|
||||||
|
func withLimitOffsetClause(query string, args []any, limit int, offset int) (string, []any) {
|
||||||
|
query += fmt.Sprintf(` LIMIT $%d OFFSET $%d`, len(args)+1, len(args)+2)
|
||||||
|
args = append(args, limit, offset)
|
||||||
|
|
||||||
|
return query, args
|
||||||
|
}
|
||||||
|
|
||||||
func NewDocumentStore(path string) *DocumentStore {
|
func NewDocumentStore(path string) *DocumentStore {
|
||||||
return &DocumentStore{
|
return &DocumentStore{
|
||||||
db: nil,
|
db: nil,
|
||||||
|
|||||||
@ -41,7 +41,7 @@ var documentStoreOpsTestCases = []documentStoreOpsTestCase{
|
|||||||
}),
|
}),
|
||||||
)
|
)
|
||||||
|
|
||||||
results, err := store.Query(ctx, collection, filter, nil)
|
results, err := store.Query(ctx, collection, filter)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return errors.WithStack(err)
|
return errors.WithStack(err)
|
||||||
}
|
}
|
||||||
@ -82,7 +82,7 @@ var documentStoreOpsTestCases = []documentStoreOpsTestCase{
|
|||||||
}),
|
}),
|
||||||
)
|
)
|
||||||
|
|
||||||
results, err := store.Query(ctx, collection, filter, nil)
|
results, err := store.Query(ctx, collection, filter)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return errors.WithStack(err)
|
return errors.WithStack(err)
|
||||||
}
|
}
|
||||||
@ -127,7 +127,7 @@ var documentStoreOpsTestCases = []documentStoreOpsTestCase{
|
|||||||
),
|
),
|
||||||
)
|
)
|
||||||
|
|
||||||
results, err := store.Query(ctx, collection, filter, nil)
|
results, err := store.Query(ctx, collection, filter)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return errors.WithStack(err)
|
return errors.WithStack(err)
|
||||||
}
|
}
|
||||||
@ -219,7 +219,7 @@ var documentStoreOpsTestCases = []documentStoreOpsTestCase{
|
|||||||
|
|
||||||
// Verify that there is no additional created document in the collection
|
// Verify that there is no additional created document in the collection
|
||||||
|
|
||||||
results, err := store.Query(ctx, collection, nil, nil)
|
results, err := store.Query(ctx, collection, nil)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return errors.WithStack(err)
|
return errors.WithStack(err)
|
||||||
}
|
}
|
||||||
@ -228,6 +228,206 @@ var documentStoreOpsTestCases = []documentStoreOpsTestCase{
|
|||||||
return errors.Errorf("len(results): expected '%v', got '%v'", e, g)
|
return errors.Errorf("len(results): expected '%v', got '%v'", e, g)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "Query order by document field",
|
||||||
|
Run: func(ctx context.Context, store storage.DocumentStore) error {
|
||||||
|
docs := []storage.Document{
|
||||||
|
{
|
||||||
|
"sortedField": 0,
|
||||||
|
"name": "Item 1",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"sortedField": 1,
|
||||||
|
"name": "Item 2",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"sortedField": 2,
|
||||||
|
"name": "Item 3",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
collection := "ordered_query_by_document_field"
|
||||||
|
|
||||||
|
for _, doc := range docs {
|
||||||
|
if _, err := store.Upsert(ctx, collection, doc); err != nil {
|
||||||
|
return errors.WithStack(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
results, err := store.Query(
|
||||||
|
ctx, collection, nil,
|
||||||
|
storage.WithOrderBy("sortedField"),
|
||||||
|
storage.WithOrderDirection(storage.OrderDirectionAsc),
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return errors.WithStack(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if e, g := 3, len(results); e != g {
|
||||||
|
return errors.Errorf("len(results): expected '%v', got '%v'", e, g)
|
||||||
|
}
|
||||||
|
|
||||||
|
if e, g := docs[0]["name"], results[0]["name"]; e != g {
|
||||||
|
return errors.Errorf("results[0][\"name\"]: expected '%v', got '%v'", e, g)
|
||||||
|
}
|
||||||
|
|
||||||
|
if e, g := docs[2]["name"], results[2]["name"]; e != g {
|
||||||
|
return errors.Errorf("results[2][\"name\"]: expected '%v', got '%v'", e, g)
|
||||||
|
}
|
||||||
|
|
||||||
|
results, err = store.Query(
|
||||||
|
ctx, collection, nil,
|
||||||
|
storage.WithOrderBy("sortedField"),
|
||||||
|
storage.WithOrderDirection(storage.OrderDirectionDesc),
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return errors.WithStack(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if e, g := 3, len(results); e != g {
|
||||||
|
return errors.Errorf("len(results): expected '%v', got '%v'", e, g)
|
||||||
|
}
|
||||||
|
|
||||||
|
if e, g := docs[2]["name"], results[0]["name"]; e != g {
|
||||||
|
return errors.Errorf("results[0][\"name\"]: expected '%v', got '%v'", e, g)
|
||||||
|
}
|
||||||
|
|
||||||
|
if e, g := docs[0]["name"], results[2]["name"]; e != g {
|
||||||
|
return errors.Errorf("results[2][\"name\"]: expected '%v', got '%v'", e, g)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "Query order by special attr",
|
||||||
|
Run: func(ctx context.Context, store storage.DocumentStore) error {
|
||||||
|
docs := []storage.Document{
|
||||||
|
{
|
||||||
|
"name": "Item 1",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "Item 2",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "Item 3",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
collection := "ordered_query_by_special_attr"
|
||||||
|
|
||||||
|
for _, doc := range docs {
|
||||||
|
if _, err := store.Upsert(ctx, collection, doc); err != nil {
|
||||||
|
return errors.WithStack(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
results, err := store.Query(
|
||||||
|
ctx, collection, nil,
|
||||||
|
storage.WithOrderBy(storage.DocumentAttrCreatedAt),
|
||||||
|
storage.WithOrderDirection(storage.OrderDirectionAsc),
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return errors.WithStack(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if e, g := 3, len(results); e != g {
|
||||||
|
return errors.Errorf("len(results): expected '%v', got '%v'", e, g)
|
||||||
|
}
|
||||||
|
|
||||||
|
if e, g := docs[0]["name"], results[0]["name"]; e != g {
|
||||||
|
return errors.Errorf("results[0][\"name\"]: expected '%v', got '%v'", e, g)
|
||||||
|
}
|
||||||
|
|
||||||
|
if e, g := docs[2]["name"], results[2]["name"]; e != g {
|
||||||
|
return errors.Errorf("results[2][\"name\"]: expected '%v', got '%v'", e, g)
|
||||||
|
}
|
||||||
|
|
||||||
|
results, err = store.Query(
|
||||||
|
ctx, collection, nil,
|
||||||
|
storage.WithOrderBy(storage.DocumentAttrCreatedAt),
|
||||||
|
storage.WithOrderDirection(storage.OrderDirectionDesc),
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return errors.WithStack(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if e, g := 3, len(results); e != g {
|
||||||
|
return errors.Errorf("len(results): expected '%v', got '%v'", e, g)
|
||||||
|
}
|
||||||
|
|
||||||
|
if e, g := docs[2]["name"], results[0]["name"]; e != g {
|
||||||
|
return errors.Errorf("results[0][\"name\"]: expected '%v', got '%v'", e, g)
|
||||||
|
}
|
||||||
|
|
||||||
|
if e, g := docs[0]["name"], results[2]["name"]; e != g {
|
||||||
|
return errors.Errorf("results[2][\"name\"]: expected '%v', got '%v'", e, g)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "Query limit and offset",
|
||||||
|
Run: func(ctx context.Context, store storage.DocumentStore) error {
|
||||||
|
docs := []storage.Document{
|
||||||
|
{"name": "Item 1"},
|
||||||
|
{"name": "Item 2"},
|
||||||
|
{"name": "Item 3"},
|
||||||
|
{"name": "Item 4"},
|
||||||
|
}
|
||||||
|
|
||||||
|
collection := "query_limit_and_offset"
|
||||||
|
|
||||||
|
for _, doc := range docs {
|
||||||
|
if _, err := store.Upsert(ctx, collection, doc); err != nil {
|
||||||
|
return errors.WithStack(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
results, err := store.Query(
|
||||||
|
ctx, collection, nil,
|
||||||
|
storage.WithLimit(2),
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return errors.WithStack(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if e, g := 2, len(results); e != g {
|
||||||
|
return errors.Errorf("len(results): expected '%v', got '%v'", e, g)
|
||||||
|
}
|
||||||
|
|
||||||
|
if e, g := docs[0]["name"], results[0]["name"]; e != g {
|
||||||
|
return errors.Errorf("results[0][\"name\"]: expected '%v', got '%v'", e, g)
|
||||||
|
}
|
||||||
|
|
||||||
|
if e, g := docs[1]["name"], results[1]["name"]; e != g {
|
||||||
|
return errors.Errorf("results[1][\"name\"]: expected '%v', got '%v'", e, g)
|
||||||
|
}
|
||||||
|
|
||||||
|
results, err = store.Query(
|
||||||
|
ctx, collection, nil,
|
||||||
|
storage.WithOffset(2),
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return errors.WithStack(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if e, g := 2, len(results); e != g {
|
||||||
|
return errors.Errorf("len(results): expected '%v', got '%v'", e, g)
|
||||||
|
}
|
||||||
|
|
||||||
|
if e, g := docs[2]["name"], results[0]["name"]; e != g {
|
||||||
|
return errors.Errorf("results[0][\"name\"]: expected '%v', got '%v'", e, g)
|
||||||
|
}
|
||||||
|
|
||||||
|
if e, g := docs[3]["name"], results[1]["name"]; e != g {
|
||||||
|
return errors.Errorf("results[1][\"name\"]: expected '%v', got '%v'", e, g)
|
||||||
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
|
|||||||
Reference in New Issue
Block a user