feat(parser): adds explicit DB parser for when we don't want a key - should probably reject keyinput on match
This commit is contained in:
parent
af049a6ade
commit
fa224d9b38
2 changed files with 21 additions and 3 deletions
|
|
@ -123,7 +123,7 @@ func (s *Store) AllStores() ([]string, error) {
|
|||
}
|
||||
|
||||
func (s *Store) FindStore(k string) (string, error) {
|
||||
_, n, err := s.parse(k, false)
|
||||
n, err := s.parseDB(k, false)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
|
@ -176,6 +176,20 @@ func (s *Store) parse(k string, defaults bool) ([]byte, string, error) {
|
|||
return []byte(key), db, nil
|
||||
}
|
||||
|
||||
func (s *Store) parseDB(v string, defaults bool) (string, error) {
|
||||
db := strings.TrimSpace(v)
|
||||
if strings.HasPrefix(db, "@") {
|
||||
db = strings.TrimPrefix(db, "@")
|
||||
}
|
||||
if db == "" {
|
||||
if defaults {
|
||||
return "default", nil
|
||||
}
|
||||
return "", fmt.Errorf("bad db format, use DB or @DB")
|
||||
}
|
||||
return strings.ToLower(db), nil
|
||||
}
|
||||
|
||||
func (s *Store) open(name string) (*badger.DB, error) {
|
||||
if name == "" {
|
||||
name = "default"
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue