《从0到1搭建一个IM项目》项目初始化

/Users/feng/go/srcCopy

go mod init HiChatCopy

HiChat   
    ├── common    //放置公共文件
    │  
    ├── config    //做配置文件
    │  
    ├── dao       //数据库crud
    │  
    ├── global    //放置各种连接池，配置等
    │   
    ├── initialize  //项目初始化文件
    │  
    ├── middlewear  //放置web中间件
    │ 
    ├── models      //数据库表设计
    │   
    ├── router           //路由
    │   
    ├── service     //对外api
    │   
    ├── test        //测试文件
    │  
    ├── main.go     //项目入口
    ├── go.mod            //项目依赖管理
    ├── go.sum            //项目依赖管理Copy

package global

import (
    "gorm.io/gorm"
)

var (
    DB            *gorm.DB
)Copy

package initialize

import (
    "fmt"
    "log"
    "os"
    "time"

  "HiChat/global"

    "gorm.io/gorm/logger"
    "gorm.io/driver/mysql"
    "gorm.io/gorm"
)

func InitDB() {
    dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local", User,
        Password, Host, Port, DBName)
    //注意：User和Password为MySQL数据库的管理员密码，Host和Port为数据库连接ip端口，DBname为要连接的数据库

    //写sql语句配置
    newLogger := logger.New(
        log.New(os.Stdout, "\r\n", log.LstdFlags), // io writer（日志输出的目标，前缀和日志包含的内容——译者注）
        logger.Config{
            SlowThreshold:             time.Second, // 慢 SQL 阈值
            LogLevel:                  logger.Info, // 日志级别
            IgnoreRecordNotFoundError: true,        // 忽略ErrRecordNotFound（记录未找到）错误
            Colorful:                  true,        // 禁用彩色打印
        },
    )

    var err error

  //将获取到的连接赋值到global.DB
    global.DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
        Logger: newLogger, //打印sql日志
    })
    if err != nil {
        panic(err)
    }
}Copy

package main

import (
    "HiChat/initialize"
)


func main() {
    //初始化数据库
    initialize.InitDB()
}Copy

go get go.uber.org/zapCopy

package initialize

import (
    "log"

    "go.uber.org/zap"
)

func InitLogger() {
    //初始化日志
    logger, err := zap.NewDevelopment()
    if err != nil {
        log.Fatal("日志初始化失败", err.Error())
    }
    //使用全局logger
    zap.ReplaceGlobals(logger)

}Copy

func main() {
    //初始化日志
    initialize.InitLogger()
    //初始化数据库
    initialize.InitDB()
}Copy