Method executeLoadData

backend/loaddata.go:115–227

(ctx *sql.Context, insert *plan.InsertInto, dst sql.InsertableTable, load *plan.LoadData, filePath string)

}

func (db *DuckBuilder) executeLoadData(ctx *sql.Context, insert *plan.InsertInto, dst sql.InsertableTable, load *plan.LoadData, filePath string) (sql.RowIter, error) {
	// Build the DuckDB INSERT INTO statement.
	var b strings.Builder
	b.Grow(256)

	keyless := sql.IsKeyless(dst.Schema())
	// TODO(fan): This is an ugly hack for MySQL Shell's utilities to work properly.
	if !keyless {
		if t, ok := dst.(*catalog.Table); ok {
			// Replicated tables do not have physical primary keys.
			// Their logical primary keys are fake and should not be used in INSERT INTO statements.
			// https://github.com/apecloud/myduckserver/issues/272
			keyless = t.ExtraTableInfo().Replicated || !t.HasPrimaryKey()
		}
	}

	b.WriteString("INSERT")
	if load.IsIgnore && !keyless {
		b.WriteString(" OR IGNORE")
	} else if load.IsReplace && !keyless {
		b.WriteString(" OR REPLACE")
	}
	b.WriteString(" INTO ")

	qualifiedTableName := catalog.ConnectIdentifiersANSI(insert.Database().Name(), dst.Name())
	b.WriteString(qualifiedTableName)

	if len(load.ColNames) > 0 {
		b.WriteString(" (")
		b.WriteString(catalog.QuoteIdentifierANSI(load.ColNames[0]))
		for _, col := range load.ColNames[1:] {
			b.WriteString(", ")
			b.WriteString(catalog.QuoteIdentifierANSI(col))
		}
		b.WriteString(")")
	}

	b.WriteString(" FROM ")
	b.WriteString("read_csv('")
	b.WriteString(filePath)
	b.WriteString("'")

	b.WriteString(", auto_detect = false")
	b.WriteString(", header = false")
	b.WriteString(", null_padding = true")

	b.WriteString(", new_line = ")
	if len(load.LinesTerminatedBy) == 1 {
		b.WriteString(singleQuotedDuckChar(load.LinesTerminatedBy))
	} else {
		b.WriteString(`'\r\n'`)
	}

	b.WriteString(", sep = ")
	b.WriteString(singleQuotedDuckChar(load.FieldsTerminatedBy))

	b.WriteString(", quote = ")
	b.WriteString(singleQuotedDuckChar(load.FieldsEnclosedBy))