77 lines
1.6 KiB
Go
77 lines
1.6 KiB
Go
|
package apm
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/url"
|
||
|
|
"sync"
|
||
|
|
)
|
||
|
|
|
||
|
|
func ParseURL(uri string) (scheme string, host string, path string) {
|
||
|
|
u, err := url.Parse(uri)
|
||
|
|
if err != nil {
|
||
|
|
return "", "", ""
|
||
|
|
}
|
||
|
|
if u.Opaque != "" {
|
||
|
|
// eg. jdbc:mysql://test_user:ouupppssss@localhost:3306/sakila?profileSQL=true
|
||
|
|
u, err := url.Parse(u.Opaque)
|
||
|
|
if err != nil {
|
||
|
|
return "", "", ""
|
||
|
|
}
|
||
|
|
return u.Scheme, u.Host, u.Path
|
||
|
|
}
|
||
|
|
return u.Scheme, u.Host, u.Path
|
||
|
|
}
|
||
|
|
|
||
|
|
var (
|
||
|
|
mutex sync.Mutex
|
||
|
|
cache sync.Map
|
||
|
|
mongoCache sync.Map
|
||
|
|
)
|
||
|
|
|
||
|
|
type urlParsedInfo struct {
|
||
|
|
scheme string
|
||
|
|
host string
|
||
|
|
path string
|
||
|
|
}
|
||
|
|
|
||
|
|
func ParseURLWithCache(uri string) (scheme string, host string, path string) {
|
||
|
|
v, ok := cache.Load(uri)
|
||
|
|
if ok {
|
||
|
|
info := v.(*urlParsedInfo)
|
||
|
|
return info.scheme, info.host, info.path
|
||
|
|
}
|
||
|
|
|
||
|
|
mutex.Lock()
|
||
|
|
defer mutex.Unlock()
|
||
|
|
v, ok = cache.Load(uri)
|
||
|
|
if ok {
|
||
|
|
info := v.(*urlParsedInfo)
|
||
|
|
return info.scheme, info.host, info.path
|
||
|
|
}
|
||
|
|
|
||
|
|
scheme, host, path = ParseURL(uri)
|
||
|
|
info := urlParsedInfo{scheme: scheme, host: host, path: path}
|
||
|
|
cache.Store(uri, &info)
|
||
|
|
return info.scheme, info.host, info.path
|
||
|
|
}
|
||
|
|
|
||
|
|
func ParseMongoURLWithCache(uri string) (scheme string, host string, path string) {
|
||
|
|
v, ok := mongoCache.Load(uri)
|
||
|
|
if ok {
|
||
|
|
info := v.(*urlParsedInfo)
|
||
|
|
return info.scheme, info.host, info.path
|
||
|
|
}
|
||
|
|
|
||
|
|
mutex.Lock()
|
||
|
|
defer mutex.Unlock()
|
||
|
|
v, ok = mongoCache.Load(uri)
|
||
|
|
if ok {
|
||
|
|
info := v.(*urlParsedInfo)
|
||
|
|
return info.scheme, info.host, info.path
|
||
|
|
}
|
||
|
|
|
||
|
|
scheme, host, path = ParseMongoURL(uri)
|
||
|
|
info := urlParsedInfo{scheme: scheme, host: host, path: path}
|
||
|
|
mongoCache.Store(uri, &info)
|
||
|
|
return info.scheme, info.host, info.path
|
||
|
|
}
|