-
Notifications
You must be signed in to change notification settings - Fork 0
/
nats.go
98 lines (83 loc) · 2.33 KB
/
nats.go
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
package cli
import (
"errors"
"fmt"
"net/url"
"strings"
)
// NatsUrls holds the URLs of the NATS servers.
type NatsUrls struct {
//Servers []string `help:"NATS server URLs." short:"s" default:"wss://megalarm:megalarm@nats.megalarm.fr:443"`
Servers []string `help:"NATS server URLs." short:"s" default:"nats://nats:4222"`
}
// NatsUser holds the user to connect to a NATS server.
type NatsUser struct {
User string `help:"NATS server user." short:"u" default:""`
}
// NatsPass holds the password to connect to a NATS server.
type NatsPass struct {
Pass string `help:"NATS server password." short:"p" default:""`
}
// NatsConfig is a configuration structure for NATS.
type NatsConfig struct {
NatsUrls
NatsUser
NatsPass
}
// NewNatsConfig creates a new NatsConfig.
func NewNatsConfig(natsUrls []string, natsUser, natsPass string) *NatsConfig {
return &NatsConfig{
NatsUrls: NatsUrls{
Servers: natsUrls,
},
NatsUser: NatsUser{
User: natsUser,
},
NatsPass: NatsPass{
Pass: natsPass,
},
}
}
// NatsConnectString returns a connection string to be used to connect to a NATS server.
func (n *NatsConfig) NatsConnectString(serverIndex int) (string, error) {
if serverIndex >= len(n.Servers) {
return "", fmt.Errorf("invalid server index %d", serverIndex)
}
connectString, err := BuildConnectString(n.Servers[serverIndex], n.User, n.Pass)
return connectString, err
}
// BuildConnectString builds a valid nats connection string given an url, user and passward
func BuildConnectString(urlStr, userStr, passwordStr string) (string, error) {
// check if url contains a protocol prefix
if !strings.Contains(urlStr, "://") {
urlStr = "nats://" + urlStr
}
u, err := url.Parse(urlStr)
if err != nil {
return "", err
}
scheme := u.Scheme
if scheme == "" {
return "", errors.New("no scheme in url")
}
host := u.Host
if host == "" && u.Path != "" {
host = u.Path
}
user := u.User.Username()
password, _ := u.User.Password()
natsUrl := scheme + "://"
if userStr != "" && passwordStr != "" {
natsUrl += userStr + ":" + passwordStr + "@"
} else if userStr != "" {
natsUrl += userStr + "@"
} else if passwordStr != "" {
natsUrl += passwordStr + "@"
} else if user != "" && password != "" {
natsUrl += user + ":" + password + "@"
} else if user != "" {
natsUrl += user + "@"
}
natsUrl += host
return natsUrl, nil
}