feat: add file type to json output

Signed-off-by: Andres Taylor <[email protected]>

go/summarize/utils.go

@@ -30,12 +30,14 @@ const (
 	traceFile
 	keysFile
 	dbInfoFile
+	transactionFile
 )
 
 var fileTypeMap = map[string]fileType{ //nolint:gochecknoglobals // this is instead of a const
-	"trace":  traceFile,
-	"keys":   keysFile,
-	"dbinfo": dbInfoFile,
+	"trace":        traceFile,
+	"keys":         keysFile,
+	"dbinfo":       dbInfoFile,
+	"transactions": transactionFile,
 }
 
 // getFileType reads the first key-value pair from a JSON file and returns the type of the file

go/testdata/small-slow-query-transactions.json

@@ -1,15 +1,18 @@
-[
-  {
-    "query-signatures": [
-      "update tblA where foo = :0 and id = :1 set apa",
-      "update tblB where bar = :0 and id = :2 set monkey"
-    ],
-    "predicates": [
-      "tblA.foo = 0",
-      "tblA.id = ?",
-      "tblB.bar = 0",
-      "tblB.id = ?"
-    ],
-    "count": 2
-  }
-]
+{
+  "fileType": "transactions",
+  "signatures": [
+    {
+      "query-signatures": [
+        "update tblA where foo = :0 and id = :1 set apa",
+        "update tblB where bar = :0 and id = :2 set monkey"
+      ],
+      "predicates": [
+        "tblA.foo = 0",
+        "tblA.id = ?",
+        "tblB.bar = 0",
+        "tblB.id = ?"
+      ],
+      "count": 2
+    }
+  ]
+}

go/transactions/transaction_signature.go

@@ -220,5 +220,10 @@ func (m *txSignatureMap) MarshalJSON() ([]byte, error) {
 		return signatures[i].Count > signatures[j].Count
 	})
 
-	return json.Marshal(signatures)
+	result := map[string]any{
+		"fileType":   "transactions",
+		"signatures": signatures,
+	}
+
+	return json.Marshal(result)
 }

go/transactions/transactions_test.go

@@ -45,4 +45,7 @@ func TestRun(t *testing.T) {
 	require.NoError(t, err)
 
 	assert.Equal(t, string(out), sb.String())
+	if t.Failed() {
+		_ = os.WriteFile("../testdata/expected/small-slow-query-transactions.json", []byte(sb.String()), 0o644)
+	}
 }