Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add a standard subscription [ci-skip] #950

Merged
merged 2 commits into from Apr 7, 2022
Merged

Add a standard subscription [ci-skip] #950

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
632b0f8
add a standard subscription [ci-skip]
ColinSullivan1 Apr 7, 2022
f6e4f8d
add geo URL and status JSON
ColinSullivan1 Apr 7, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
54 changes: 41 additions & 13 deletions examples/nats-echo/main.go
View file
Open in desktop
Expand Up @@ -44,17 +44,27 @@ func showUsageAndExit(exitcode int) {
}

func printMsg(m *nats.Msg, i int) {
log.Printf("[#%d] Echoing to [%s]: %q", i, m.Reply, m.Data)
log.Printf("[#%d] Echoing from [%s] to [%s]: %q", i, m.Subject, m.Reply, m.Data)
}

func printStatusMsg(m *nats.Msg, i int) {
log.Printf("[#%d] Sending status from [%s] to [%s]: %q", i, m.Subject, m.Reply, m.Data)
}

type serviceStatus struct {
Id string `json:"id"`
Geo string `json:"geo"`
}

func main() {
var urls = flag.String("s", nats.DefaultURL, "The nats server URLs (separated by comma)")
var userCreds = flag.String("creds", "", "User Credentials File")
var nkeyFile = flag.String("nkey", "", "NKey Seed File")
var serviceId = flag.String("id", "NATS Echo Service", "Identifier for this service")
var showTime = flag.Bool("t", false, "Display timestamps")
var showHelp = flag.Bool("h", false, "Show help message")
var geoloc = flag.Bool("geo", false, "Display geo location of echo service")
var geo string
var geo string = "unknown"

log.SetFlags(0)
flag.Usage = usage
Expand All @@ -74,7 +84,7 @@ func main() {
geo = lookupGeo()
}
// Connect Options.
opts := []nats.Option{nats.Name("NATS Echo Service")}
opts := []nats.Option{nats.Name(*serviceId)}
opts = setupConnOptions(opts)

if *userCreds != "" && *nkeyFile != "" {
Expand All @@ -101,19 +111,29 @@ func main() {
log.Fatal(err)
}

subj, i := args[0], 0
subj, iEcho, iStatus := args[0], 0, 0
statusSubj := subj + ".status"

nc.QueueSubscribe(subj, "echo", func(msg *nats.Msg) {
i++
iEcho++
printMsg(msg, iEcho)
if msg.Reply != "" {
printMsg(msg, i)
// Just echo back what they sent us.
if geo != "" {
m := fmt.Sprintf("[%s]: %q", geo, msg.Data)
nc.Publish(msg.Reply, []byte(m))
var payload []byte
if geo != "unknown" {
payload = []byte(fmt.Sprintf("[%s]: %q", geo, msg.Data))
} else {
nc.Publish(msg.Reply, msg.Data)
payload = msg.Data
}
nc.Publish(msg.Reply, payload)
}
})
nc.Subscribe(statusSubj, func(msg *nats.Msg) {
iStatus++
printStatusMsg(msg, iStatus)
if msg.Reply != "" {
payload, _ := json.Marshal(&serviceStatus{Id: *serviceId, Geo: geo})
nc.Publish(msg.Reply, payload)
}
})
nc.Flush()
Expand All @@ -122,9 +142,11 @@ func main() {
log.Fatal(err)
}

log.Printf("Echo Service ID: [%s]", *serviceId)
log.Printf("Echo Service listening on [%s]\n", subj)
log.Printf("Echo Service (Status) listening on [%s]\n", statusSubj)

// Now handle signal to terminate so we cam drain on exit.
// Now handle signal to terminate so we can drain on exit.
c := make(chan os.Signal, 1)
signal.Notify(c, syscall.SIGINT)

Expand Down Expand Up @@ -176,9 +198,15 @@ type geo struct {
// lookup our current region and country..
func lookupGeo() string {
c := &http.Client{Timeout: 2 * time.Second}
resp, err := c.Get("https://ipapi.co/json")

url := os.Getenv("ECHO_SVC_GEO_URL")
if len(url) == 0 {
url = "https://ipapi.co/json"
}

resp, err := c.Get(url)
if err != nil || resp == nil {
log.Fatalf("Could not retrive geo location data: %v", err)
log.Fatalf("Could not retrieve geo location data: %v", err)
}
defer resp.Body.Close()
body, _ := ioutil.ReadAll(resp.Body)
Expand Down