query_mappers.go

package proteus

import (
	"os"

	"github.com/rickar/props"
)

type MapMapper map[string]string

func (mm MapMapper) Map(name string) string {
	return mm[name]
}

type propFileMapper struct {
	properties *props.Properties
}

func (pm propFileMapper) Map(name string) string {
	val, _ := pm.properties.Get(name)
	return val
}

func PropFileToQueryMapper(name string) (QueryMapper, error) {
	file, err := os.Open(name)
	if err != nil {
		return nil, err
	}
	defer file.Close()
	properties, err := props.Read(file)
	if err != nil {
		return nil, err
	}
	return propFileMapper{properties}, nil
}