main.go

package main

import (
	_ "fmt"
	"log"
	"net/http"
	"socketgo/config"
	"socketgo/service"
	"socketgo/sql"
)

func main() {
	// 初始化Nsq生产者
	errorNo, err := service.InitProducer(config.ProducerAddr)
	if err != nil {
		switch errorNo {
		case 1:
			log.Fatalf("init producer failed：%v\n", err)
			return
		case 2:
			log.Fatalf("fail to ping %v\n", err)
		}
	}
	// 初始化Nsq消费者
	service.InitConsumer("Message", "Message-channel", config.ConsumerAddr)

	// 初始化数据库连接
	sql.InitDb()

	// 初始化WebSocket
	http.HandleFunc("/", service.Index)                           // 注册首页路由
	http.HandleFunc("/ws", service.WsHandler)                     // 注册Ws路由
	http.HandleFunc("/login", service.Login)                      // 注册登录路由
	http.HandleFunc("/register", service.Register)                // 注册注册路由
	http.HandleFunc("/revise/name", service.ReviseName)           // 修改昵称
	http.HandleFunc("/revise/fontColor", service.ReviseFontColor) // 修改字体颜色

	http.Handle("/static/", http.StripPrefix("/static/", http.FileServer(http.Dir("static")))) // 注册静态资源路由
	defer sql.DB.Close()
	// panic(http.ListenAndServeTLS(":8080", "cer/imoniang.crt", "cer/imoniang.key", nil)) // SSL
	panic(http.ListenAndServe(":8080", nil)) // 设置监听信息
}