38 lines
1014 B
Go
38 lines
1014 B
Go
|
package infrastructure
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
oclib "cloud.o-forge.io/core/oc-lib"
|
||
|
|
"cloud.o-forge.io/core/oc-lib/models/peer"
|
||
|
|
"cloud.o-forge.io/core/oc-lib/tools"
|
||
|
|
)
|
||
|
|
|
||
|
|
func ListenNATS() {
|
||
|
|
tools.NewNATSCaller().ListenNats(map[tools.NATSMethod]func(tools.NATSResponse){
|
||
|
|
tools.CREATE_PEER: func(resp tools.NATSResponse) {
|
||
|
|
p := &peer.Peer{}
|
||
|
|
err := json.Unmarshal(resp.Payload, p)
|
||
|
|
if err == nil {
|
||
|
|
search := p.PeerID
|
||
|
|
if p.Relation == peer.SELF {
|
||
|
|
search = fmt.Sprintf("%v", peer.SELF.EnumIndex())
|
||
|
|
}
|
||
|
|
access := oclib.NewRequestAdmin(oclib.LibDataEnum(oclib.PEER), nil)
|
||
|
|
if data := access.Search(nil, search, false); len(data.Data) > 0 {
|
||
|
|
pp := data.Data[0].(*peer.Peer)
|
||
|
|
access.UpdateOne(pp.Serialize(pp), pp.GetID())
|
||
|
|
} else {
|
||
|
|
access.StoreOne(p.Serialize(p))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
},
|
||
|
|
tools.REMOVE_PEER: func(tools.NATSResponse) {
|
||
|
|
p := &peer.Peer{}
|
||
|
|
access := oclib.NewRequestAdmin(oclib.LibDataEnum(oclib.PEER), nil)
|
||
|
|
access.DeleteOne(p.GetID())
|
||
|
|
},
|
||
|
|
})
|
||
|
|
}
|