Compare commits
10 Commits
Author | SHA1 | Date | |
---|---|---|---|
2efc1a155e
|
|||
45f8b0f918
|
|||
2cdcbec364
|
|||
f815bff582
|
|||
73278439b0
|
|||
0a589a2565
|
|||
179666c8f9
|
|||
02d36e3d69
|
|||
9733bb42a8
|
|||
3d2005a8a2
|
2
.gitignore
vendored
Normal file
2
.gitignore
vendored
Normal file
@ -0,0 +1,2 @@
|
|||||||
|
tests/buildenv/*
|
||||||
|
tests/bin/*
|
@ -1,3 +1,6 @@
|
|||||||
#!/usr/bin/env bash
|
#!/usr/bin/env bash
|
||||||
go fmt *.go
|
go fmt *.go
|
||||||
git add *.go
|
go fmt tests/*.go
|
||||||
|
|
||||||
|
git add *.go tests/*.go
|
||||||
|
git add tests/*.go
|
||||||
|
48
datatype.go
48
datatype.go
@ -1,22 +1,40 @@
|
|||||||
package envconf
|
package envconf
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"fmt"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
type DataType int
|
type DataType uint32
|
||||||
|
|
||||||
const (
|
const (
|
||||||
TypeNone DataType = iota
|
sizeBitmask DataType = 0xffff
|
||||||
TypeInt DataType = iota
|
typeBitmask DataType = sizeBitmask << 16
|
||||||
TypeMetric DataType = iota
|
|
||||||
TypeDuration DataType = iota
|
|
||||||
TypeString DataType = iota
|
|
||||||
TypeDirectory DataType = iota
|
|
||||||
TypeBool DataType = iota
|
|
||||||
TypeHex DataType = iota
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
TypeNone DataType = iota << 16
|
||||||
|
TypeInt DataType = iota << 16
|
||||||
|
TypeMetric DataType = iota << 16
|
||||||
|
TypeDuration DataType = iota << 16
|
||||||
|
TypeString DataType = iota << 16
|
||||||
|
TypeDirectory DataType = iota << 16
|
||||||
|
TypeBool DataType = iota << 16
|
||||||
|
TypeHex DataType = iota << 16
|
||||||
|
)
|
||||||
|
|
||||||
|
func FixedHex(size uint16) DataType {
|
||||||
|
return TypeHex | DataType(size)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dtype DataType) baseType() DataType {
|
||||||
|
return dtype & typeBitmask
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dtype DataType) typeAndSize() (DataType, int) {
|
||||||
|
return (dtype & typeBitmask), int(dtype & sizeBitmask)
|
||||||
|
}
|
||||||
|
|
||||||
type cValue struct {
|
type cValue struct {
|
||||||
dtype DataType
|
dtype DataType
|
||||||
intval int64
|
intval int64
|
||||||
@ -28,18 +46,24 @@ type cValue struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (dtype DataType) parse(key string, str string) (ret cValue) {
|
func (dtype DataType) parse(key string, str string) (ret cValue) {
|
||||||
info, ok := tInfo[dtype]
|
rdtype, size := dtype.typeAndSize()
|
||||||
|
info, ok := tInfo[rdtype]
|
||||||
if ok {
|
if ok {
|
||||||
ret = info.parser(key, str)
|
ret = info.parser(key, str, size)
|
||||||
if len(ret.binval) == 0 {
|
if len(ret.binval) == 0 {
|
||||||
ret.binval = make([]byte, 0, 0)
|
ret.binval = make([]byte, 0, 0)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func (dtype DataType) String() string {
|
func (dtype DataType) String() string {
|
||||||
info, ok := tInfo[dtype]
|
rdtype, size := dtype.typeAndSize()
|
||||||
|
info, ok := tInfo[rdtype]
|
||||||
if ok {
|
if ok {
|
||||||
|
if size > 0 {
|
||||||
|
return fmt.Sprintf("%s%d", info.name, size)
|
||||||
|
}
|
||||||
return info.name
|
return info.name
|
||||||
}
|
}
|
||||||
return "invalid"
|
return "invalid"
|
||||||
|
57
envconf.go
57
envconf.go
@ -23,7 +23,7 @@ type cEntry struct {
|
|||||||
type Config struct {
|
type Config struct {
|
||||||
parsed bool
|
parsed bool
|
||||||
env map[string]cEntry
|
env map[string]cEntry
|
||||||
mapEnv map[string]map[string]cEntry
|
mapEnv map[string]map[string]string
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewConfig returns an envconf.Config that is used to read configuration from environment variables.
|
// NewConfig returns an envconf.Config that is used to read configuration from environment variables.
|
||||||
@ -33,27 +33,31 @@ func NewConfig() *Config {
|
|||||||
config := new(Config)
|
config := new(Config)
|
||||||
config.parsed = false
|
config.parsed = false
|
||||||
config.env = make(map[string]cEntry)
|
config.env = make(map[string]cEntry)
|
||||||
config.mapEnv = make(map[string]map[string]cEntry)
|
config.mapEnv = make(map[string]map[string]string)
|
||||||
|
|
||||||
for _, v := range os.Environ() {
|
for _, v := range os.Environ() {
|
||||||
splitted := strings.SplitN(v, "=", 2)
|
splitted := strings.SplitN(v, "=", 2)
|
||||||
if len(splitted) == 2 {
|
if len(splitted) == 2 {
|
||||||
key := cleanKey(splitted[0])
|
key := cleanKey(splitted[0])
|
||||||
left, right, mappable := keySplit(key)
|
val := splitted[1]
|
||||||
|
splitted = strings.Split(key, "_")
|
||||||
if unicode.IsLetter(getFirstRune(key)) {
|
if unicode.IsLetter(getFirstRune(key)) {
|
||||||
var entry cEntry
|
var entry cEntry
|
||||||
entry.value = splitted[1]
|
entry.value = val
|
||||||
entry.dtype = TypeNone
|
entry.dtype = TypeNone
|
||||||
entry.unset = false
|
entry.unset = false
|
||||||
entry.empty = false
|
entry.empty = false
|
||||||
config.env[key] = entry
|
config.env[key] = entry
|
||||||
if mappable {
|
if len(splitted) > 1 {
|
||||||
if len(config.mapEnv[left]) == 0 {
|
for count, _ := range splitted {
|
||||||
config.mapEnv[left] = make(map[string]cEntry)
|
left := strings.Join(splitted[:count], "_")
|
||||||
|
right := strings.Join(splitted[count:], "_")
|
||||||
|
if len(config.mapEnv[left]) == 0 {
|
||||||
|
config.mapEnv[left] = make(map[string]string)
|
||||||
|
}
|
||||||
|
fmt.Printf("left=(%s), right=(%s), key=(%s)\n", left, right, key)
|
||||||
|
config.mapEnv[left][right] = key
|
||||||
}
|
}
|
||||||
|
|
||||||
config.mapEnv[left][right] = entry
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -82,11 +86,14 @@ func (c *Config) Define(key string, dtype DataType) {
|
|||||||
// Variables without a defined type will be ignored by Parse.
|
// Variables without a defined type will be ignored by Parse.
|
||||||
func (c *Config) DefineMap(key string, dtype DataType) {
|
func (c *Config) DefineMap(key string, dtype DataType) {
|
||||||
key = cleanKey(key)
|
key = cleanKey(key)
|
||||||
|
fmt.Println(c.mapEnv)
|
||||||
entries, ok := c.mapEnv[key]
|
entries, ok := c.mapEnv[key]
|
||||||
if ok {
|
if ok {
|
||||||
for mapKey, entry := range entries {
|
for _, key = range entries {
|
||||||
|
fmt.Printf("key-from-define-map (%s)\n", key)
|
||||||
|
entry := c.env[key]
|
||||||
entry.dtype = dtype
|
entry.dtype = dtype
|
||||||
c.mapEnv[key][mapKey] = entry
|
c.env[key] = entry
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -136,13 +143,6 @@ func (c *Config) Parse() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
for k, v := range c.mapEnv {
|
|
||||||
for mk, mv := range v {
|
|
||||||
mv.parsed = mv.dtype.parse(k+"_"+mk, mv.value)
|
|
||||||
c.mapEnv[k][mk] = mv
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if failed {
|
if failed {
|
||||||
for k, v := range c.env {
|
for k, v := range c.env {
|
||||||
if (v.parsed.err == nil) && v.unset {
|
if (v.parsed.err == nil) && v.unset {
|
||||||
@ -234,17 +234,7 @@ func (c *Config) Status() (ok bool) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
for _, v := range c.mapEnv {
|
|
||||||
for _, mv := range v {
|
|
||||||
err := mv.parsed.err
|
|
||||||
if err != nil {
|
|
||||||
ok = false
|
|
||||||
if !mv.empty {
|
|
||||||
fmt.Fprintln(os.Stderr, err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if !ok {
|
if !ok {
|
||||||
fmt.Fprintln(os.Stderr, "")
|
fmt.Fprintln(os.Stderr, "")
|
||||||
for _, v := range c.env {
|
for _, v := range c.env {
|
||||||
@ -287,7 +277,7 @@ func (c *Config) getRaw(key string, dtype DataType) (val cValue) {
|
|||||||
if c.parsed {
|
if c.parsed {
|
||||||
key = cleanKey(key)
|
key = cleanKey(key)
|
||||||
entry, ok := c.env[key]
|
entry, ok := c.env[key]
|
||||||
if ok && (entry.dtype == dtype) {
|
if ok && (entry.dtype.baseType() == dtype.baseType()) {
|
||||||
return entry.parsed
|
return entry.parsed
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -303,8 +293,9 @@ func (c *Config) getRawMap(key string, dtype DataType) (empty map[string]cValue)
|
|||||||
|
|
||||||
if ok {
|
if ok {
|
||||||
for k, v := range entries {
|
for k, v := range entries {
|
||||||
if (v.dtype == dtype) && (v.parsed.err == nil) {
|
entry := c.env[v]
|
||||||
retval[k] = v.parsed
|
if (entry.dtype.baseType() == dtype.baseType()) && (entry.parsed.err == nil) {
|
||||||
|
retval[k] = entry.parsed
|
||||||
} else {
|
} else {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
22
parsers.go
22
parsers.go
@ -11,7 +11,7 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
func parseInt(key string, str string) (ret cValue) {
|
func parseInt(key string, str string, _ int) (ret cValue) {
|
||||||
val, err := strconv.ParseInt(str, 10, 64)
|
val, err := strconv.ParseInt(str, 10, 64)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
ret.intval = val
|
ret.intval = val
|
||||||
@ -21,7 +21,7 @@ func parseInt(key string, str string) (ret cValue) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func parseMetric(key string, str string) (ret cValue) {
|
func parseMetric(key string, str string, _ int) (ret cValue) {
|
||||||
mod := int64(1)
|
mod := int64(1)
|
||||||
str = strings.ToUpper(str)
|
str = strings.ToUpper(str)
|
||||||
if strings.HasSuffix(str, "K") {
|
if strings.HasSuffix(str, "K") {
|
||||||
@ -46,7 +46,7 @@ func parseMetric(key string, str string) (ret cValue) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func parseDuration(key string, str string) (ret cValue) {
|
func parseDuration(key string, str string, _ int) (ret cValue) {
|
||||||
val, err := time.ParseDuration(str)
|
val, err := time.ParseDuration(str)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
ret.durval = val
|
ret.durval = val
|
||||||
@ -56,7 +56,7 @@ func parseDuration(key string, str string) (ret cValue) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func parseBool(key string, str string) (ret cValue) {
|
func parseBool(key string, str string, _ int) (ret cValue) {
|
||||||
val, err := strconv.ParseBool(str)
|
val, err := strconv.ParseBool(str)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
ret.boolval = val
|
ret.boolval = val
|
||||||
@ -66,17 +66,21 @@ func parseBool(key string, str string) (ret cValue) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func parseHex(key string, str string) (ret cValue) {
|
func parseHex(key string, str string, size int) (ret cValue) {
|
||||||
val, err := hex.DecodeString(str)
|
val, err := hex.DecodeString(str)
|
||||||
if err == nil {
|
if err == nil && (size == 0 || size == len(val)) {
|
||||||
ret.binval = val
|
ret.binval = val
|
||||||
} else {
|
} else {
|
||||||
ret.err = errors.New(fmt.Sprintf(`Environment variable "%s" is not of type hex.`, key))
|
if size == 0 {
|
||||||
|
ret.err = errors.New(fmt.Sprintf(`Environment variable "%s" is not of type hex.`, key))
|
||||||
|
} else {
|
||||||
|
ret.err = errors.New(fmt.Sprintf(`Environment variable "%s" is not of type hex%d.`, key, size))
|
||||||
|
}
|
||||||
}
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func parseDirectory(_ string, str string) (ret cValue) {
|
func parseDirectory(_ string, str string, _ int) (ret cValue) {
|
||||||
wd, err := os.Getwd()
|
wd, err := os.Getwd()
|
||||||
if err == nil {
|
if err == nil {
|
||||||
if path.IsAbs(str) {
|
if path.IsAbs(str) {
|
||||||
@ -90,7 +94,7 @@ func parseDirectory(_ string, str string) (ret cValue) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func parseString(_ string, str string) (ret cValue) {
|
func parseString(_ string, str string, _ int) (ret cValue) {
|
||||||
ret.strval = str
|
ret.strval = str
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
22
tests/build.sh
Executable file
22
tests/build.sh
Executable file
@ -0,0 +1,22 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
dir=$(dirname $(realpath $0))
|
||||||
|
export GO111MODULE=on
|
||||||
|
export GOPROXY=direct
|
||||||
|
export GOPATH=${dir}/buildenv/deps
|
||||||
|
|
||||||
|
mkdir -p ${dir}/buildenv
|
||||||
|
chmod -R 755 ${dir}/buildenv
|
||||||
|
rm -rf ${dir}/buildenv
|
||||||
|
|
||||||
|
mkdir -p "${dir}/buildenv/src"
|
||||||
|
mkdir -p ${GOPATH}/src
|
||||||
|
mkdir -p ${dir}/bin
|
||||||
|
|
||||||
|
cp -r "${dir}/main.go" "${dir}/buildenv/src/main.go"
|
||||||
|
|
||||||
|
cd ${dir}/buildenv
|
||||||
|
|
||||||
|
go mod init src
|
||||||
|
go mod tidy
|
||||||
|
|
||||||
|
go build -o "${dir}/bin/test" "src/main.go"
|
17
tests/main.go
Normal file
17
tests/main.go
Normal file
@ -0,0 +1,17 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"git.purser.it/roypur/envconf"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
conf := envconf.NewConfig()
|
||||||
|
conf.Define("this_is_a_map", envconf.TypeInt)
|
||||||
|
conf.Define("test_map", envconf.TypeInt)
|
||||||
|
conf.Parse()
|
||||||
|
conf.Status()
|
||||||
|
|
||||||
|
fmt.Println(conf.GetMapInt("test_map"))
|
||||||
|
fmt.Println(conf.GetInt("this_is_a_map"))
|
||||||
|
}
|
@ -1,7 +1,7 @@
|
|||||||
package envconf
|
package envconf
|
||||||
|
|
||||||
type dataTypeInfo struct {
|
type dataTypeInfo struct {
|
||||||
parser func(string, string) cValue
|
parser func(string, string, int) cValue
|
||||||
name string
|
name string
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -15,6 +15,7 @@ func init() {
|
|||||||
var strInfo dataTypeInfo
|
var strInfo dataTypeInfo
|
||||||
var dirInfo dataTypeInfo
|
var dirInfo dataTypeInfo
|
||||||
var hexInfo dataTypeInfo
|
var hexInfo dataTypeInfo
|
||||||
|
|
||||||
var boolInfo dataTypeInfo
|
var boolInfo dataTypeInfo
|
||||||
|
|
||||||
intInfo.name = "int"
|
intInfo.name = "int"
|
||||||
@ -30,6 +31,7 @@ func init() {
|
|||||||
durInfo.parser = parseDuration
|
durInfo.parser = parseDuration
|
||||||
dirInfo.parser = parseDirectory
|
dirInfo.parser = parseDirectory
|
||||||
strInfo.parser = parseString
|
strInfo.parser = parseString
|
||||||
|
|
||||||
hexInfo.parser = parseHex
|
hexInfo.parser = parseHex
|
||||||
boolInfo.parser = parseBool
|
boolInfo.parser = parseBool
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user