Implement ":copyfrom" as bulk insert on node-mysql2 driver

examples/authors/mysql/query.sql

@@ -13,6 +13,13 @@ INSERT INTO authors (
  ?, ?
 );
 
+/* name: CreateAuthors :copyfrom */
+INSERT INTO authors (
+ name, bio
+) VALUES (
+ ?, ?
+);
+
 /* name: CreateAuthorReturnId :execlastid */
 INSERT INTO authors (
  name, bio

src/app.ts

@@ -77,6 +77,12 @@ interface Driver{
  params: Parameter[],
  columns: Column[]
  ) => Node;
+ copyfromDecl: (
+ name: string,
+ text: string,
+ iface: string | undefined,
+ params: Parameter[]
+ ) => Node;
 }
 
 function createNodeGenerator(options: Options): Driver{
@@ -143,7 +149,8 @@ function codegen(input: GenerateRequest): GenerateResponse{
  queryDecl(
  textName,
  `-- name: ${query.name} ${query.cmd}
-${query.text}`
+${query.text}`,
+ query.cmd === ":copyfrom"
  )
  );
 
@@ -197,6 +204,12 @@ ${query.text}`
  );
  break;
  }
+ case ":copyfrom":{
+ nodes.push(
+ driver.copyfromDecl(lowerName, textName, argIface, query.params)
+ );
+ break;
+ }
  }
  if (nodes){
  files.push(
@@ -220,7 +233,11 @@ function readInput(): GenerateRequest{
  return GenerateRequest.fromBinary(buffer);
 }
 
-function queryDecl(name: string, sql: string){
+function queryDecl(name: string, sql: string, isCopyFrom = false){
+ const modifiedSql = isCopyFrom 
+ ? sql.replace(/\(\s*\?(?:\s*,\s*\?)*\s*\)/g, "?")
+ : sql;
+
  return factory.createVariableStatement(
  [factory.createToken(SyntaxKind.ExportKeyword)],
  factory.createVariableDeclarationList(
@@ -229,10 +246,10 @@ function queryDecl(name: string, sql: string){
  factory.createIdentifier(name),
  undefined,
  undefined,
- factory.createNoSubstitutionTemplateLiteral(sql, sql)
+ factory.createNoSubstitutionTemplateLiteral(modifiedSql, modifiedSql)
  ),
  ],
- NodeFlags.Const //| NodeFlags.Constant | NodeFlags.Constant
+ NodeFlags.Const
  )
  );
 }

src/drivers/better-sqlite3.ts

@@ -435,4 +435,15 @@ export class Driver{
  "better-sqlite3 driver currently does not support :execlastid"
  );
  }
+
+ copyfromDecl(
+ funcName: string,
+ queryName: string,
+ argIface: string | undefined,
+ params: Parameter[]
+ ): Node{
+ throw new Error(
+ "better-sqlite3 driver currently does not support :copyfrom"
+ );
+ }
 }

src/drivers/mysql2.ts

@@ -749,4 +749,118 @@ export class Driver{
  )
  );
  }
+
+ copyfromDecl(
+ funcName: string,
+ queryName: string,
+ argIface: string | undefined,
+ params: Parameter[]
+ ){
+ const funcParams = [
+ factory.createParameterDeclaration(
+ undefined,
+ undefined,
+ factory.createIdentifier("client"),
+ undefined,
+ factory.createTypeReferenceNode(
+ factory.createIdentifier("Client"),
+ undefined
+ ),
+ undefined
+ ),
+ factory.createParameterDeclaration(
+ undefined,
+ undefined,
+ factory.createIdentifier("args"),
+ undefined,
+ factory.createArrayTypeNode(
+ argIface
+ ? factory.createTypeReferenceNode(
+ factory.createIdentifier(argIface),
+ undefined
+ )
+ : factory.createTypeLiteralNode([])
+ ),
+ undefined
+ ),
+ ];
+
+ return factory.createFunctionDeclaration(
+ [
+ factory.createToken(SyntaxKind.ExportKeyword),
+ factory.createToken(SyntaxKind.AsyncKeyword),
+ ],
+ undefined,
+ factory.createIdentifier(funcName),
+ undefined,
+ funcParams,
+ factory.createTypeReferenceNode(factory.createIdentifier("Promise"), [
+ factory.createKeywordTypeNode(SyntaxKind.VoidKeyword),
+ ]),
+ factory.createBlock(
+ [
+ factory.createExpressionStatement(
+ factory.createAwaitExpression(
+ factory.createCallExpression(
+ factory.createPropertyAccessExpression(
+ factory.createIdentifier("client"),
+ factory.createIdentifier("query")
+ ),
+ undefined,
+ [
+ factory.createObjectLiteralExpression(
+ [
+ factory.createPropertyAssignment(
+ factory.createIdentifier("sql"),
+ factory.createIdentifier(queryName)
+ ),
+ factory.createPropertyAssignment(
+ factory.createIdentifier("values"),
+ factory.createCallExpression(
+ factory.createPropertyAccessExpression(
+ factory.createIdentifier("args"),
+ factory.createIdentifier("map")
+ ),
+ undefined,
+ [
+ factory.createArrowFunction(
+ undefined,
+ undefined,
+ [
+ factory.createParameterDeclaration(
+ undefined,
+ undefined,
+ factory.createIdentifier("arg"),
+ undefined,
+ undefined,
+ undefined
+ ),
+ ],
+ undefined,
+ factory.createToken(SyntaxKind.EqualsGreaterThanToken),
+ factory.createArrayLiteralExpression(
+ params.map((param, i) =>
+ factory.createPropertyAccessExpression(
+ factory.createIdentifier("arg"),
+ factory.createIdentifier(argName(i, param.column))
+ )
+ ),
+ false
+ )
+ ),
+ ]
+ )
+ ),
+ ],
+ true
+ ),
+ ]
+ )
+ )
+ ),
+ ],
+ true
+ )
+ );
+ }
 }

src/drivers/pg.ts

@@ -794,4 +794,13 @@ export class Driver{
  ): FunctionDeclaration{
  throw new Error("pg driver currently does not support :execlastid");
  }
+
+ copyfromDecl(
+ funcName: string,
+ queryName: string,
+ argIface: string | undefined,
+ params: Parameter[]
+ ): Node{
+ throw new Error("pg driver currently does not support :copyfrom");
+ }
 }

src/drivers/postgres.ts

@@ -1,6 +1,7 @@
 import{
  SyntaxKind,
  NodeFlags,
+ Node,
  TypeNode,
  factory,
  FunctionDeclaration,
@@ -619,4 +620,13 @@ export class Driver{
  ): FunctionDeclaration{
  throw new Error("postgres driver currently does not support :execlastid");
  }
+
+ copyfromDecl(
+ funcName: string,
+ queryName: string,
+ argIface: string | undefined,
+ params: Parameter[]
+ ): Node{
+ throw new Error("postgres driver currently does not support :copyfrom");
+ }
 }