99 lines
1.5 KiB
Go
99 lines
1.5 KiB
Go
package storage
|
|
|
|
import (
|
|
"fmt"
|
|
"net/url"
|
|
)
|
|
|
|
type DbMode uint8
|
|
|
|
const (
|
|
DBTYPE_NONE = iota
|
|
DBTYPE_MEM
|
|
DBTYPE_FS
|
|
DBTYPE_GDBM
|
|
DBTYPE_POSTGRES
|
|
)
|
|
|
|
const (
|
|
DBMODE_ANY DbMode = iota
|
|
DBMODE_BINARY
|
|
DBMODE_TEXT
|
|
)
|
|
|
|
const (
|
|
STORETYPE_STATE = iota
|
|
STORETYPE_RESOURCE
|
|
STORETYPE_USER
|
|
_STORETYPE_MAX
|
|
)
|
|
|
|
var (
|
|
DbModeDebug = []string{"ANY", "BIN", "TXT"}
|
|
DbTypeDebug = []string{"NONE", "MEM", "FS", "GDBM", "POSTGRES"}
|
|
DbStoreDebug = []string{"STATE", "RESOURCE", "USER"}
|
|
)
|
|
|
|
type Conns map[int8]ConnData
|
|
|
|
func NewConns() Conns {
|
|
c := make(Conns)
|
|
return c
|
|
}
|
|
|
|
func (c Conns) Set(conn ConnData, typ int8) {
|
|
if typ < 0 || typ >= _STORETYPE_MAX {
|
|
panic(fmt.Errorf("invalid store type: %d", typ))
|
|
}
|
|
c[typ] = conn
|
|
}
|
|
|
|
func (c Conns) Have(conn *ConnData) int8 {
|
|
for i := range _STORETYPE_MAX {
|
|
ii := int8(i)
|
|
v, ok := c[ii]
|
|
if !ok {
|
|
continue
|
|
}
|
|
if v.Raw() == conn.Raw() {
|
|
if v.Mode() == DBMODE_ANY || v.Mode() == conn.Mode() {
|
|
return ii
|
|
}
|
|
}
|
|
}
|
|
return -1
|
|
}
|
|
|
|
type ConnData struct {
|
|
typ int
|
|
str string
|
|
domain string
|
|
mode DbMode
|
|
}
|
|
|
|
func (cd *ConnData) DbType() int {
|
|
return cd.typ
|
|
}
|
|
|
|
func (cd ConnData) String() string {
|
|
return fmt.Sprintf("conn: %s, mod %s, typ %s", cd.str, DbModeDebug[uint8(cd.mode)], DbTypeDebug[uint8(cd.typ)])
|
|
}
|
|
|
|
func (cd *ConnData) Domain() string {
|
|
return cd.domain
|
|
}
|
|
|
|
func (cd *ConnData) Mode() DbMode {
|
|
return cd.mode
|
|
}
|
|
|
|
func (cd *ConnData) Path() string {
|
|
v, _ := url.Parse(cd.str)
|
|
v.RawQuery = ""
|
|
return v.String()
|
|
}
|
|
|
|
func (cd *ConnData) Raw() string {
|
|
return cd.str
|
|
}
|