App file changes and other actions.

Author: malexanderlim

components/azure_sql/actions/execute-query/execute-query.mjs

@@ -5,7 +5,7 @@ export default {
   name: "Execute Query",
   description: "Executes a SQL query and returns the results. [See the documentation](https://learn.microsoft.com/en-us/sql/t-sql/queries/select-transact-sql?view=azuresqldb-current)",
   type: "action",
-  version: "0.0.5",
+  version: "0.0.6",
   props: {
     app,
     query: {
@@ -20,18 +20,26 @@ export default {
       description: "The inputs to the query. These will be available as @input_parameter in the query. For example, if you provide an input named 'id', you can use @id in the query.",
       optional: true,
     },
+    requestTimeout: {
+      propDefinition: [
+        app,
+        "requestTimeout",
+      ],
+    },
   },
   run({ $ }) {
     const {
       app,
       inputs,
       query,
+      requestTimeout,
     } = this;
 
     return app.executeQuery({
       $,
       query,
       inputs,
+      requestTimeout,
       summary: () => "Successfully executed query.",
     });
   },

components/azure_sql/azure_sql.app.mjs

@@ -28,6 +28,13 @@ export default {
         return recordset.map(({ COLUMN_NAME: columnName }) => columnName);
       },
     },
+    requestTimeout: {
+      type: "integer",
+      label: "Request Timeout",
+      description: "The timeout for query execution in seconds. Default is 60 seconds if not specified.",
+      optional: true,
+      default: 60,
+    },
   },
   methods: {
     ...sqlProxy.methods,
@@ -37,7 +44,7 @@ export default {
     },
     getConfig() {
       const {
-        host, username, password, port, database,
+        host, username, password, port, database, requestTimeout,
       } = this.$auth;
       return {
         user: username,
@@ -51,21 +58,13 @@ export default {
         options: {
           encrypt: true,
           port: Number(port),
-          requestTimeout: 60000,
+          requestTimeout: (requestTimeout || 60) * 1000, // Convert to milliseconds
         },
       };
     },
     getConnection() {
       return sql.connect(this.getConfig());
     },
-    /**
-     * A helper method to get the schema of the database. Used by other features
-     * (like the `sql` prop) to enrich the code editor and provide the user with
-     * auto-complete and fields suggestion.
-     *
-     * @returns {DbInfo} The schema of the database, which is a
-     * JSON-serializable object.
-     */
     async getSchema() {
       const sql = `
         SELECT t.TABLE_SCHEMA AS tableSchema,
@@ -98,15 +97,6 @@ export default {
         return acc;
       }, {});
     },
-    /**
-     * Adapts the arguments to `executeQuery` so that they can be consumed by
-     * the SQL proxy (when applicable). Note that this method is not intended to
-     * be used by the component directly.
-     * @param {object} preparedStatement The prepared statement to be sent to the DB.
-     * @param {string} preparedStatement.sql The prepared SQL query to be executed.
-     * @param {string[]} preparedStatement.values The values to replace in the SQL query.
-     * @returns {object} - The adapted query and parameters.
-     */
     proxyAdapter(preparedStatement = {}) {
       const { query } = preparedStatement;
       const inputs = preparedStatement?.inputs || {};
@@ -121,16 +111,6 @@ export default {
         params: [],
       };
     },
-    /**
-     * A method that performs the inverse transformation of `proxyAdapter`.
-     *
-     * @param {object} proxyArgs - The output of `proxyAdapter`.
-     * @param {string} proxyArgs.query - The SQL query to be executed.
-     * @param {string[]} proxyArgs.params - The values to replace in the SQL
-     * query.
-     * @returns {object} - The adapted query and parameters, compatible with
-     * `executeQuery`.
-     */
     executeQueryAdapter(proxyArgs = {}) {
       let { query } = proxyArgs;
       const params = proxyArgs?.params || [];
@@ -148,10 +128,16 @@ export default {
     },
     async executeQuery(preparedStatement = {}) {
       let connection;
-      const { query } = preparedStatement;
+      const {
+        query, requestTimeout,
+      } = preparedStatement;
       const inputs = preparedStatement?.inputs || {};
       try {
-        connection = await sql.connect(this.getClientConfiguration());
+        const config = this.getConfig();
+        if (requestTimeout) {
+          config.options.requestTimeout = requestTimeout * 1000; // Convert to milliseconds
+        }
+        connection = await sql.connect(config);
         const input =
           Object.entries(inputs)
             .reduce((req, inputArgs) =>

components/azure_sql/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pipedream/azure_sql",
-  "version": "0.1.4",
+  "version": "0.1.5",
   "description": "Pipedream Microsoft Azure SQL Database Components",
   "main": "azure_sql.app.mjs",
   "keywords": [