Skip to content

Commit e6c1f0c

Browse files
authored
fix: Generate JavaScript Code from plugin-pb (#66)
Co-authored-by: cq-bot <[email protected]>
1 parent a6e7fd2 commit e6c1f0c

File tree

2 files changed

+202
-0
lines changed

2 files changed

+202
-0
lines changed

protos/plugin/v3/plugin.proto

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,8 @@ service Plugin {
2929
rpc Write(stream Write.Request) returns (Write.Response);
3030
// Transform resources.
3131
rpc Transform(stream Transform.Request) returns (stream Transform.Response);
32+
// Transform schemas.
33+
rpc TransformSchema(stream TransformSchema.Request) returns (stream TransformSchema.Response);
3234
// Send signal to flush and close open connections
3335
rpc Close(Close.Request) returns (Close.Response);
3436
// Validate and test the connections used by the plugin
@@ -203,6 +205,17 @@ message Transform {
203205
}
204206
}
205207

208+
message TransformSchema {
209+
message Request {
210+
// marshalled arrow.Schema
211+
bytes schema = 1;
212+
}
213+
message Response {
214+
// marshalled arrow.Schema
215+
bytes schema = 1;
216+
}
217+
}
218+
206219
message Close {
207220
message Request {}
208221
message Response {}

protos/plugin/v3/plugin.ts

Lines changed: 189 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2890,6 +2890,182 @@ export namespace cloudquery.plugin.v3 {
28902890
}
28912891
}
28922892
}
2893+
export class TransformSchema extends pb_1.Message {
2894+
#one_of_decls: number[][] = [];
2895+
constructor(data?: any[] | {}) {
2896+
super();
2897+
pb_1.Message.initialize(this, Array.isArray(data) ? data : [], 0, -1, [], this.#one_of_decls);
2898+
if (!Array.isArray(data) && typeof data == "object") { }
2899+
}
2900+
static fromObject(data: {}): TransformSchema {
2901+
const message = new TransformSchema({});
2902+
return message;
2903+
}
2904+
toObject() {
2905+
const data: {} = {};
2906+
return data;
2907+
}
2908+
serialize(): Uint8Array;
2909+
serialize(w: pb_1.BinaryWriter): void;
2910+
serialize(w?: pb_1.BinaryWriter): Uint8Array | void {
2911+
const writer = w || new pb_1.BinaryWriter();
2912+
if (!w)
2913+
return writer.getResultBuffer();
2914+
}
2915+
static deserialize(bytes: Uint8Array | pb_1.BinaryReader): TransformSchema {
2916+
const reader = bytes instanceof pb_1.BinaryReader ? bytes : new pb_1.BinaryReader(bytes), message = new TransformSchema();
2917+
while (reader.nextField()) {
2918+
if (reader.isEndGroup())
2919+
break;
2920+
switch (reader.getFieldNumber()) {
2921+
default: reader.skipField();
2922+
}
2923+
}
2924+
return message;
2925+
}
2926+
serializeBinary(): Uint8Array {
2927+
return this.serialize();
2928+
}
2929+
static deserializeBinary(bytes: Uint8Array): TransformSchema {
2930+
return TransformSchema.deserialize(bytes);
2931+
}
2932+
}
2933+
export namespace TransformSchema {
2934+
export class Request extends pb_1.Message {
2935+
#one_of_decls: number[][] = [];
2936+
constructor(data?: any[] | {
2937+
schema?: Uint8Array;
2938+
}) {
2939+
super();
2940+
pb_1.Message.initialize(this, Array.isArray(data) ? data : [], 0, -1, [], this.#one_of_decls);
2941+
if (!Array.isArray(data) && typeof data == "object") {
2942+
if ("schema" in data && data.schema != undefined) {
2943+
this.schema = data.schema;
2944+
}
2945+
}
2946+
}
2947+
get schema() {
2948+
return pb_1.Message.getFieldWithDefault(this, 1, new Uint8Array(0)) as Uint8Array;
2949+
}
2950+
set schema(value: Uint8Array) {
2951+
pb_1.Message.setField(this, 1, value);
2952+
}
2953+
static fromObject(data: {
2954+
schema?: Uint8Array;
2955+
}): Request {
2956+
const message = new Request({});
2957+
if (data.schema != null) {
2958+
message.schema = data.schema;
2959+
}
2960+
return message;
2961+
}
2962+
toObject() {
2963+
const data: {
2964+
schema?: Uint8Array;
2965+
} = {};
2966+
if (this.schema != null) {
2967+
data.schema = this.schema;
2968+
}
2969+
return data;
2970+
}
2971+
serialize(): Uint8Array;
2972+
serialize(w: pb_1.BinaryWriter): void;
2973+
serialize(w?: pb_1.BinaryWriter): Uint8Array | void {
2974+
const writer = w || new pb_1.BinaryWriter();
2975+
if (this.schema.length)
2976+
writer.writeBytes(1, this.schema);
2977+
if (!w)
2978+
return writer.getResultBuffer();
2979+
}
2980+
static deserialize(bytes: Uint8Array | pb_1.BinaryReader): Request {
2981+
const reader = bytes instanceof pb_1.BinaryReader ? bytes : new pb_1.BinaryReader(bytes), message = new Request();
2982+
while (reader.nextField()) {
2983+
if (reader.isEndGroup())
2984+
break;
2985+
switch (reader.getFieldNumber()) {
2986+
case 1:
2987+
message.schema = reader.readBytes();
2988+
break;
2989+
default: reader.skipField();
2990+
}
2991+
}
2992+
return message;
2993+
}
2994+
serializeBinary(): Uint8Array {
2995+
return this.serialize();
2996+
}
2997+
static deserializeBinary(bytes: Uint8Array): Request {
2998+
return Request.deserialize(bytes);
2999+
}
3000+
}
3001+
export class Response extends pb_1.Message {
3002+
#one_of_decls: number[][] = [];
3003+
constructor(data?: any[] | {
3004+
schema?: Uint8Array;
3005+
}) {
3006+
super();
3007+
pb_1.Message.initialize(this, Array.isArray(data) ? data : [], 0, -1, [], this.#one_of_decls);
3008+
if (!Array.isArray(data) && typeof data == "object") {
3009+
if ("schema" in data && data.schema != undefined) {
3010+
this.schema = data.schema;
3011+
}
3012+
}
3013+
}
3014+
get schema() {
3015+
return pb_1.Message.getFieldWithDefault(this, 1, new Uint8Array(0)) as Uint8Array;
3016+
}
3017+
set schema(value: Uint8Array) {
3018+
pb_1.Message.setField(this, 1, value);
3019+
}
3020+
static fromObject(data: {
3021+
schema?: Uint8Array;
3022+
}): Response {
3023+
const message = new Response({});
3024+
if (data.schema != null) {
3025+
message.schema = data.schema;
3026+
}
3027+
return message;
3028+
}
3029+
toObject() {
3030+
const data: {
3031+
schema?: Uint8Array;
3032+
} = {};
3033+
if (this.schema != null) {
3034+
data.schema = this.schema;
3035+
}
3036+
return data;
3037+
}
3038+
serialize(): Uint8Array;
3039+
serialize(w: pb_1.BinaryWriter): void;
3040+
serialize(w?: pb_1.BinaryWriter): Uint8Array | void {
3041+
const writer = w || new pb_1.BinaryWriter();
3042+
if (this.schema.length)
3043+
writer.writeBytes(1, this.schema);
3044+
if (!w)
3045+
return writer.getResultBuffer();
3046+
}
3047+
static deserialize(bytes: Uint8Array | pb_1.BinaryReader): Response {
3048+
const reader = bytes instanceof pb_1.BinaryReader ? bytes : new pb_1.BinaryReader(bytes), message = new Response();
3049+
while (reader.nextField()) {
3050+
if (reader.isEndGroup())
3051+
break;
3052+
switch (reader.getFieldNumber()) {
3053+
case 1:
3054+
message.schema = reader.readBytes();
3055+
break;
3056+
default: reader.skipField();
3057+
}
3058+
}
3059+
return message;
3060+
}
3061+
serializeBinary(): Uint8Array {
3062+
return this.serialize();
3063+
}
3064+
static deserializeBinary(bytes: Uint8Array): Response {
3065+
return Response.deserialize(bytes);
3066+
}
3067+
}
3068+
}
28933069
export class Close extends pb_1.Message {
28943070
#one_of_decls: number[][] = [];
28953071
constructor(data?: any[] | {}) {
@@ -3341,6 +3517,15 @@ export namespace cloudquery.plugin.v3 {
33413517
responseSerialize: (message: Transform.Response) => Buffer.from(message.serialize()),
33423518
responseDeserialize: (bytes: Buffer) => Transform.Response.deserialize(new Uint8Array(bytes))
33433519
},
3520+
TransformSchema: {
3521+
path: "/cloudquery.plugin.v3.Plugin/TransformSchema",
3522+
requestStream: true,
3523+
responseStream: true,
3524+
requestSerialize: (message: TransformSchema.Request) => Buffer.from(message.serialize()),
3525+
requestDeserialize: (bytes: Buffer) => TransformSchema.Request.deserialize(new Uint8Array(bytes)),
3526+
responseSerialize: (message: TransformSchema.Response) => Buffer.from(message.serialize()),
3527+
responseDeserialize: (bytes: Buffer) => TransformSchema.Response.deserialize(new Uint8Array(bytes))
3528+
},
33443529
Close: {
33453530
path: "/cloudquery.plugin.v3.Plugin/Close",
33463531
requestStream: false,
@@ -3370,6 +3555,7 @@ export namespace cloudquery.plugin.v3 {
33703555
abstract Read(call: grpc_1.ServerWritableStream<Read.Request, Read.Response>): void;
33713556
abstract Write(call: grpc_1.ServerReadableStream<Write.Request, Write.Response>, callback: grpc_1.sendUnaryData<Write.Response>): void;
33723557
abstract Transform(call: grpc_1.ServerDuplexStream<Transform.Request, Transform.Response>): void;
3558+
abstract TransformSchema(call: grpc_1.ServerDuplexStream<TransformSchema.Request, TransformSchema.Response>): void;
33733559
abstract Close(call: grpc_1.ServerUnaryCall<Close.Request, Close.Response>, callback: grpc_1.sendUnaryData<Close.Response>): void;
33743560
abstract TestConnection(call: grpc_1.ServerUnaryCall<TestConnection.Request, TestConnection.Response>, callback: grpc_1.sendUnaryData<TestConnection.Response>): void;
33753561
}
@@ -3404,6 +3590,9 @@ export namespace cloudquery.plugin.v3 {
34043590
Transform: GrpcChunkServiceInterface<Transform.Request, Transform.Response> = (metadata?: grpc_1.Metadata | grpc_1.CallOptions, options?: grpc_1.CallOptions): grpc_1.ClientDuplexStream<Transform.Request, Transform.Response> => {
34053591
return super.Transform(metadata, options);
34063592
};
3593+
TransformSchema: GrpcChunkServiceInterface<TransformSchema.Request, TransformSchema.Response> = (metadata?: grpc_1.Metadata | grpc_1.CallOptions, options?: grpc_1.CallOptions): grpc_1.ClientDuplexStream<TransformSchema.Request, TransformSchema.Response> => {
3594+
return super.TransformSchema(metadata, options);
3595+
};
34073596
Close: GrpcUnaryServiceInterface<Close.Request, Close.Response> = (message: Close.Request, metadata: grpc_1.Metadata | grpc_1.CallOptions | grpc_1.requestCallback<Close.Response>, options?: grpc_1.CallOptions | grpc_1.requestCallback<Close.Response>, callback?: grpc_1.requestCallback<Close.Response>): grpc_1.ClientUnaryCall => {
34083597
return super.Close(message, metadata, options, callback);
34093598
};

0 commit comments

Comments
 (0)