column.go

package sq

import "time"

type colmode int

const (
	colmodeInsert colmode = iota
	colmodeUpdate
)

// Column keeps track of what the values mapped to what Field in an InsertQuery/SelectQuery.
type Column struct {
	// mode determines if INSERT or UPDATE
	mode colmode
	// INSERT
	rowStart      bool
	rowEnd        bool
	firstField    string
	insertColumns Fields
	rowValues     RowValues
	// UPDATE
	assignments Assignments
}

// Set maps the value to the Field.
func (col *Column) Set(field Field, value interface{}) {
	if field == nil {
		// should I panic with an error here instead?
		return
	}
	switch col.mode {
	case colmodeUpdate:
		col.assignments = append(col.assignments, FieldAssignment{
			Field: field,
			Value: value,
		})
	case colmodeInsert:
		fallthrough
	default:
		name := field.GetName()
		if !col.rowStart {
			col.rowStart = true
			col.firstField = name
			col.insertColumns = append(col.insertColumns, field)
			col.rowValues = append(col.rowValues, RowValue{value})
			return
		}
		switch name {
		case col.firstField: // Start a new RowValue
			if !col.rowEnd {
				col.rowEnd = true
			}
			col.rowValues = append(col.rowValues, RowValue{value})
		default: // Append to last RowValue
			if !col.rowEnd {
				col.insertColumns = append(col.insertColumns, field)
			}
			last := len(col.rowValues) - 1
			col.rowValues[last] = append(col.rowValues[last], value)
		}
	}
}

// SetBool maps the bool value to the BooleanField.
func (col *Column) SetBool(field BooleanField, value bool) {
	col.Set(field, value)
}

// SetFloat64 maps the float64 value to the NumberField.
func (col *Column) SetFloat64(field NumberField, value float64) {
	col.Set(field, value)
}

// SetInt maps the int value to the NumberField.
func (col *Column) SetInt(field NumberField, value int) {
	col.Set(field, value)
}

// SetInt64 maps the int64 value to the NumberField.
func (col *Column) SetInt64(field NumberField, value int64) {
	col.Set(field, value)
}

// SetString maps the string value to the StringField.
func (col *Column) SetString(field StringField, value string) {
	col.Set(field, value)
}

// SetTime maps the time.Time value to the TimeField.
func (col *Column) SetTime(field TimeField, value time.Time) {
	col.Set(field, value)
}