route/internal/server/storage.go

35 lines
620 B
Go
Raw Normal View History

package server
import (
"context"
2017-09-30 13:41:35 +00:00
"git.xeserv.us/xena/route/internal/database"
"git.xeserv.us/xena/route/lib/tun2"
)
type storageWrapper struct {
database.Storage
}
var (
_ tun2.Storage = &storageWrapper{}
)
func (s *storageWrapper) HasToken(token string) (string, []string, error) {
t, err := s.Storage.GetToken(context.Background(), token)
if err != nil {
return "", nil, err
}
return t.Owner, t.Scopes, nil
}
func (s *storageWrapper) HasRoute(domain string) (string, error) {
r, err := s.Storage.GetRoute(context.Background(), domain)
if err != nil {
return "", err
}
return r.Creator, nil
}