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 tls and basic auth support #318

Merged
merged 2 commits into from
Jul 1, 2020
Merged

add tls and basic auth support #318

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
30af8f0
add tls and basic auth support
lmtyler Jul 1, 2020
ecd8e00
changes from PR review
lmtyler Jul 1, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
26 changes: 25 additions & 1 deletion examples/stan-bench/main.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,12 +18,15 @@ import (
"fmt"
"io/ioutil"
"log"
"path/filepath"
"strings"
"sync"
"sync/atomic"
"time"

"github.com/nats-io/nats.go"
"github.com/nats-io/nats.go/bench"

"github.com/nats-io/stan.go"
)

Expand All @@ -40,7 +43,7 @@ const (
)

func usage() {
log.Fatalf("Usage: stan-bench [-s server (%s)] [-c CLUSTER_ID] [-id CLIENT_ID] [-qgroup QUEUE_GROUP_NAME] [-np NUM_PUBLISHERS] [-ns NUM_SUBSCRIBERS] [-n NUM_MSGS] [-ms MESSAGE_SIZE] [-csv csvfile] [-mpa MAX_NUMBER_OF_PUBLISHED_ACKS_INFLIGHT] [-io] [-sync] [--creds credentials_file] <subject>\n", nats.DefaultURL)
log.Fatalf("Usage: stan-bench [-s server (%s)] [-c CLUSTER_ID] [-id CLIENT_ID] [-qgroup QUEUE_GROUP_NAME] [-np NUM_PUBLISHERS] [-ns NUM_SUBSCRIBERS] [-n NUM_MSGS] [-ms MESSAGE_SIZE] [-csv csvfile] [-mpa MAX_NUMBER_OF_PUBLISHED_ACKS_INFLIGHT] [-io] [-sync] [--creds credentials_file] [-cd PATH_TO_CERTS] [-cf CERTIFICATE_FILE] [-ck CERTIFICATE_KEY] [-u USERID] [-pw PASSWORD] <subject>\n", nats.DefaultURL)
}

var (
Expand All @@ -66,6 +69,11 @@ func main() {
var csvFile = flag.String("csv", "", "Save bench data to csv file")
var queue = flag.String("qgroup", "", "Queue group name")
var userCreds = flag.String("creds", "", "Credentials File")
var certDir = flag.String("cd", "", "path to certs to load")
var certFile = flag.String("cf", "", "certificate file")
var certKey = flag.String("ck", "", "certificate key")
var user = flag.String("u", "", "user id")
var pswd = flag.String("pw", "", "password")

log.SetFlags(0)
flag.Usage = usage
Expand All @@ -83,6 +91,22 @@ func main() {
opts = append(opts, nats.UserCredentials(*userCreds))
}

// Use BasicAuth
if len(*user) > 0 && len(*pswd) > 0 {
opts = append(opts, nats.UserInfo(*user, *pswd))
}

if strings.Contains(*urls, "tls://") {
if len(*certDir) > 0 {
certificateFiles, _ := filepath.Glob(filepath.Join(*certDir, "*.pem"))
opts = append(opts, nats.RootCAs(certificateFiles...))
} else if len(*certFile) > 0 && len(*certKey) > 0 {
opts = append(opts, nats.ClientCert(*certFile, *certKey))
} else {
usage()
}
}

benchmark = bench.NewBenchmark("NATS Streaming", *numSubs, *numPubs)

var startwg sync.WaitGroup
Expand Down