上一篇文章里，我们定了项目基本目录结构。现在来回顾一下我的思路：
- 创建入口文件；
- 指定配置、日志目录；
- 指定数据库、队列、缓存组件目录；
- 创建数据处理的controller目录；
- 指定公共函数等杂项。
接下来，我们还是按照这个顺序填充代码。
入口文件
func main() {
config.InitConfig()
logger.InitLogger()
}
暂且忽略一些实现细节，我们需要先初始化配置和日志，先来看看日志，我选用Uber的zap日志库。来看看日志模块怎么写？
日志
package logger
import (
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
"gopkg.in/natefinch/lumberjack.v2"
)
var l *zap.Logger
func InitLogger(logPath, logLevel string) error {
hook := lumberjack.Logger{
Filename: logPath,
MaxSize: 1024,
MaxBackups: 3,
MaxAge: 7,
Compress: true,
}
w := zapcore.AddSync(&hook)
var level zapcore.Level
switch logLevel {
case "debug":
level = zap.DebugLevel
case "info":
level = zap.InfoLevel
case "error":
level = zap.ErrorLevel
default:
level = zap.DebugLevel
}
encoderConfig := zap.NewProductionEncoderConfig()
encoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder
core := zapcore.NewCore(
zapcore.NewConsoleEncoder(encoderConfig),
w,
level,
)
l = zap.New(core)
return nil
}
func GetLogger() *zap.Logger {
return l
}
我们先定义一个包级的全局变量l，类型是*zap.Logger，并创建了InitLogger和GetLogger这两个函数。因为，zap不支持日志归档，所以在InitLogger中定义了一个lumberjack的hook，用来归档日志。我们可以看到InitLogger这个方法有两个入参：logPath和logLevel。一般来讲，这些参数应该是放在配置文件里的，接下来我们来写配置。
配置
我们简单地创建一个config.json文件来存放配置：
config.json:
{
"log_config": {
"log_path": "{your_path}",
"log_level": "debug"
}
}
然后在config.go中创建相应的结构体，之后定义InitConfig和GetConfig方法，在InitConfig方法中，我读取了配置文件的内容，然后解析到结构体中，并处理了错误，如果有错误信息的话，我会将错误信息包裹一层，方便以后的错误定位。InitConfig有一个入参，就是配置文件的路径，这个参数我会从命令行中获取。
config.go:
package config
import (
"encoding/json"
"github.com/pkg/errors"
"io/ioutil"
)
type LogConfig struct {
LogPath string `json:"log_path"`
LogLevel string `json:"log_level"`
}
type Config struct {
LogConfig LogConfig `json:"log_config"`
}
var conf Config
func InitConfig(configPath string) error {
configFile, err := ioutil.ReadFile(configPath)
if err != nil {
err = errors.Wrap(err, "Read config file failed.")
return err
}
err = json.Unmarshal(configFile, &conf)
if err != nil {
err = errors.Wrap(err, "Unmarshal config file failed.")
return err
}
return nil
}
func GetConfig() Config {
return conf
}
我们当然不止这么一点配置，我们还有数据库、缓存等配置没有添加，但不急，我们先理出一个路线，之后的配置可以依葫芦画瓢。
调整入口文件
好，日志和配置的初始化大致写好了，我们回过头来看一下入口文件的调整：
package main
import (
"flag"
"fmt"
"github.com/TomatoMr/awesomeframework/config"
"github.com/TomatoMr/awesomeframework/logger"
"os"
)
func main() {
var configPath string
flag.StringVar(&configPath, "config", "", "配置文件路径")
flag.Parse()
if configPath == "" {
fmt.Printf("Config Path must be assigned.")
os.Exit(1)
}
var err error
err = config.InitConfig(configPath)
if err != nil {
fmt.Printf("Init config failed. Error is %v", err)
os.Exit(1)
}
logConfig := config.GetConfig().LogConfig
err = logger.InitLogger(logConfig.LogPath, logConfig.LogLevel)
if err != nil {
fmt.Printf("Init logger failed. Error is %v", err)
os.Exit(1)
}
logger.GetLogger().Info("Init success.")
}
我们调整了入口文件，从命令行里获取配置文件路径，之后初始化了配置和日志，最后打印初始化结果。
测试一下
首先，编译一下：
$ go build
接着修改一下你的config.json文件的log_path，然后在命令行指定一下你的配置文件路径：
$ awesomeframework --config={$your_path}/config.json
最后，来看一下运行正不正常，日志文件的打印如下：
2020-01-19T20:41:57.506+0800 info Init success.
小结
到目前为止，我们初始化了配置和日志，接下去要开始初始化数据库等组件。可以在https://github.com/TomatoMr/awesomeframework
找到今天的代码。未完待续……
欢迎关注我的公众号：onepunchgo。
有疑问加站长微信联系（非本文作者）
