Run CREATE statement in backfill script (#2109)

For the case where we want to start populating a new MySQL instance but the Rekor server storage provider is not actually switched to MySQL yet, ensure the table is created before trying to copy the data. Signed-off-by: Colleen Murphy <[email protected]>
sigstore · May 8, 2024 · 891581e · 891581e
1 parent 7f43202
commit 891581e
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/cmd/backfill-index/main.go b/cmd/backfill-index/main.go
@@ -69,7 +69,14 @@ import (
 )
 
 const (
-	mysqlWriteStmt = "INSERT IGNORE INTO EntryIndex (EntryKey, EntryUUID) VALUES (:key, :uuid)"
+	mysqlWriteStmt       = "INSERT IGNORE INTO EntryIndex (EntryKey, EntryUUID) VALUES (:key, :uuid)"
+	mysqlCreateTableStmt = `CREATE TABLE IF NOT EXISTS EntryIndex (
+		PK BIGINT UNSIGNED NOT NULL AUTO_INCREMENT,
+		EntryKey varchar(512) NOT NULL,
+		EntryUUID char(80) NOT NULL,
+		PRIMARY KEY(PK),
+		UNIQUE(EntryKey, EntryUUID)
+	)`
 )
 
 type provider int
@@ -190,6 +197,9 @@ func getIndexClient(backend provider) (indexClient, error) {
 		if err = dbClient.Ping(); err != nil {
 			return nil, err
 		}
+		if _, err = dbClient.Exec(mysqlCreateTableStmt); err != nil {
+			return nil, err
+		}
 		return &mysqlClient{client: dbClient}, nil
 	default:
 		return nil, fmt.Errorf("could not create client for unexpected provider")