zhangzengfei
2024-10-22 a254bc563003a9e7b3a8f1307df38b8ae4274a4f
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
package repository
 
import (
    "time"
 
    "gat1400Exchange/models"
    "gat1400Exchange/vo"
)
 
const (
    deviceTypeApe int = iota
    deviceTypeCascade
    deviceTypeSubPlatform
)
 
type ApeRepository struct {
}
 
func NewApeRepository() ApeRepository {
    return ApeRepository{}
}
 
func (a *ApeRepository) Keepalive(id string, devType int) {
    switch devType {
    case deviceTypeApe:
        var m models.Ape
        m.Keepalive(id)
    case deviceTypeCascade:
        var m models.Cascade
        m.Keepalive(id)
    case deviceTypeSubPlatform:
        var m models.SubPlatform
        m.Keepalive(id)
    }
}
 
func (a *ApeRepository) Create(id string) error {
    var ape models.Ape
 
    // 设备存在
    if err := ape.FindById(id); err == nil {
        return nil
    }
 
    ape.Id = id
    ape.Name = id
    ape.HeartbeatTime = time.Now().Format("2006-01-02 15:04:05")
    ape.Ext = vo.Ape{
        ApeID:            id,
        Name:             "",
        Model:            "",
        IPAddr:           "",
        IPV6Addr:         "",
        Port:             0,
        Longitude:        0,
        Latitude:         0,
        PlaceCode:        "",
        Place:            "",
        OrgCode:          "",
        CapDirection:     0,
        MonitorDirection: "",
        MonitorAreaDesc:  "",
        IsOnline:         "1",
        OwnerApsID:       "",
        UserID:           "",
        Password:         "",
        FunctionType:     "2",
    }
 
    return ape.Save()
}
 
func (a *ApeRepository) List() ([]models.Ape, error) {
    var ape models.Ape
 
    return ape.FindAll()
}
 
func (a *ApeRepository) Save(fromId string, req *vo.Ape) error {
    var ape models.Ape
    _ = ape.FindById(req.ApeID)
 
    ape.Id = req.ApeID
    ape.Name = req.Name
    ape.FromId = fromId
    ape.Ext = *req
 
    return ape.Save()
}
 
func (a *ApeRepository) HandleNotification(fromId string, apes []vo.Ape) {
    for _, a := range apes {
        var ape models.Ape
 
        err := ape.FindById(a.ApeID)
        if err == nil {
            ape.Ext = a
        } else {
            ape = models.Ape{
                Id:            a.ApeID,
                Name:          a.Name,
                FromId:        fromId,
                HeartbeatTime: "",
                Ext:           a,
            }
        }
 
        ape.Save()
    }
}
 
func (a *ApeRepository) CollectDeviceType() map[string]int {
    var ret = make(map[string]int, 0)
 
    var ap models.Ape
    if list, err := ap.FindAll(); err == nil {
        for _, item := range list {
            ret[item.Id] = deviceTypeApe
        }
    }
 
    var ca models.Cascade
    if list, err := ca.FindAll(); err == nil {
        for _, item := range list {
            ret[item.Id] = deviceTypeCascade
        }
    }
 
    var su models.SubPlatform
    if list, err := su.FindAll(); err == nil {
        for _, item := range list {
            ret[item.Id] = deviceTypeSubPlatform
        }
    }
 
    return ret
}