@mastra/mysql 0.3.0 → 0.3.1

package/dist/index.js

@@ -294,6 +294,16 @@ var StoreOperationsMySQL = class extends StoreOperations {
       const pkColumns = tableConfig.compositePrimaryKey.map((col) => quoteIdentifier(col, "primary key column")).join(", ");
       extraConstraints.push(`PRIMARY KEY (${pkColumns})`);
     }
+    if (tableName === TABLE_WORKFLOW_SNAPSHOT) {
+      extraConstraints.push(
+        `PRIMARY KEY (${quoteIdentifier("workflow_name", "column name")}, ${quoteIdentifier("run_id", "column name")})`
+      );
+    }
+    if (tableName === TABLE_SPANS) {
+      extraConstraints.push(
+        `PRIMARY KEY (${quoteIdentifier("traceId", "column name")}, ${quoteIdentifier("spanId", "column name")})`
+      );
+    }
     return `CREATE TABLE IF NOT EXISTS ${tableIdent} (${[...columns, ...extraConstraints].filter(Boolean).join(", ")}) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci`;
   }
   isKeyColumn(tableName, columnName) {
@@ -6886,7 +6896,8 @@ var ObservabilityMySQL = class _ObservabilityMySQL extends ObservabilityStorage
     statements.push(
       generateTableSQL({
         tableName: TABLE_SPANS,
-        schema: TABLE_SCHEMAS[TABLE_SPANS]
+        schema: TABLE_SCHEMAS[TABLE_SPANS],
+        compositePrimaryKey: ["traceId", "spanId"]
       })
     );
     for (const idx of _ObservabilityMySQL.getDefaultIndexDefs()) {
@@ -9872,7 +9883,13 @@ var WorkflowsMySQL = class _WorkflowsMySQL extends WorkflowsStorage {
    * Exports DDL statements for all managed tables.
    */
   static getExportDDL() {
-    return [generateTableSQL({ tableName: TABLE_WORKFLOW_SNAPSHOT, schema: TABLE_SCHEMAS[TABLE_WORKFLOW_SNAPSHOT] })];
+    return [
+      generateTableSQL({
+        tableName: TABLE_WORKFLOW_SNAPSHOT,
+        schema: TABLE_SCHEMAS[TABLE_WORKFLOW_SNAPSHOT],
+        compositePrimaryKey: ["workflow_name", "run_id"]
+      })
+    ];
   }
   constructor({
     operations,
@@ -10050,6 +10067,17 @@ var WorkflowsMySQL = class _WorkflowsMySQL extends WorkflowsStorage {
     const updatedAtValue = updatedAt ?? now;
     try {
       const tableName = formatTableName(TABLE_WORKFLOW_SNAPSHOT);
+      const [updateResult] = await this.pool.execute(
+        `UPDATE ${tableName}
+         SET ${quoteIdentifier("resourceId", "column name")} = ?,
+             ${quoteIdentifier("snapshot", "column name")} = ?,
+             ${quoteIdentifier("updatedAt", "column name")} = ?
+         WHERE ${quoteIdentifier("workflow_name", "column name")} = ? AND ${quoteIdentifier("run_id", "column name")} = ?`,
+        [resourceId ?? null, JSON.stringify(snapshot), transformToSqlValue(updatedAtValue), workflowName, runId]
+      );
+      if (updateResult.affectedRows > 0) {
+        return;
+      }
       await this.pool.execute(
         `INSERT INTO ${tableName} (${quoteIdentifier("workflow_name", "column name")}, ${quoteIdentifier("run_id", "column name")}, ${quoteIdentifier("resourceId", "column name")}, ${quoteIdentifier("snapshot", "column name")}, ${quoteIdentifier("createdAt", "column name")}, ${quoteIdentifier("updatedAt", "column name")})
          VALUES (?, ?, ?, ?, ?, ?)