column_map.go

// Copyright 2019 The Xorm Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package statements

import (
	"strings"

	"github.com/fairyhunter13/xorm/schemas"
)

type columnMap []string

func (m columnMap) Contain(colName string) bool {
	if len(m) == 0 {
		return false
	}

	n := len(colName)
	for _, mk := range m {
		if len(mk) != n {
			continue
		}
		if strings.EqualFold(mk, colName) {
			return true
		}
	}

	return false
}

func (m columnMap) Len() int {
	return len(m)
}

func (m columnMap) IsEmpty() bool {
	return len(m) == 0
}

func (m *columnMap) Add(colName string) bool {
	if m.Contain(colName) {
		return false
	}
	*m = append(*m, colName)
	return true
}

func getFlagForColumn(m map[string]bool, col *schemas.Column) (val bool, has bool) {
	if len(m) == 0 {
		return false, false
	}

	n := len(col.Name)

	for mk := range m {
		if len(mk) != n {
			continue
		}
		if strings.EqualFold(mk, col.Name) {
			return m[mk], true
		}
	}

	return false, false
}