104 lines
1.9 KiB
Go
104 lines
1.9 KiB
Go
package ini_config
|
||
|
||
import (
|
||
"bufio"
|
||
"errors"
|
||
"fmt"
|
||
"io"
|
||
"os"
|
||
"strings"
|
||
)
|
||
|
||
// ParseFile
|
||
// @description: 从文件读取配置文件，并转化为map对象
|
||
// parameter:
|
||
// @filePath:文件路径
|
||
// return:
|
||
// @map[string]string: 配置信息
|
||
// @error:错误信息
|
||
func ParseFile(filePath string) (map[string]string, error) {
|
||
fn, err := os.Open(filePath)
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
defer fn.Close()
|
||
|
||
kvMap := make(map[string]string)
|
||
rd := bufio.NewReader(fn)
|
||
for {
|
||
data, _, err := rd.ReadLine()
|
||
if err != nil || io.EOF == err {
|
||
break
|
||
}
|
||
|
||
line := strings.TrimSpace(string(data))
|
||
if line == "" || line[0:1] == "#" {
|
||
continue
|
||
}
|
||
|
||
var k string
|
||
var v string
|
||
k, v, err = Parse(line)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
kvMap[k] = v
|
||
}
|
||
|
||
return kvMap, nil
|
||
}
|
||
|
||
// Parse
|
||
// @description: 转换单行字符串为配置对象
|
||
// parameter:
|
||
// @line:单行内容
|
||
// return:
|
||
// @string:key
|
||
// @string:value
|
||
// @error:错误信息
|
||
func Parse(line string) (string, string, error) {
|
||
ls := strings.SplitN(line, "=", 2)
|
||
if len(ls) != 2 {
|
||
return "", "", errors.New(fmt.Sprintf("配置%s中未找到=分隔符", line))
|
||
}
|
||
|
||
var key = strings.TrimSpace(ls[0])
|
||
var value string
|
||
|
||
i := strings.LastIndex(ls[1], "#")
|
||
if i < 0 {
|
||
value = strings.TrimSpace(ls[1])
|
||
} else {
|
||
s := ls[1][:i]
|
||
value = strings.TrimSpace(s)
|
||
}
|
||
|
||
return key, value, nil
|
||
}
|
||
|
||
// ParseMultipleLines
|
||
// @description: 转换多行字符串为配置对象
|
||
// parameter:
|
||
// @content:多行内容(通过\n换行)
|
||
// return:
|
||
// @map[string]string:配置信息
|
||
// @error:错误信息
|
||
func ParseMultipleLines(content string) (map[string]string, error) {
|
||
lines := strings.Split(content, "\n")
|
||
kvMap := make(map[string]string)
|
||
for _, line := range lines {
|
||
if line == "" || line[0:1] == "#" {
|
||
continue
|
||
}
|
||
|
||
k, v, err := Parse(line)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
kvMap[k] = v
|
||
}
|
||
|
||
return kvMap, nil
|
||
}
|