sqlite3/ext/csv/types.go

package csv

import (
	_ "embed"
	"strings"

	"github.com/ncruces/go-sqlite3/util/vtabutil"
)

type affinity byte

const (
	blob    affinity = 0
	text    affinity = 1
	numeric affinity = 2
	integer affinity = 3
	real    affinity = 4
)

func getColumnAffinities(schema string) []affinity {
	tab, err := vtabutil.Parse(schema)
	if err != nil {
		return nil
	}
	defer tab.Close()

	types := make([]affinity, tab.NumColumns())
	for i := range types {
		col := tab.Column(i)
		types[i] = getAffinity(col.Type())
	}
	return types
}

func getAffinity(declType string) affinity {
	// https://sqlite.org/datatype3.html#determination_of_column_affinity
	if declType == "" {
		return blob
	}
	name := strings.ToUpper(declType)
	if strings.Contains(name, "INT") {
		return integer
	}
	if strings.Contains(name, "CHAR") || strings.Contains(name, "CLOB") || strings.Contains(name, "TEXT") {
		return text
	}
	if strings.Contains(name, "BLOB") {
		return blob
	}
	if strings.Contains(name, "REAL") || strings.Contains(name, "FLOA") || strings.Contains(name, "DOUB") {
		return real
	}
	return numeric
}