package cache
|
|
import (
|
"basic.com/pubsub/cache.git/shardmap"
|
"basic.com/pubsub/protomsg.git"
|
"basic.com/valib/bhomeclient.git"
|
"basic.com/valib/bhomedbapi.git"
|
"basic.com/valib/c_bhomebus.git/proto/source/bhome_msg"
|
"basic.com/valib/logger.git"
|
"encoding/json"
|
"errors"
|
"strconv"
|
)
|
|
var cMap *shardmap.ShardMap
|
|
const (
|
SERVER_KEY = "SERVERINFO" //服务器信息
|
SCENERULES = "SCENERULES" //摄像机场景列表
|
)
|
|
func UpdateBySub(msg *bhome_msg.MsgPublish) {
|
switch string(msg.Topic) {
|
case bhomeclient.Proc_System_Service:
|
initServerInfo()
|
case bhomeclient.Proc_Scene_Service:
|
var protoMsg protomsg.DbChangeMessage
|
if err := json.Unmarshal(msg.Data, &protoMsg); err == nil {
|
initCameraRules()
|
}
|
}
|
}
|
|
//缓存一些其他数据库的数据，并通过订阅消息实时更新
|
func InitDataCache() {
|
cMap = shardmap.New(uint8(32))
|
|
initServerInfo() //初始化服务器配置信息
|
|
initCameraRules()
|
}
|
|
func initServerInfo() {
|
var api bhomedbapi.SysSetApi
|
b, s := api.GetServerInfo()
|
logger.Debug("GetServerInfo b:", b, "s:", s)
|
if b && s.ServerId != ""{
|
cMap.Set(SERVER_KEY,s)
|
}
|
}
|
|
func GetServerInfo() (conf protomsg.LocalConfig,err error) {
|
config, b := cMap.Get(SERVER_KEY)
|
if b {
|
return config.(protomsg.LocalConfig),nil
|
} else {
|
initServerInfo()
|
nc, nb := cMap.Get(SERVER_KEY)
|
if nb {
|
return nc.(protomsg.LocalConfig), nil
|
} else {
|
return conf,errors.New("conf not found")
|
}
|
}
|
}
|
|
|
func initCameraRules() {
|
var api bhomedbapi.CameraRuleApi
|
b,allCamRules := api.FindAll()
|
logger.Debug("initCameraRules b:", b, " len(allCamRules):", len(allCamRules))
|
if b {
|
cMap.Set(SCENERULES, allCamRules)
|
}
|
}
|
|
func GetCameraRules() ([]protomsg.CameraAndRules, error) {
|
allRules, b := cMap.Get(SCENERULES)
|
if b {
|
return allRules.([]protomsg.CameraAndRules), nil
|
} else {
|
initCameraRules()
|
nr, nb := cMap.Get(SCENERULES)
|
if nb {
|
return nr.([]protomsg.CameraAndRules), nil
|
} else {
|
return nil, errors.New("no CameraAndRules cached")
|
}
|
}
|
}
|
|
//获取weedfs的访问地址
|
func GetWeedfsUri() (serverId string,uri string,err error) {
|
localConf, err2 := GetServerInfo()
|
if err2 !=nil || localConf.WebPicIp == "" {
|
logger.Debug("localConfig is wrong!!!")
|
return "", "", errors.New("localConfig err")
|
}
|
weedfsUri := "http://"+localConf.WebPicIp+":"+strconv.Itoa(int(localConf.WebPicPort))+"/submit?collection="+localConf.ServerId+"-persistent"
|
return localConf.ServerId, weedfsUri, nil
|
}
|
|
//获取ES的访问地址
|
func GetESConfig() (serverId string, ip string,port int,err error) {
|
localConf, err2 := GetServerInfo()
|
if err2 !=nil || localConf.WebPicIp == "" {
|
logger.Debug("localConfig is wrong!!!")
|
return "", "", 0, errors.New("localConfig err")
|
}
|
return localConf.ServerId, localConf.AlarmIp, int(localConf.AlarmPort), nil
|
}
|
