fix(setup): update chunks in step 11 (#5965)

cmd/setup/11.go

@@ -2,14 +2,25 @@ package setup
 
 import (
 	"context"
+	"database/sql"
 	_ "embed"
+	"time"
+
+	"github.com/cockroachdb/cockroach-go/v2/crdb"
+	"github.com/zitadel/logging"
 
 	"github.com/zitadel/zitadel/internal/database"
 )
 
 var (
-	//go:embed 11.sql
+	//go:embed 11/11_add_column.sql
 	addEventCreatedAt string
+	//go:embed 11/11_fetch_events.sql
+	fetchCreatedAt string
+	//go:embed 11/11_fill_column.sql
+	fillCreatedAt string
+	//go:embed 11/11_set_column.sql
+	setCreatedAtDetails string
 )
 
 type AddEventCreatedAt struct {
@@ -24,6 +35,59 @@ func (mig *AddEventCreatedAt) Execute(ctx context.Context) error {
 		return err
 	}
 	_, err := mig.dbClient.ExecContext(ctx, addEventCreatedAt)
+	if err != nil {
+		return err
+	}
+
+	for {
+		var count int
+		err = crdb.ExecuteTx(ctx, mig.dbClient.DB, nil, func(tx *sql.Tx) error {
+			rows, err := tx.Query(fetchCreatedAt)
+			if err != nil {
+				return err
+			}
+			defer rows.Close()
+
+			data := make(map[string]time.Time, 20)
+			for rows.Next() {
+				count++
+				var (
+					id           string
+					creationDate time.Time
+				)
+
+				err = rows.Scan(&id, &creationDate)
+				if err != nil {
+					return err
+				}
+
+				data[id] = creationDate
+
+			}
+			if err := rows.Err(); err != nil {
+				return err
+			}
+
+			for id, creationDate := range data {
+				_, err = tx.Exec(fillCreatedAt, creationDate, id)
+				if err != nil {
+					return err
+				}
+			}
+
+			return nil
+		})
+		if err != nil {
+			return err
+		}
+		logging.WithFields("count", count).Debug("creation dates set")
+		if count < 20 {
+			break
+		}
+	}
+
+	logging.Info("set details")
+	_, err = mig.dbClient.ExecContext(ctx, setCreatedAtDetails)
 	return err
 }