iot init

main.go

@@ -2,19 +2,25 @@ package main
 
 import (
 	"context"
-	"pandax/pkg/config"
-
+	"fmt"
 	"github.com/XM-GO/PandaKit/logger"
+	"github.com/XM-GO/PandaKit/rediscli"
 	"github.com/XM-GO/PandaKit/restfulx"
 	"github.com/XM-GO/PandaKit/starter"
 	"github.com/spf13/cobra"
 	"log"
+	"m7s.live/engine/v4"
+	_ "m7s.live/plugin/gb28181/v4"
+	_ "m7s.live/plugin/jessica/v4"
+	_ "m7s.live/plugin/rtmp/v4"
 	"os"
 	"os/signal"
-	"pandax/apps/job/jobs"
+	"pandax/iothub"
+	"pandax/pkg/config"
 	"pandax/pkg/global"
 	"pandax/pkg/initialize"
 	"pandax/pkg/middleware"
+	"pandax/pkg/tdengine"
 	"syscall"
 )
 
@@ -23,46 +29,60 @@ var (
 )
 
 var rootCmd = &cobra.Command{
-	Use:   "panda is the main component in the panda.",
-	Short: `panda is go gin frame`,
+	Use:   "pandax is the main component in the panda.",
+	Short: `pandax is go go-restful frame`,
 	PreRun: func(cmd *cobra.Command, args []string) {
 		if configFile != "" {
 			global.Conf = config.InitConfig(configFile)
+			log.Println(global.Conf.Log.Level)
 			global.Log = logger.InitLog(global.Conf.Log.File.GetFilename(), global.Conf.Log.Level)
 			dbGorm := starter.DbGorm{Type: global.Conf.Server.DbType}
 			if global.Conf.Server.DbType == "mysql" {
 				dbGorm.Dsn = global.Conf.Mysql.Dsn()
 				dbGorm.MaxIdleConns = global.Conf.Mysql.MaxIdleConns
 				dbGorm.MaxOpenConns = global.Conf.Mysql.MaxOpenConns
-			} else {
-				dbGorm.Dsn = global.Conf.Postgresql.PgDsn()
-				dbGorm.MaxIdleConns = global.Conf.Postgresql.MaxIdleConns
-				dbGorm.MaxOpenConns = global.Conf.Postgresql.MaxOpenConns
 			}
 			global.Db = dbGorm.GormInit()
+			global.Log.Infof("%s连接成功", global.Conf.Server.DbType)
+			client, err := rediscli.NewRedisClient(global.Conf.Redis.Host, global.Conf.Redis.Password, global.Conf.Redis.Port)
+			if err != nil {
+				global.Log.Panic("Redis连接错误")
+			} else {
+				global.Log.Info("Redis连接成功")
+			}
+			global.RedisDb = client
+			tDengine, err := tdengine.NewTdengine(global.Conf.Taos.Username, global.Conf.Taos.Password, global.Conf.Taos.Host, global.Conf.Taos.Database)
+			if err != nil {
+				global.Log.Panic("Tdengine连接错误")
+			} else {
+				global.Log.Info("Tdengine连接成功")
+			}
+			global.TdDb = tDengine
 			initialize.InitTable()
+			// 初始化事件监听
+			initialize.InitEvents()
 		} else {
 			global.Log.Panic("请配置config")
 		}
 	},
 	Run: func(cmd *cobra.Command, args []string) {
-		restfulx.UseAfterHandlerInterceptor(middleware.OperationHandler)
 		// 前置 函数
 		restfulx.UseBeforeHandlerInterceptor(middleware.PermissionHandler)
 		// 后置 函数
 		restfulx.UseAfterHandlerInterceptor(middleware.LogHandler)
-		go func() {
-			// 启动系统调度任务
-			jobs.InitJob()
-			jobs.Setup()
-		}()
+		restfulx.UseAfterHandlerInterceptor(middleware.OperationHandler)
 
 		app := initialize.InitRouter()
 		global.Log.Info("路由初始化完成")
 		app.Start(context.TODO())
+		//开启IOTHUB
+		hs := iothub.InitEmqxHook(fmt.Sprintf(":%d", global.Conf.Server.GrpcPort))
+		//开启视频服务
+		go engine.Run(context.Background(), "config.yml")
 		stop := make(chan os.Signal, 1)
 		signal.Notify(stop, syscall.SIGTERM, os.Interrupt)
 		<-stop
+		hs.Stop()
 		if err := app.Stop(context.TODO()); err != nil {
 			log.Fatalf("fatal app stop: %s", err)
 			os.Exit(-3)