You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
98 lines
2.6 KiB
98 lines
2.6 KiB
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"os/signal"
|
|
"regexp"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
rabbitmq "github.com/wagslane/go-rabbitmq"
|
|
)
|
|
|
|
var publisher *rabbitmq.Publisher
|
|
|
|
func publishHandler() {
|
|
pubch := make(chan os.Signal, 1)
|
|
signal.Notify(pubch, syscall.SIGUSR1, syscall.SIGUSR2)
|
|
|
|
//create test message to send
|
|
ptest := KzMessage{
|
|
EventCategory: "notification",
|
|
EventName: "push_req",
|
|
AccountId: "12a0f57aec7a40a1ccf6d959521d5682",
|
|
AppName: appname,
|
|
AppVersion: version,
|
|
CallId: uuid.New().String(),
|
|
MsgId: uuid.New().String(),
|
|
Node: myHostname,
|
|
ServerId: appname + "@" + myHostname,
|
|
Description: "Test message from RuhNet RabbitHunter",
|
|
To: "1234",
|
|
}
|
|
|
|
msg1, err := json.Marshal(ptest)
|
|
if err != nil {
|
|
logit(3, "Unable to marshal ptest into JSON! Published test message will be blank.")
|
|
}
|
|
|
|
msg2 := []byte(`{"test":true, "message":"test from ` + appnameFull + `","node":"` + myHostname + `"}`) //default second message
|
|
|
|
for {
|
|
select {
|
|
case sig := <-pubch:
|
|
logit(5, "Received signal "+fmt.Sprintf("%s", sig)+" on pubch channel. Publishing message... ")
|
|
var msg []byte
|
|
switch sig {
|
|
case syscall.SIGUSR1:
|
|
msg = msg1
|
|
case syscall.SIGUSR2:
|
|
msg = msg2
|
|
msgFromFile, err := readPubMessageFile(appconf.PubMessageFile)
|
|
if err == nil {
|
|
msg = msgFromFile
|
|
}
|
|
}
|
|
logit(7, "Publishing message to exchange '"+appconf.AmqpPubExch+"' with routing key '"+appconf.AmqpPubRoutingKey+"': \n"+string(msg))
|
|
err := publisher.Publish(msg,
|
|
[]string{appconf.AmqpPubRoutingKey},
|
|
rabbitmq.WithPublishOptionsContentType("application/json"),
|
|
rabbitmq.WithPublishOptionsExchange(appconf.AmqpPubExch),
|
|
//rabbitmq.WithPublishOptionsExchange("amq.fanout"),
|
|
)
|
|
if err != nil {
|
|
logit(3, "Error publishing message: "+err.Error())
|
|
fmt.Println("Error publishing message: " + err.Error())
|
|
} else {
|
|
fmt.Println("PUBLISHED: \n" + string(msg) + "\n")
|
|
}
|
|
case <-done:
|
|
fmt.Println("Shutting myself down...")
|
|
return
|
|
case <-time.After(100 * time.Millisecond):
|
|
//case <-time.After(time.Second):
|
|
//fmt.Println("tick")
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
func readPubMessageFile(filePath string) (msg []byte, err error) {
|
|
jsonFile, err := os.Open(filePath)
|
|
if err != nil {
|
|
logit(4, "Could not open config file: "+filePath+"\n"+err.Error())
|
|
return msg, err
|
|
}
|
|
defer jsonFile.Close()
|
|
fileBytes, _ := ioutil.ReadAll(jsonFile)
|
|
|
|
//strip out // comments from file:
|
|
re := regexp.MustCompile(`([\s]//.*)|(^//.*)`)
|
|
fileCleanedBytes := re.ReplaceAll(fileBytes, nil)
|
|
|
|
return fileCleanedBytes, err
|
|
}
|