Method createTable

plugins/destination/mysql/client/schema.go:145–196

(ctx context.Context, table *schema.Table)

}

func (c *Client) createTable(ctx context.Context, table *schema.Table) error {
	totalColumns := len(table.Columns)
	primaryKeysIndices := []int{}

	builder := strings.Builder{}
	builder.WriteString("CREATE TABLE ")
	builder.WriteString(identifier(table.Name))
	builder.WriteString(" (\n  ")
	for i, column := range table.Columns {
		builder.WriteString(identifier(column.Name))
		builder.WriteString(" ")
		builder.WriteString(arrowTypeToMySqlStr(column.Type))

		if column.PrimaryKey {
			primaryKeysIndices = append(primaryKeysIndices, i)
		} else {
			// Primary keys are implicitly not null and unique, so we only need to add these constraints if the column is not a primary key
			if column.Unique {
				builder.WriteString(" UNIQUE")
			}
			if column.NotNull {
				builder.WriteString(" NOT NULL")
			}
		}

		if i < totalColumns-1 {
			builder.WriteString(",\n  ")
		}
	}
	if len(primaryKeysIndices) > 0 {
		builder.WriteString(",\n  ")
		builder.WriteString(" PRIMARY KEY (")
		lengthPerPk := c.maxIndexLength / len(primaryKeysIndices)
		for i, pk := range primaryKeysIndices {
			column := table.Columns[pk]
			builder.WriteString(identifier(column.Name))
			sqlType := arrowTypeToMySqlStr(column.Type)
			if sqlType == "blob" || sqlType == "text" {
				// `blob/text` SQL types require specifying prefix length to use for the primary key
				// https://dev.mysql.com/doc/refman/8.0/en/innodb-limits.html
				builder.WriteString("(" + strconv.Itoa(lengthPerPk) + ")")
			}
			if i < len(primaryKeysIndices)-1 {
				builder.WriteString(", ")
			}
		}
		builder.WriteString(")\n")
	}
	builder.WriteString(") CHARACTER SET utf8mb4;")
	_, err := c.db.ExecContext(ctx, builder.String())
	return err
}

func (c *Client) dropTable(ctx context.Context, table *schema.Table) error {
	_, err := c.db.ExecContext(ctx, "DROP TABLE "+identifier(table.Name))