【代码篇】从零开始一步步搭建自己的golang框架(四)

wx5e1abbbb0a5e5 · · 178 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

上篇文章讲到配置和日志已经初始化了,这篇文章接着把数据库和redis初始化完成。

初始化数据库

数据库orm我选用xorm。首先在config.json和config.go添加对应的数据库配置。
config.json:

  "db_config": {
    "db_host": "127.0.0.1",
    "db_port": "3306",
    "db_user": "root",
    "db_password": "123456",
    "db_name": "test"
  }

config.go

type DBConfig struct {
    DbHost     string `json:"db_host"`
    DbPort     string `json:"db_port"`
    DbUser     string `json:"db_user"`
    DbPassword string `json:"db_password"`
    DbName     string `json:"db_name"`
}

接着,初始化数据库:

package db

import (
    "github.com/TomatoMr/awesomeframework/config"
    _ "github.com/go-sql-driver/mysql"
    "github.com/go-xorm/xorm"
    "github.com/pkg/errors"
)

var engine *xorm.Engine

func InitEngine() error {
    var err error
    conf := config.GetConfig()
    engine, err = xorm.NewEngine("mysql", conf.DBConfig.DbUser+
        ":"+conf.DBConfig.DbPassword+"@tcp("+conf.DBConfig.DbHost+":"+conf.DBConfig.DbPort+")/"+conf.DBConfig.DbName+"?charset=utf8")
    if err != nil {
        err = errors.Wrap(err, "InitEngine1")
        return err
    }
    err = engine.Ping()
    if err != nil {
        err = errors.Wrap(err, "InitEngine2")
        return err
    }
    return nil
}

func GetEngine() *xorm.Engine {
    return engine
}

我们再创建一个sql,等会测试连接用:

DROP DATABASE IF EXISTS `test`;
CREATE DATABASE `test`;
USE `test`;
DROP TABLE IF EXISTS `users`;
CREATE TABLE `users`
(
    `id`   int(11)     NOT NULL AUTO_INCREMENT COMMENT 'id',
    `name` varchar(40) NOT NULL COMMENT '名字',
    `age`  int         NOT NULL DEFAULT 0 COMMENT '年龄',
    PRIMARY KEY (`id`)
) ENGINE = InnoDB COMMENT 'users';

初始化redis

同样先添加配置:
config.json:

"redis_config": {
    "addr": "127.0.0.1:6379",
    "password": "",
    "db": 0
  }

config.go:

type RedisConfig struct {
    Addr     string `json:"addr"`
    Password string `json:"password"`
    DB       int    `json:"db"`
}

初始化redis:

func InitRedis() error {
    conf := config.GetConfig()
    client = redis.NewClient(&redis.Options{
        Addr:     conf.RedisConfig.Addr,
        Password: conf.RedisConfig.Password,
        DB:       conf.RedisConfig.DB,
    })

    pong, err := client.Ping().Result()
    if err != nil {
        err = errors.Wrap(err, "InitRedis")
        return err
    }
    logger.GetLogger().Info("Redis ping:", zap.String("ping", pong))
    return nil
}

调整入口文件

err = db.InitEngine()
    if err != nil {
        fmt.Printf("Init DB failed. Error is %v", err)
        os.Exit(1)
    }

    err = redis.InitRedis()
    if err != nil {
        fmt.Printf("Init Redis failed. Error is %v", err)
        os.Exit(1)
    }

测试一下

使用上面的sql,创建一个数据库叫做test,然后搭建redis,接着我们测试一下:

编译:

go build

运行:

awesomeframework --config=./config/config.json

看下日志打印:

2020-01-20T20:09:26.798+0800    info    Redis ping: {"ping": "PONG"}
2020-01-20T20:09:26.798+0800    info    Init success.

小结

这篇文章写了数据库和redis的初始化,可以在https://github.com/TomatoMr/awesomeframework找到今天的代码。未完待续……


欢迎关注我的公众号:onepunchgo,给我留言。

image


有疑问加站长微信联系(非本文作者)

本文来自:51CTO博客

感谢作者:wx5e1abbbb0a5e5

查看原文:【代码篇】从零开始一步步搭建自己的golang框架(四)

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

178 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传