Add flexibility in subject matching for nats_jetstream

Signed-off-by: Evan Anderson <[email protected]>

protocol/nats_jetstream/v2/receiver.go

@@ -8,6 +8,7 @@ package nats_jetstream
 import (
 	"context"
 	"io"
+	"strings"
 	"sync"
 
 	"github.com/nats-io/nats.go"
@@ -89,10 +90,20 @@ func NewConsumerFromConn(conn *nats.Conn, stream, subject string, jsmOpts []nats
 
 	streamInfo, err := jsm.StreamInfo(stream, jsmOpts...)
 
+	subjectMatch := stream + ".*"
+	if !strings.HasPrefix(subject, stream) {
+		// Use an empty subject parameter in conjunction with
+		// nats.ConsumerFilterSubjects
+		subjectMatch = ""
+	} else if strings.Count(subject[len(stream):], ".") > 1 {
+		// More than one "." in the remainder of subject, use ".>" to match
+		subjectMatch = stream + ".>"
+	}
+
 	if streamInfo == nil || err != nil && err.Error() == "stream not found" {
 		_, err = jsm.AddStream(&nats.StreamConfig{
 			Name:     stream,
-			Subjects: []string{stream + ".*"},
+			Subjects: []string{subjectMatch},
 		})
 		if err != nil {
 			return nil, err