一、.环境准备首先安装gin框架官网根据文档安装操作2.安装mysql驱动go get -u gorm.io/driver/mysql3.安装redis依赖go get -u github.com/go-redis/redis/v84安装websocket依赖go get github.com/gorilla/websocket二、链接MySQL和redis1.在根目录下新建config/config.yml如下mysql: localhost: 127.0.0.1 port: 3306 username: root password: admin123 database: ginchat redis: host: 127.0.0.1 port: 6379 password: selectDb: 5 polSize: 30 minIdleConn: 302.untils目录下负责链接redismysqlkafak等操作地址untils/system_init.go/** * Created by GoLand * User: lingm * Date: 2023/5/28 * Time: 下午 05:02 * Author: 现在的努力是为了小时候吹过的NB * Atom: 小白从不写注释 */ package untils import ( context fmt github.com/go-redis/redis/v8 github.com/spf13/viper gorm.io/gorm gorm.io/gorm/logger log os time gorm.io/driver/mysql ) //声明全局数据库链接 var DB *gorm.DB var Red *redis.Client // 读取mysql连接的配置参数 func InitConfig() (host string, database string, port string, username string, password string) { viper.SetConfigName(app) viper.AddConfigPath(config) err : viper.ReadInConfig() if err ! nil { fmt.Printf(读取配置文件错误, err) } host viper.GetString(mysql.localhost) database viper.GetString(mysql.database) port viper.GetString(mysql.port) username viper.GetString(mysql.username) password viper.GetString(mysql.password) return host, database, port, username, password } // 初始化连接mysql func InitMySQL() { //自定义日志打印sql语句 newLogger : logger.New( log.New(os.Stdout, \r\n, log.LstdFlags), logger.Config{ SlowThreshold: time.Second, //慢SQL阈值 LogLevel: logger.Info, //级别 Colorful: true, //彩色 }, ) host, database, port, username, password : InitConfig() urlList : username : password tcp( host : port )/ database ?charsetutf8mb4parseTimetruelocLocal DB, _ gorm.Open(mysql.Open(urlList), gorm.Config{Logger: newLogger}) fmt.Println( MySQL inited 。。。。) } // 读取redis配置文件 func RedisConfig() (host string, selectDb int, port string, password string, polSize int, minIdleConn int) { viper.SetConfigName(app) viper.AddConfigPath(config) err : viper.ReadInConfig() if err ! nil { fmt.Printf(读取redis配置文件错误, err) } host viper.GetString(redis.host) selectDb viper.GetInt(redis.selectDb) port viper.GetString(redis.port) password viper.GetString(redis.password) polSize viper.GetInt(redis.polSize) minIdleConn viper.GetInt(redis.minIdleConn) return host, selectDb, port, password, polSize, minIdleConn } // 初始化连接redis func InitRedis() { host, selectDb, port, password, polSize, minIdleConn : RedisConfig() localhost : host : port fmt.Println(地址打印, localhost) Red redis.NewClient(redis.Options{ Addr: localhost, Password: password, DB: selectDb, PoolSize: polSize, MinIdleConns: minIdleConn, }) fmt.Println( Redis inited 。。。。) } const ( PublishKey websocket ) // 发起Publish 发消息到redis func Publish(c context.Context, channel string, msg string) error { var err error err Red.Publish(c, channel, msg).Err() return err } // Subscribe订阅redis消息 func Subscribe(c context.Context, channel string) (string, error) { sub : Red.PSubscribe(c, channel) fmt.Println(Subscribe。。。, sub) msg, err : sub.ReceiveMessage(c) return msg.Payload, err }3.main.go文件引用untils链接package main import ( fmt go-study/router go-study/untils ) func main() { //引用untils下的mysql和redis untils.InitMySQL() untils.InitRedis() r : router.Router() r.Run(:8081) }三、搭建websocket骨架1.在控制器层新建如下// 防止跨域站点为找请求 var upGrade websocket.Upgrader{ CheckOrigin: func(r *http.Request) bool { return true }, } //消息发送 func SendMsg(c *gin.Context) { ws, err : upGrade.Upgrade(c.Writer, c.Request, nil) if err ! nil { fmt.Println(ws数据打印, err) return } defer func(ws *websocket.Conn) { err ws.Close() if err ! nil { fmt.Println(错误的ws日志打印) } }(ws) MsgHandler(ws, c) } // 消息中间件 func MsgHandler(ws *websocket.Conn, c context.Context) { msg, err : untils.Subscribe(c, untils.PublishKey) if err ! nil { fmt.Println(错误打印, err) } tm : time.Now().Format(2006-01-02 15:04:05) m : fmt.Sprintf([ws][%s]:%s, tm, msg) err ws.WriteMessage(1, []byte(m)) if err ! nil { fmt.Println(消息异常打印, err) } }3.将控制器层加入到路由/** * Created by GoLand * User: lingm * Date: 2023/5/28 * Time: 下午 04:24 * Author: 现在的努力是为了小时候吹过的NB * Atom: 小白从不写注释 */ package router func Router() *gin.Engine { r : gin.Default() docs.SwaggerInfo.BasePath r.GET(/swagger/*any, ginSwagger.WrapHandler(swaggerfiles.Handler)) r.GET(/user/sendMsg, controller.SendMsg) return r }4.启动项目如下5.调试websocket地址这里推荐可以使用WebSocket在线测试工具如果网站失效大家可以自行百度一下如下图这里需要注意的就是redis启动后期可能需要记录数据到redis记录到redis有个好处可以设置国企时间自动清除不用链接mysql数据库。避坑主要在于测试webscoket的地方可能会跨域需要加入如下代码解决跨域问题// 防止跨域站点为找请求 var upGrade websocket.Upgrader{ CheckOrigin: func(r *http.Request) bool { fmt.Println(控制台打印数据) return true }, }本期就到这里谢谢大家支持 go--------gin+mysql+redis+websocket搭建聊天
一、.环境准备首先安装gin框架官网根据文档安装操作2.安装mysql驱动go get -u gorm.io/driver/mysql3.安装redis依赖go get -u github.com/go-redis/redis/v84安装websocket依赖go get github.com/gorilla/websocket二、链接MySQL和redis1.在根目录下新建config/config.yml如下mysql: localhost: 127.0.0.1 port: 3306 username: root password: admin123 database: ginchat redis: host: 127.0.0.1 port: 6379 password: selectDb: 5 polSize: 30 minIdleConn: 302.untils目录下负责链接redismysqlkafak等操作地址untils/system_init.go/** * Created by GoLand * User: lingm * Date: 2023/5/28 * Time: 下午 05:02 * Author: 现在的努力是为了小时候吹过的NB * Atom: 小白从不写注释 */ package untils import ( context fmt github.com/go-redis/redis/v8 github.com/spf13/viper gorm.io/gorm gorm.io/gorm/logger log os time gorm.io/driver/mysql ) //声明全局数据库链接 var DB *gorm.DB var Red *redis.Client // 读取mysql连接的配置参数 func InitConfig() (host string, database string, port string, username string, password string) { viper.SetConfigName(app) viper.AddConfigPath(config) err : viper.ReadInConfig() if err ! nil { fmt.Printf(读取配置文件错误, err) } host viper.GetString(mysql.localhost) database viper.GetString(mysql.database) port viper.GetString(mysql.port) username viper.GetString(mysql.username) password viper.GetString(mysql.password) return host, database, port, username, password } // 初始化连接mysql func InitMySQL() { //自定义日志打印sql语句 newLogger : logger.New( log.New(os.Stdout, \r\n, log.LstdFlags), logger.Config{ SlowThreshold: time.Second, //慢SQL阈值 LogLevel: logger.Info, //级别 Colorful: true, //彩色 }, ) host, database, port, username, password : InitConfig() urlList : username : password tcp( host : port )/ database ?charsetutf8mb4parseTimetruelocLocal DB, _ gorm.Open(mysql.Open(urlList), gorm.Config{Logger: newLogger}) fmt.Println( MySQL inited 。。。。) } // 读取redis配置文件 func RedisConfig() (host string, selectDb int, port string, password string, polSize int, minIdleConn int) { viper.SetConfigName(app) viper.AddConfigPath(config) err : viper.ReadInConfig() if err ! nil { fmt.Printf(读取redis配置文件错误, err) } host viper.GetString(redis.host) selectDb viper.GetInt(redis.selectDb) port viper.GetString(redis.port) password viper.GetString(redis.password) polSize viper.GetInt(redis.polSize) minIdleConn viper.GetInt(redis.minIdleConn) return host, selectDb, port, password, polSize, minIdleConn } // 初始化连接redis func InitRedis() { host, selectDb, port, password, polSize, minIdleConn : RedisConfig() localhost : host : port fmt.Println(地址打印, localhost) Red redis.NewClient(redis.Options{ Addr: localhost, Password: password, DB: selectDb, PoolSize: polSize, MinIdleConns: minIdleConn, }) fmt.Println( Redis inited 。。。。) } const ( PublishKey websocket ) // 发起Publish 发消息到redis func Publish(c context.Context, channel string, msg string) error { var err error err Red.Publish(c, channel, msg).Err() return err } // Subscribe订阅redis消息 func Subscribe(c context.Context, channel string) (string, error) { sub : Red.PSubscribe(c, channel) fmt.Println(Subscribe。。。, sub) msg, err : sub.ReceiveMessage(c) return msg.Payload, err }3.main.go文件引用untils链接package main import ( fmt go-study/router go-study/untils ) func main() { //引用untils下的mysql和redis untils.InitMySQL() untils.InitRedis() r : router.Router() r.Run(:8081) }三、搭建websocket骨架1.在控制器层新建如下// 防止跨域站点为找请求 var upGrade websocket.Upgrader{ CheckOrigin: func(r *http.Request) bool { return true }, } //消息发送 func SendMsg(c *gin.Context) { ws, err : upGrade.Upgrade(c.Writer, c.Request, nil) if err ! nil { fmt.Println(ws数据打印, err) return } defer func(ws *websocket.Conn) { err ws.Close() if err ! nil { fmt.Println(错误的ws日志打印) } }(ws) MsgHandler(ws, c) } // 消息中间件 func MsgHandler(ws *websocket.Conn, c context.Context) { msg, err : untils.Subscribe(c, untils.PublishKey) if err ! nil { fmt.Println(错误打印, err) } tm : time.Now().Format(2006-01-02 15:04:05) m : fmt.Sprintf([ws][%s]:%s, tm, msg) err ws.WriteMessage(1, []byte(m)) if err ! nil { fmt.Println(消息异常打印, err) } }3.将控制器层加入到路由/** * Created by GoLand * User: lingm * Date: 2023/5/28 * Time: 下午 04:24 * Author: 现在的努力是为了小时候吹过的NB * Atom: 小白从不写注释 */ package router func Router() *gin.Engine { r : gin.Default() docs.SwaggerInfo.BasePath r.GET(/swagger/*any, ginSwagger.WrapHandler(swaggerfiles.Handler)) r.GET(/user/sendMsg, controller.SendMsg) return r }4.启动项目如下5.调试websocket地址这里推荐可以使用WebSocket在线测试工具如果网站失效大家可以自行百度一下如下图这里需要注意的就是redis启动后期可能需要记录数据到redis记录到redis有个好处可以设置国企时间自动清除不用链接mysql数据库。避坑主要在于测试webscoket的地方可能会跨域需要加入如下代码解决跨域问题// 防止跨域站点为找请求 var upGrade websocket.Upgrader{ CheckOrigin: func(r *http.Request) bool { fmt.Println(控制台打印数据) return true }, }本期就到这里谢谢大家支持