go-fly/config/config.go

package config

import (
	"encoding/json"
	"fmt"
	"github.com/taoshihan1991/imaptool/tools"
	"io/ioutil"
	"os"
)

const Dir = "config/"
const AccountConf = Dir + "account.json"
const MysqlConf = Dir + "mysql.json"
func GetMysql() map[string]string {
	var mysql map[string]string
	isExist, _ := tools.IsFileExist(MysqlConf)
	if !isExist {
		return mysql
	}
	info, err := ioutil.ReadFile(MysqlConf)
	if err != nil {
		return mysql
	}

	err = json.Unmarshal(info, &mysql)
	return mysql
}
func GetAccount() map[string]string {
	var account map[string]string
	isExist, _ := tools.IsFileExist(AccountConf)
	if !isExist {
		return account
	}
	info, err := ioutil.ReadFile(AccountConf)
	if err != nil {
		return account
	}

	err = json.Unmarshal(info, &account)
	return account
}
func GetUserInfo(uid string) map[string]string {
	var userInfo map[string]string
	userFile := Dir + "sess_" + uid + ".json"
	isExist, _ := tools.IsFileExist(userFile)
	if !isExist {
		return userInfo
	}
	info, err := ioutil.ReadFile(userFile)
	if err != nil {
		return userInfo
	}

	err = json.Unmarshal(info, &userInfo)
	return userInfo
}
func SetUserInfo(uid string, info map[string]string) {
	userFile := Dir + "sess_" + uid + ".json"
	isExist, _ := tools.IsFileExist(Dir)
	if !isExist {
		os.Mkdir(Dir, os.ModePerm)
	}
	file, _ := os.OpenFile(userFile, os.O_RDWR|os.O_CREATE, os.ModePerm)
	str := "{\r\n"
	for k, v := range info {
		str += fmt.Sprintf(`"%s":"%s",`, k, v)
	}
	str += fmt.Sprintf(`"session_id":"%s"%s}`, uid, "\r\n")
	file.WriteString(str)
}