dsn.go

package gohive

import (
	"fmt"
	"regexp"
)

type config struct {
	User   string
	Passwd string
	Addr   string
}

var (
	// Regexp syntax: https://github.com/google/re2/wiki/Syntax
	reDSN        = regexp.MustCompile(`(.+@)?([^@/]+)`)
	reUserPasswd = regexp.MustCompile(`([^:@]+)(:[^:@]+)?@`)
)

// parseDSN requires DSN names in the format [user[:password]@]addr/dbname.
func parseDSN(dsn string) (*config, error) {
	// Please read https://play.golang.org/p/_CSLvl1AxOX before code review.
	sub := reDSN.FindStringSubmatch(dsn)
	if len(sub) != 3 {
		return nil, fmt.Errorf("The DSN %s doesn't match [user[:password]@]addr", dsn)
	}

	up := reUserPasswd.FindStringSubmatch(sub[1])
	if len(up) == 3 {
		if len(up[2]) > 0 {
			return &config{User: up[1], Passwd: up[2][1:], Addr: sub[2]}, nil
		}
		return &config{User: up[1], Addr: sub[2]}, nil
	}
	return &config{Addr: sub[2]}, nil
}