qixiaoning
2025-08-21 e38654fe9eff4562da4f18f8f018aed7879d493c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
package main
 
import (
    "basic.com/valib/bhomeclient.git"
    "basic.com/valib/bhomedbapi.git"
    "basic.com/valib/logger.git"
    "basic.com/valib/version.git"
    "context"
    "flag"
    "github.com/robfig/cron"
    "os"
    "os/signal"
    "syscall"
    "vamicro/compTable-service/controllers"
    "vamicro/compTable-service/models"
    "vamicro/compTable-service/service"
    "vamicro/config"
    cc "vamicro/gb28181-service/cache"
 
    "net/http"
    _ "net/http/pprof"
)
 
var (
    procName = "compTable-service"
    proc     = &bhomeclient.ProcInfo{
        Name: procName, //进程名称
        ID:   procName, //进程id
        Info: "",       //进程的描述信息,用于区分同一进程名称下多个进程
    }
    env = flag.String("e", "pro", "")
)
 
func init() {
    flag.Parse()
    vaversion.Usage()
 
    config.Init(*env)
    // 日志初始化
    var logFile = config.LogConf.Path + "vamicro-" + procName + ".log"
    logger.InitLogger(logFile, config.LogConf.Level, config.LogConf.MaxSize, config.LogConf.MaxBackups, config.LogConf.MaxAge)
    logger.Info("log init success !")
}
 
func main() {
    models.Init()
    defer models.CloseDB()
 
    go func() {
        logger.Info(http.ListenAndServe("0.0.0.0:7078", nil))
    }()
 
    ctx, cancel := context.WithCancel(context.Background())
    fm, pubTopics := initFuncMap()
    var reg = &bhomeclient.RegisterInfo{
        Proc:     *proc,
        Channel:  nil,
        PubTopic: pubTopics,
        SubTopic: []string{},
    }
 
    q := make(chan os.Signal, 1)
    signal.Notify(q, os.Interrupt, os.Kill, syscall.SIGTERM)
 
    ms, err := bhomeclient.NewMicroNode(ctx, q, config.Server.AnalyServerId, reg, logger.Debug)
    if err != nil {
        return
    }
 
    bhomedbapi.InitLog(logger.Debug)
    bhomedbapi.InitGetNetNode(ms.GetLocalNetNodeByTopic)
    bhomedbapi.InitDoReq(ms.RequestOnly)
 
    go ms.StartServer(fm)
    cc.InitDataCache()
    models.InitSync()
 
    c := cron.New()
    c.AddFunc("@hourly", func() {
        //检查底库有效期状态
        service.InitDbTableEnableStatus()
    })
    c.Start()
 
    <-q
    ms.DeRegister()
    cancel()
    ms.Free()
}
 
const urlPrefix = "/data/api-v"
 
func initFuncMap() (map[string]bhomeclient.MicroFunc, []string) {
    funcMap := make(map[string]bhomeclient.MicroFunc)
    fc := new(controllers.FileController)
    pc := new(controllers.DbPersonController)
    tc := new(controllers.DbTableController)
 
    funcMap[urlPrefix+"/dbperson/fileupload"] = fc.Fileupload
    funcMap[urlPrefix+"/dbperson/moreFileUpload"] = fc.MoreFileUpload
    funcMap[urlPrefix+"/dbperson/fileUploadTest"] = fc.UploadPersonTest
    funcMap[urlPrefix+"/dbtable/addDbTableInfo"] = tc.Add
    funcMap[urlPrefix+"/dbtable/findAllDbTablesByCurServer"] = tc.FindAllDbTablesByCurServer
    funcMap[urlPrefix+"/dbtable/findAllDbTables"] = tc.FindAllDbTables
    funcMap[urlPrefix+"/dbtable/deleteDBtablesById"] = tc.DeleteById
    funcMap[urlPrefix+"/dbtable/dbtablesByIds"] = tc.DbtablesByIds
    funcMap[urlPrefix+"/dbtable/findAllLocalDbTables"] = tc.FindAllLocalDbTables
    funcMap[urlPrefix+"/dbtable/queryDbTables"] = tc.QueryDbTables
    funcMap[urlPrefix+"/dbtable/updateDbTables"] = tc.UpdateDbTables
    funcMap[urlPrefix+"/dbtable/updateDbTableStatus"] = tc.UpdateDbTableStatus
 
    funcMap[urlPrefix+"/dbperson/move"] = pc.Move
    funcMap[urlPrefix+"/dbperson/addDbPerson"] = pc.AddDbPerson
    funcMap[urlPrefix+"/dbperson/copy"] = pc.Copy
    funcMap[urlPrefix+"/dbperson/dbPersonInfoByIds"] = pc.DbPersonInfoByIds
    funcMap[urlPrefix+"/dbperson/deleteDbPersonById"] = pc.DeleteDbPerson
    funcMap[urlPrefix+"/dbperson/deleteMoreDbPerson"] = pc.DeleteMoreDbPerson
    funcMap[urlPrefix+"/dbperson/deletePersonsByTbId"] = pc.DeletePersonsByTbId
    funcMap[urlPrefix+"/dbperson/findLikePersonIds"] = pc.FindLikePersonIds
    funcMap[urlPrefix+"/dbperson/getPersonTotal"] = pc.GetPersonTotal
    funcMap[urlPrefix+"/dbperson/getPersonsCompareCacheBase"] = pc.GetPersonsCompareCacheBase
    funcMap[urlPrefix+"/dbperson/joinDbTable"] = pc.JoinDbTable
    funcMap[urlPrefix+"/dbperson/findLikePersonIds"] = pc.FindLikePersonIds
    funcMap[urlPrefix+"/dbperson/multiUploadCarNo"] = pc.MultiUploadCarNo
    funcMap[urlPrefix+"/dbperson/queryDbPersonsByTbId"] = pc.QueryDbPersonsByTbId
    funcMap[urlPrefix+"/dbperson/extract"] = pc.FaceExtract
    funcMap[urlPrefix+"/dbperson/queryDbPersonsByCompare"] = pc.QueryDbPersonsByCompare
    funcMap[urlPrefix+"/dbperson/updateDbPerson"] = pc.UpdateDbPerson
    funcMap[urlPrefix+"/dbperson/updateFace"] = pc.UpdateFace
 
    /*********************************字典**********************************/
    funcMap[urlPrefix+"/dbperson/dicByType"] = pc.FindDicByType
    /*******************************太原营业厅*******************************/
 
    var pubTopics []string
    for key, _ := range funcMap {
        pubTopics = append(pubTopics, key)
    }
    return funcMap, pubTopics
}