4
1
Fork 0
backend-core/nats_handlers.go

102 lines
1.8 KiB
Go
Raw Normal View History

package main
import (
2019-10-23 13:49:43 +08:00
"encoding/json"
"log"
2019-10-23 13:49:43 +08:00
"github.com/nats-io/go-nats"
)
func (h *Handler) ContactHandler(msg *nats.Msg) {
2019-10-23 13:49:43 +08:00
// Validate JSON
updateMsg := UpdateMsg{}
err := json.Unmarshal(msg.Data, &updateMsg)
if err != nil {
log.Println(err)
return
}
2019-10-23 13:49:43 +08:00
contact := Contact{}
err = json.Unmarshal([]byte(updateMsg.Data), &contact)
if err != nil {
log.Println(err)
return
}
2019-10-23 13:49:43 +08:00
// Transmit
for _, conn := range h.contactConnections {
conn <- msg.Data
}
}
func (h *Handler) ConversationHandler(msg *nats.Msg) {
2019-10-23 13:49:43 +08:00
// Validate JSON
updateMsg := UpdateMsg{}
err := json.Unmarshal(msg.Data, &updateMsg)
if err != nil {
log.Println(err)
return
}
2019-10-23 13:49:43 +08:00
conversation := Conversation{}
err = json.Unmarshal([]byte(updateMsg.Data), &conversation)
if err != nil {
log.Println(err)
return
}
2019-10-23 13:49:43 +08:00
// Transmit
for _, conn := range h.conversationConnections {
conn <- msg.Data
}
}
func (h *Handler) UserHandler(msg *nats.Msg) {
2019-10-23 13:49:43 +08:00
// Validate JSON
updateMsg := UpdateMsg{}
err := json.Unmarshal(msg.Data, &updateMsg)
if err != nil {
log.Println(err)
return
}
2019-10-23 13:49:43 +08:00
user := User{}
err = json.Unmarshal([]byte(updateMsg.Data), &user)
if err != nil {
log.Println(err)
return
}
2019-10-23 13:49:43 +08:00
// Transmit
for _, conn := range h.userConnections {
conn <- msg.Data
}
}
func (h *Handler) MemberHandler(msg *nats.Msg) {
2019-10-23 13:49:43 +08:00
// Validate JSON
updateMsg := UpdateMsg{}
err := json.Unmarshal(msg.Data, &updateMsg)
if err != nil {
log.Println(err)
return
}
2019-10-23 13:49:43 +08:00
member := Member{}
err = json.Unmarshal([]byte(updateMsg.Data), &member)
if err != nil {
log.Println(err)
return
}
2019-10-23 13:49:43 +08:00
// Get transmit channel
if channels, ok := h.memberConnections[member.Conversation]; ok {
// Transmit
for _, conn := range channels {
conn <- msg.Data
}
} else {
log.Printf("member conversation %s not found\n", member.Conversation)
}
}