Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions .changeset/witty-ducks-work.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
---
'@powersync/service-core': minor
---

Added support for user parameters when making a StreamingSyncRequest.
6 changes: 5 additions & 1 deletion packages/service-core/src/routes/socket-route.ts
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,11 @@ export const sync_stream_reactive: SocketRouteGenerator = (router) =>

const controller = new AbortController();

const syncParams: SyncParameters = normalizeTokenParameters(context.token_payload?.parameters ?? {});
const syncParams: SyncParameters = normalizeTokenParameters(
context.token_payload?.parameters ?? {},
params.parameters ?? {}
);

const storage = system.storage;
// Sanity check before we start the stream
const cp = await storage.getActiveCheckpoint();
Expand Down
5 changes: 4 additions & 1 deletion packages/service-core/src/routes/sync-stream.ts
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,10 @@ export const syncStreamed: RouteGenerator = (router) =>
}

const params: util.StreamingSyncRequest = payload.params;
const syncParams: SyncParameters = normalizeTokenParameters(payload.context.token_payload!.parameters ?? {});
const syncParams: SyncParameters = normalizeTokenParameters(
payload.context.token_payload!.parameters ?? {},
payload.params.parameters ?? {}
);

const storage = system.storage;
// Sanity check before we start the stream
Expand Down
7 changes: 6 additions & 1 deletion packages/service-core/src/util/protocol-types.ts
Original file line number Diff line number Diff line change
Expand Up @@ -89,7 +89,12 @@ export const StreamingSyncRequest = t.object({
/**
* Data is received in a serialized BSON Buffer
*/
binary_data: t.boolean.optional()
binary_data: t.boolean.optional(),

/**
* Client parameters to be passed to the sync rules.
*/
parameters: t.record(t.any).optional()
});

export type StreamingSyncRequest = t.Decoded<typeof StreamingSyncRequest>;
Expand Down
20 changes: 11 additions & 9 deletions packages/service-core/test/src/sync_rules.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -100,27 +100,29 @@ bucket_definitions:
const rules = SqlSyncRules.fromYaml(`
bucket_definitions:
mybucket:
parameters: SELECT token_parameters.user_id
parameters: SELECT token_parameters.user_id, user_parameters.device_id
data:
- SELECT id, description FROM assets WHERE assets.user_id = bucket.user_id AND NOT assets.archived
- SELECT id, description FROM assets WHERE assets.user_id = bucket.user_id AND assets.device_id = bucket.device_id AND NOT assets.archived
`);
const bucket = rules.bucket_descriptors[0];
expect(bucket.bucket_parameters).toEqual(['user_id']);
expect(bucket.bucket_parameters).toEqual(['user_id', 'device_id']);
const param_query = bucket.global_parameter_queries[0];
expect(param_query.bucket_parameters).toEqual(['user_id']);
expect(rules.getStaticBucketIds(normalizeTokenParameters({ user_id: 'user1' }))).toEqual(['mybucket["user1"]']);
expect(param_query.bucket_parameters).toEqual(['user_id', 'device_id']);
expect(rules.getStaticBucketIds(normalizeTokenParameters({ user_id: 'user1' }, { device_id: 'device1' }))).toEqual([
'mybucket["user1","device1"]'
]);

const data_query = bucket.data_queries[0];
expect(data_query.bucket_parameters).toEqual(['user_id']);
expect(data_query.bucket_parameters).toEqual(['user_id', 'device_id']);
expect(
rules.evaluateRow({
sourceTable: ASSETS,
record: { id: 'asset1', description: 'test', user_id: 'user1' }
record: { id: 'asset1', description: 'test', user_id: 'user1', device_id: 'device1' }
})
).toEqual([
{
ruleId: '1',
bucket: 'mybucket["user1"]',
bucket: 'mybucket["user1","device1"]',
id: 'asset1',
data: {
id: 'asset1',
Expand All @@ -132,7 +134,7 @@ bucket_definitions:
expect(
rules.evaluateRow({
sourceTable: ASSETS,
record: { id: 'asset1', description: 'test', user_id: 'user1', archived: 1 }
record: { id: 'asset1', description: 'test', user_id: 'user1', archived: 1, device_id: 'device1' }
})
).toEqual([]);
});
Expand Down