tank/rest/config.go

package rest

import (
	"github.com/json-iterator/go"
	"io/ioutil"
	"time"
	"unsafe"
)

const (
	//用户身份的cookie字段名
	COOKIE_AUTH_KEY = "_ak"


	//数据库表前缀 tank200表示当前应用版本是tank:2.0.x版，数据库结构发生变化必然是中型升级
	TABLE_PREFIX = "tank20_"

	//当前版本
	VERSION = "2.0.0"
)

/*
如果你需要在本地127.0.0.1创建默认的数据库和账号，使用以下语句。
create database tank;
grant all privileges on tank.* to tank identified by 'tank123';
flush privileges;
*/
var CONFIG = &Config{}

//依赖外部定义的变量。
type Config struct {
	//默认监听端口号
	ServerPort int
	//网站是否已经完成安装
	Installed bool
	//上传的文件路径，要求不以/结尾。如果没有指定，默认在根目录下的matter文件夹中。eg: /var/www/matter
	MatterPath string
	//数据库连接信息。
	MysqlUrl string
	//配置文件中的项
	Item *ConfigItem
}

//和tank.json文件中的键值一一对应。
type ConfigItem struct {
	//默认监听端口号
	ServerPort int
	//上传的文件路径，要求不以/结尾。如果没有指定，默认在根目录下的matter文件夹中。eg: /var/www/matter
	MatterPath string
	//mysql相关配置。
	//数据库端口
	MysqlPort int
	//数据库Host
	MysqlHost string
	//数据库名字
	MysqlSchema string
	//用户名
	MysqlUsername string
	//密码
	MysqlPassword string
}

//验证配置文件的正确性。
func (this *ConfigItem) validate() bool {

	if this.ServerPort == 0 {
		LOGGER.Error("ServerPort 未配置")
		return false
	} else {
		//只要配置文件中有配置端口，就使用。
		CONFIG.ServerPort = this.ServerPort
	}

	if this.MysqlUsername == "" {
		LOGGER.Error("MysqlUsername 未配置")
		return false
	}

	if this.MysqlPassword == "" {
		LOGGER.Error("MysqlPassword 未配置")
		return false
	}

	if this.MysqlHost == "" {
		LOGGER.Error("MysqlHost 未配置")
		return false
	}

	if this.MysqlPort == 0 {
		LOGGER.Error("MysqlPort 未配置")
		return false
	}

	if this.MysqlSchema == "" {
		LOGGER.Error("MysqlSchema 未配置")
		return false
	}

	return true

}

//验证配置文件是否完好
func (this *Config) Init() {

	//JSON初始化
	jsoniter.RegisterTypeDecoderFunc("time.Time", func(ptr unsafe.Pointer, iter *jsoniter.Iterator) {
		//如果使用time.UTC，那么时间会相差8小时
		t, err := time.ParseInLocation("2006-01-02 15:04:05", iter.ReadString(), time.Local)
		if err != nil {
			iter.Error = err
			return
		}
		*((*time.Time)(ptr)) = t
	})

	jsoniter.RegisterTypeEncoderFunc("time.Time", func(ptr unsafe.Pointer, stream *jsoniter.Stream) {
		t := *((*time.Time)(ptr))
		//如果使用time.UTC，那么时间会相差8小时
		stream.WriteString(t.Local().Format("2006-01-02 15:04:05"))
	}, nil)

	//默认从6010端口启动
	this.ServerPort = 6010

	this.ReadFromConfigFile()

}

//系统如果安装好了就调用这个方法。
func (this *Config) ReadFromConfigFile() {

	//读取配置文件
	filePath := GetConfPath() + "/tank.json"
	content, err := ioutil.ReadFile(filePath)
	if err != nil {
		LOGGER.Warn("无法找到配置文件：%s 即将进入安装过程！", filePath)
		this.Installed = false
	} else {
		this.Item = &ConfigItem{}
		LOGGER.Warn("读取配置文件：%s", filePath)
		err := jsoniter.ConfigCompatibleWithStandardLibrary.Unmarshal(content, this.Item)
		if err != nil {
			LOGGER.Error("配置文件格式错误！ 即将进入安装过程！")
			this.Installed = false
			return
		}

		//验证项是否齐全
		itemValidate := this.Item.validate()
		if !itemValidate {
			LOGGER.Error("配置文件信息不齐全！ 即将进入安装过程！")
			this.Installed = false
			return
		}

		//使用配置项中的文件路径
		if this.Item.MatterPath == "" {
			this.MatterPath = GetHomePath() + "/matter"
		} else {
			this.MatterPath = this.Item.MatterPath
		}
		MakeDirAll(CONFIG.MatterPath)

		//使用配置项中的端口
		if this.Item.ServerPort != 0 {
			this.ServerPort = this.Item.ServerPort
		}

		this.MysqlUrl = GetMysqlUrl(this.Item.MysqlPort, this.Item.MysqlHost, this.Item.MysqlSchema, this.Item.MysqlUsername, this.Item.MysqlPassword)
		this.Installed = true

		LOGGER.Info("使用配置文件：%s", filePath)
		LOGGER.Info("上传文件存放路径：%s", this.MatterPath)
	}
}

//系统如果安装好了就调用这个方法。
func (this *Config) InstallOk() {

	this.ReadFromConfigFile()
}