gh-ost/go/sql/types.go

/*
   Copyright 2016 GitHub Inc.
	 See https://github.com/github/gh-osc/blob/master/LICENSE
*/

package sql

import (
	"fmt"
	"reflect"
	"strings"
)

// ColumnList makes for a named list of columns
type ColumnList []string

// ParseColumnList parses a comma delimited list of column names
func ParseColumnList(columns string) *ColumnList {
	result := ColumnList(strings.Split(columns, ","))
	return &result
}

func (this *ColumnList) String() string {
	return strings.Join(*this, ",")
}

func (this *ColumnList) Equals(other *ColumnList) bool {
	return reflect.DeepEqual(*this, *other)
}

// UniqueKey is the combination of a key's name and columns
type UniqueKey struct {
	Name        string
	Columns     ColumnList
	HasNullable bool
}

// IsPrimary cehcks if this unique key is primary
func (this *UniqueKey) IsPrimary() bool {
	return this.Name == "PRIMARY"
}

func (this *UniqueKey) String() string {
	return fmt.Sprintf("%s: %s; has nullable: %+v", this.Name, this.Columns, this.HasNullable)
}

type ColumnValues struct {
	abstractValues []interface{}
	ValuesPointers []interface{}
}

func NewColumnValues(length int) *ColumnValues {
	result := &ColumnValues{
		abstractValues: make([]interface{}, length),
		ValuesPointers: make([]interface{}, length),
	}
	for i := 0; i < length; i++ {
		result.ValuesPointers[i] = &result.abstractValues[i]
	}

	return result
}

func (this *ColumnValues) AbstractValues() []interface{} {
	return this.abstractValues
}

func (this *ColumnValues) BinaryValues() [][]uint8 {
	result := make([][]uint8, len(this.abstractValues), len(this.abstractValues))
	for i, val := range this.abstractValues {
		result[i] = val.([]uint8)
	}
	return result
}

func (this *ColumnValues) String() string {
	stringValues := []string{}
	for _, val := range this.BinaryValues() {
		stringValues = append(stringValues, string(val))
	}
	return strings.Join(stringValues, ",")
}