Create & Init Project...

app/infra/databus/dsn/dsn.go

@@ -0,0 +1,70 @@
+package dsn
+
+import (
+	"errors"
+	"strings"
+)
+
+var (
+	errInvalidDSN = errors.New("invalid dsn params")
+)
+
+// DSN is a configuration parsed from a DSN string
+// key:secret@group/topic=?&role=?
+type DSN struct {
+	Key    string // app key
+	Secret string // app secret
+	Group  string // kafka group
+	Topic  string // kafka topic
+	Role   string // pub or sub
+	Color  string // env color
+}
+
+// ParseDSN parse databus info.
+func ParseDSN(s string) (*DSN, error) {
+	if strings.Count(s, "@") != 1 || strings.Count(s, "/") != 1 || strings.Count(s, ":") != 1 {
+		return nil, errInvalidDSN
+	}
+	var (
+		c      = &DSN{}
+		params string
+	)
+	i := len(s) - 1
+	var j, k int
+	for j = i; j >= 0; j-- {
+		// found key:passwd
+		if s[j] == '@' {
+			for k = 0; k < j; k++ {
+				if s[k] == ':' {
+					c.Secret = s[k+1 : j]
+					break
+				}
+			}
+			c.Key = s[:k]
+			break
+		}
+	}
+	// group
+	for k = j + 1; k < i; k++ {
+		if s[k] == '/' {
+			break
+		}
+	}
+	c.Group = s[j+1 : k]
+	params = s[k+1:]
+	for _, v := range strings.Split(params, "&") {
+		param := strings.SplitN(v, "=", 2)
+		if len(param) != 2 {
+			continue
+		}
+		switch value := param[1]; strings.ToLower(param[0]) {
+		case "topic":
+			c.Topic = value
+		case "role":
+			c.Role = value
+		case "color":
+			c.Color = value
+		}
+	}
+	return c, nil
+}