Change internal structure in Property object

This commit is contained in:
Fabio Scotto di Santolo
2017-11-14 18:30:23 +01:00
parent ccfe5ab055
commit e82bb45331
3 changed files with 33 additions and 13 deletions

View File

@@ -9,10 +9,10 @@ import (
)
// ParseFunction -- type
type ParseFunction func(string, string) (map[string]string, error)
type ParseFunction func(string, string) (map[int]Pair, error)
// Default parse method for parsing key - value file
func defaultParse(path, fileName string) (m map[string]string, err error) {
func defaultParse(path, fileName string) (m map[int]Pair, err error) {
absolutePathFile, err := filepath.Abs(filepath.Join(path, fileName))
if err != nil {
return nil, err
@@ -23,8 +23,9 @@ func defaultParse(path, fileName string) (m map[string]string, err error) {
}
defer file.Close()
m = make(map[string]string)
m = make(map[int]Pair)
reader := bufio.NewReader(file)
index := 0
for {
line, err := reader.ReadString('\n')
@@ -37,7 +38,7 @@ func defaultParse(path, fileName string) (m map[string]string, err error) {
value = strings.TrimSpace(strings.Replace(line[equal+1:], "\"", "", -1))
}
// assign the values map
m[key] = value
m[index] = Pair{key, value}
}
}
if err == io.EOF {
@@ -46,6 +47,7 @@ func defaultParse(path, fileName string) (m map[string]string, err error) {
if err != nil {
return nil, err
}
index++
}
return m, nil
}

View File

@@ -7,11 +7,16 @@ import (
"strings"
)
type Pair struct {
First string
Second string
}
// Properties -- type
type Properties struct {
fileName string
path string
values map[string]string
values map[int]Pair
length int
}
@@ -20,7 +25,7 @@ func New(path, fileName string) Properties {
return Properties{
fileName: fileName,
path: filepath.Clean(path),
values: make(map[string]string),
values: make(map[int]Pair),
length: 0,
}
}
@@ -51,7 +56,7 @@ func (p Properties) Length() int {
}
// Values -- Getter values of property file
func (p Properties) Values() map[string]string {
func (p Properties) Values() map[int]Pair {
return p.values
}
@@ -61,7 +66,7 @@ func (p *Properties) Put(key, value string) error {
return e.New("Key value is nil")
}
if p.values != nil {
p.values[key] = value
p.values[p.length+1] = Pair{key, value}
p.length++
} else {
return e.New("Property values is nil")
@@ -74,10 +79,11 @@ func (p Properties) Get(key string) (string, error) {
if key == "" || len(strings.TrimSpace(key)) == 0 {
return "", e.New("Key value is nil")
}
if _, ok := p.values[key]; !ok {
index := p.index(key)
if _, ok := p.values[index]; !ok {
return "", e.New("Key not found")
}
return p.values[key], nil
return p.values[index].Second, nil
}
// Remove -- Remove property with key
@@ -86,7 +92,8 @@ func (p *Properties) Remove(key string) (string, error) {
return "", e.New("Key value is nil")
}
lenghtBefore := len(p.values)
delete(p.values, key)
index := p.index(key)
delete(p.values, index)
if len(p.values) != lenghtBefore {
p.length--
}
@@ -95,12 +102,24 @@ func (p *Properties) Remove(key string) (string, error) {
// GetProperties -- Get all key value in Properties object
func (p Properties) GetProperties() (keys []string) {
for key := range p.values {
for i := range p.values {
key := p.values[i].First
keys = append(keys, key)
}
return keys
}
func (p Properties) index(key string) int {
index := -1
for i, item := range p.values {
if item.First == key {
index = i
break
}
}
return index
}
// DefaultLoad -- Load file in Properties object using default parse function
func (p *Properties) DefaultLoad() (int, error) {
return p.Load(defaultParse)

View File

@@ -10,7 +10,6 @@ import (
// StoringFunction -- type
type StoringFunction func(Properties) (*os.File, error)
// FIXME when you remove a property it not work fine
func defaultStore(p Properties) (*os.File, error) {
absolutePathFile, err := filepath.Abs(filepath.Join(p.Path(), p.FileName()))
if err != nil {