table.go

package exceltesting

import (
	"fmt"
	"strings"

	"golang.org/x/exp/slices"
)

// functionNames はDBMSの関数名の一覧です
var functionNames = []string{
	"current_timestamp",
}

// table は投入対象のテーブルです
type table struct {
	name    string
	columns []string
	data    [][]string
}

// buildSQL はINSERTステートメントを作成します
func (t *table) buildInsertSQL() string {
	var valueSQLExp string
	for j, row := range t.data {
		rowSQLExp := "("
		for i, cell := range row {
			if i >= 1 {
				rowSQLExp = fmt.Sprintf("%s, ", rowSQLExp)
			}
			if cell == "" {
				rowSQLExp += "null"
			} else if slices.Contains(functionNames, cell) {
				rowSQLExp += cell
			} else {
				rowSQLExp += fmt.Sprintf("'%s'", cell)
			}
		}
		rowSQLExp += ")"
		if j == 0 {
			valueSQLExp = rowSQLExp
			continue
		}
		valueSQLExp = valueSQLExp + "," + rowSQLExp
	}

	sql := fmt.Sprintf("INSERT INTO %s (%s) VALUES%s;\n", t.name, t.sqlColumnExp(), valueSQLExp)
	return sql
}

func (t *table) sqlColumnExp() string {
	return strings.Join(t.columns, ",")
}

// merge は引数のカラムと値を t にマージします
// カラムが t に存在しない場合のみマージし、t にも引数の columns にも存在する場合は t を優先してマージしません
func (t *table) merge(columns []dbColumn) {
	tmp := t.DeepCopy()
	for _, dc := range columns {
		if slices.Contains(tmp.columns, dc.name) {
			continue
		}
		t.columns = append(t.columns, dc.name)
		for i := range t.data {
			t.data[i] = append(t.data[i], dc.data)
		}
	}
}

// DeepCopy generates a deep copy of table
func (t table) DeepCopy() table {
	var cp table = t
	if t.columns != nil {
		cp.columns = make([]string, len(t.columns))
		copy(cp.columns, t.columns)
	}
	if t.data != nil {
		cp.data = make([][]string, len(t.data))
		copy(cp.data, t.data)
		for i2 := range t.data {
			if t.data[i2] != nil {
				cp.data[i2] = make([]string, len(t.data[i2]))
				copy(cp.data[i2], t.data[i2])
			}
		}
	}
	return cp
}