From a464ae7f01e47395f3a5386d1b1d0b5f4223a29a Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期四, 05 九月 2019 15:02:25 +0800
Subject: [PATCH] update dbapi

---
 util/sqlite.go |   69 +++++++++++++++++++---------------
 1 files changed, 38 insertions(+), 31 deletions(-)

diff --git a/util/sqlite.go b/util/sqlite.go
index 4f3e000..fce2371 100644
--- a/util/sqlite.go
+++ b/util/sqlite.go
@@ -1,12 +1,12 @@
 package util
 
 import (
-	"fmt"
-
+	"basic.com/dbapi.git"
 	"basic.com/pubsub/protomsg.git"
-	"basic.com/r/dbapi.git"
+	"basic.com/valib/gopherdiscovery.git"
+	"flag"
 	"github.com/gogo/protobuf/proto"
-	"github.com/long/test/gopherdiscovery"
+	"taskpubsub/logger"
 )
 
 /*************************
@@ -23,14 +23,17 @@
 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 = "tcp://192.168.1.11:40007"
-//var urlPubSub = "tcp://192.168.1.11:50007"
+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 urlServer = "tcp://127.0.0.1:40007"
-var urlPubSub = "tcp://127.0.0.1:50007"
+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)
 var TaskSdkflag = make(chan bool)
@@ -38,62 +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("")
 	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("")
 		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) {
-	clientOne, _ := gopherdiscovery.ClientWithSub(urlServer, urlPubSub, "ip:192.168.1.124")
-	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 {
-		fmt.Println("client: ", x)
-		Getdata(x)
+		updateDbData(x)
 	}
 }

--
Gitblit v1.8.0