重构代码

chenshijun

2019-07-27 b5560d3cae1164f016ca9339592eda6b9008fb9f

 util/sqlite.go

@@ -1,13 +1,12 @@
package util

import (
   "fmt"

    "flag"
   "basic.com/pubsub/protomsg.git"
   "basic.com/dbapi.git"
   "basic.com/pubsub/protomsg.git"
   "basic.com/valib/gopherdiscovery.git"
   "flag"
   "github.com/gogo/protobuf/proto"
    "basic.com/valib/gopherdiscovery.git"
   "taskpubsub/logger"
)

/*************************
@@ -24,15 +23,16 @@
var CameraIds []protomsg.Camera
var CameraTasks []protomsg.CameraAndTaskInfo

var TaskSdks []protomsg.TaskSdkInfo
//var TaskSdks []protomsg.TaskSdkInfo
var TaskSdks []protomsg.TaskSdkRun

var Sdklist []string
var Sdkinfos []protomsg.Sdk

var urlServer = flag.String("urlServer","tcp://127.0.0.1:40007","heartbeat address of url server")
var urlPubSub = flag.String("urlPubsub","tcp://127.0.0.1:5007", "heartbeat pubsub address of url server")
var urlServer = flag.String("urlServer", "tcp://127.0.0.1:40007", "heartbeat address of url server")
var urlPubSub = flag.String("urlPubsub", "tcp://127.0.0.1:50007", "heartbeat pubsub address of url server")

var dbip = flag.String("dbip","127.0.0.1","address of database ip")
var dbip = flag.String("dbip", "127.0.0.1", "address of database ip")
var dbport = flag.Int("dbport", 8001, "port of database port")

var Sdkflag = make(chan bool)
@@ -41,66 +41,66 @@

var newsdkmsg = &protomsg.DbChangeMessage{}



func processinit(initchan chan bool) {
func initDbData(initchan chan bool) {
   CameraIds = camval.FindAll()

   fmt.Println("==============camera camera with task ================")
   logger.Info("==============camera camera with task ================")
   CameraTasks = camval.FindAllCameraAndTask()
   fmt.Println(CameraTasks)
   logger.Info(CameraTasks)

   TaskSdks = taskapi.FindAll()
   TaskSdks = taskapi.FindAllTaskSdkRun()

   Sdklist = sdkapi.GetAllSdkIds()
    Sdkinfos = sdkapi.FindAll("")
   Sdkinfos = sdkapi.FindAll("")
   initchan <- true
}

func Getdata(opt []byte) {
func updateDbData(opt []byte) {
   if err := proto.Unmarshal(opt, newsdkmsg); err != nil {
      fmt.Println("publichshMessage ", err)
      logger.Error("publichshMessage ", err)
      return
   }

   switch newsdkmsg.Table {
   case protomsg.TableChanged_T_Camera:
      fmt.Println("update camera")
      logger.Info("update camera")
      CameraIds = camval.FindAll()
      Cameraflag <- true
      logger.Info("update camera finish.")

   case protomsg.TableChanged_T_CameraTask:
      fmt.Println("update cameratask")
      logger.Info("update cameratask")
      CameraTasks = camval.FindAllCameraAndTask()
      fmt.Println("update cameratask finished!")
      logger.Info("update cameratask finished!")

   case protomsg.TableChanged_T_TaskSdk:
      fmt.Println("update tasksdk")
      TaskSdks = taskapi.FindAll()
      logger.Info("update tasksdk")
      TaskSdks = taskapi.FindAllTaskSdkRun()
      TaskSdkflag <- true

   case protomsg.TableChanged_T_Sdk:
      fmt.Println("update sdk")
      logger.Info("update sdk")
      Sdklist = sdkapi.GetAllSdkIds()
        Sdkinfos = sdkapi.FindAll("")
      Sdkinfos = sdkapi.FindAll("")
      Sdkflag <- true
      fmt.Println("update finished!")

   default:
      fmt.Println("unknow type operation")
      logger.Info("unknow type operation")
   }
   fmt.Println(newsdkmsg)

   logger.Info(newsdkmsg)
}

func Init(initchan chan bool) {
    dbapi.Init(*dbip, *dbport)
   clientOne, _ := gopherdiscovery.ClientWithSub(*urlServer, *urlPubSub, "ip:local")
   recvinit := clientOne.HeartBeatMsg()
   fmt.Println(<-recvinit)
   processinit(initchan)
   dbapi.Init(*dbip, *dbport)
   client, _ := gopherdiscovery.ClientWithSub(*urlServer, *urlPubSub, "ip:local")
   recvinit := client.HeartBeatMsg()
   _ = <-recvinit

   peers, _ := clientOne.Peers()
   initDbData(initchan)

   peers, _ := client.Peers()
   for x := range peers {
      Getdata(x)
      updateDbData(x)
   }
}