Skip to content

Commit 5a8f3c3

Browse files
authored
fix: Flatten tables in GetTables gRPC call (#57)
1 parent 7e39769 commit 5a8f3c3

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

src/grpc/plugin.ts

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ import { pluginV3 } from '@cloudquery/plugin-pb-javascript';
22
import grpc = require('@grpc/grpc-js');
33

44
import { Plugin } from '../plugin/plugin.js';
5-
import { encodeTables } from '../schema/table.js';
5+
import { encodeTables, flattenTables } from '../schema/table.js';
66

77
export class MigrateTable extends pluginV3.cloudquery.plugin.v3.Sync.MessageMigrateTable {}
88
export class DeleteStale extends pluginV3.cloudquery.plugin.v3.Write.MessageDeleteStale {}
@@ -81,8 +81,12 @@ export class PluginServer extends pluginV3.cloudquery.plugin.v3.UnimplementedPlu
8181
this.plugin
8282
.tables({ tables, skipTables, skipDependentTables })
8383
.then((tables) => {
84+
const flattened = flattenTables(tables);
8485
// eslint-disable-next-line promise/no-callback-in-promise
85-
return callback(null, new pluginV3.cloudquery.plugin.v3.GetTables.Response({ tables: encodeTables(tables) }));
86+
return callback(
87+
null,
88+
new pluginV3.cloudquery.plugin.v3.GetTables.Response({ tables: encodeTables(flattened) }),
89+
);
8690
})
8791
.catch((error) => {
8892
// eslint-disable-next-line promise/no-callback-in-promise

0 commit comments

Comments
 (0)