(table *schema.Table, rowCount int, overwrite bool)
| 12 | ) |
| 13 | |
| 14 | func getInsertQueryBuild(table *schema.Table, rowCount int, overwrite bool) *strings.Builder { |
| 15 | builder := strings.Builder{} |
| 16 | builder.WriteString("INSERT INTO " + identifier(table.Name)) |
| 17 | builder.WriteString(" (") |
| 18 | |
| 19 | for i, col := range table.Columns { |
| 20 | builder.WriteString(identifier(col.Name)) |
| 21 | if i < len(table.Columns)-1 { |
| 22 | builder.WriteString(", ") |
| 23 | } |
| 24 | } |
| 25 | builder.WriteString(") VALUES ") |
| 26 | |
| 27 | for i := 1; i <= rowCount; i++ { |
| 28 | if i > 1 { |
| 29 | builder.WriteString(", ") |
| 30 | } |
| 31 | builder.WriteString("(") |
| 32 | builder.WriteString(strings.TrimSuffix(strings.Repeat("?,", len(table.Columns)), ",")) |
| 33 | builder.WriteString(")") |
| 34 | } |
| 35 | |
| 36 | if overwrite { |
| 37 | builder.WriteString(" ON DUPLICATE KEY UPDATE ") |
| 38 | for i, col := range table.Columns { |
| 39 | builder.WriteString(fmt.Sprintf("%s = VALUES(%s)", identifier(col.Name), identifier(col.Name))) |
| 40 | if i < len(table.Columns)-1 { |
| 41 | builder.WriteString(", ") |
| 42 | } |
| 43 | } |
| 44 | } |
| 45 | |
| 46 | return &builder |
| 47 | } |
| 48 | |
| 49 | func logTablesWithTruncation(logger zerolog.Logger, tables map[string]bool) { |
| 50 | if len(tables) == 0 { |
no test coverage detected