zhangzengfei
2023-10-08 2cd1af13bc4e7aec4c85b9fe88db2d294af6468f
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
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"
    //"net/http"
    "os"
    "os/signal"
    "syscall"
    "vamicro/config"
    "vamicro/sysinfo-service/controllers"
    "vamicro/sysinfo-service/service"
)
 
var (
    procName = service.ProcName
    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() {
    // pprof 用于分析性能
    //go func() {
    //    logger.Info(http.ListenAndServe("0.0.0.0:6079", nil))
    //}()
 
    fm, pubTopics := initFuncMap()
    ctx, cancel := context.WithCancel(context.Background())
    var reg = &bhomeclient.RegisterInfo{
        Proc:        *proc,
        Channel:     nil,
        PubTopic:    pubTopics,
        SubTopic:    []string{},
        SubNetTopic: []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.InitGetNetNode(ms.GetLocalNetNodeByTopic)
    bhomedbapi.InitDoReq(ms.RequestOnly)
    bhomedbapi.InitLog(logger.Debug)
 
    go ms.StartServer(fm)
 
    <-q
 
    ms.DeRegister()
    cancel()
    ms.Free()
}
 
const urlPrefix = "/data/api-v"
 
func initFuncMap() (map[string]bhomeclient.MicroFunc, []string) {
    funcMap := make(map[string]bhomeclient.MicroFunc)
 
    funcMap[urlPrefix+"/sysinfo/showSystemStates"] = controllers.ShowSystemStates
    funcMap[urlPrefix+"/sysinfo/showService"] = controllers.ShowVasystemService
    funcMap[urlPrefix+"/sysinfo/showProcesses"] = controllers.ShowVasystemProcesses
 
    var pubTopics []string
    for key, _ := range funcMap {
        pubTopics = append(pubTopics, key)
    }
 
    return funcMap, pubTopics
}