50 lines
1.4 KiB
Go
50 lines
1.4 KiB
Go
|
// Copyright 2016 The Chihaya Authors. All rights reserved.
|
||
|
|
// Use of this source code is governed by the BSD 2-Clause license,
|
||
|
|
// which can be found in the LICENSE file.
|
||
|
|
|
||
|
|
package store
|
||
|
|
|
||
|
|
import "fmt"
|
||
|
|
|
||
|
|
var clientStoreDrivers = make(map[string]ClientStoreDriver)
|
||
|
|
|
||
|
|
// ClientStore represents an interface for manipulating clientIDs.
|
||
|
|
type ClientStore interface {
|
||
|
|
CreateClient(clientID string) error
|
||
|
|
FindClient(peerID string) (bool, error)
|
||
|
|
DeleteClient(clientID string) error
|
||
|
|
}
|
||
|
|
|
||
|
|
// ClientStoreDriver represents an interface for creating a handle to the
|
||
|
|
// storage of swarms.
|
||
|
|
type ClientStoreDriver interface {
|
||
|
|
New(*Config) (ClientStore, error)
|
||
|
|
}
|
||
|
|
|
||
|
|
// RegisterClientStoreDriver makes a driver available by the provided name.
|
||
|
|
//
|
||
|
|
// If this function is called twice with the same name or if the driver is nil,
|
||
|
|
// it panics.
|
||
|
|
func RegisterClientStoreDriver(name string, driver ClientStoreDriver) {
|
||
|
|
if driver == nil {
|
||
|
|
panic("store: could not register nil ClientStoreDriver")
|
||
|
|
}
|
||
|
|
if _, dup := clientStoreDrivers[name]; dup {
|
||
|
|
panic("store: could not register duplicate ClientStoreDriver: " + name)
|
||
|
|
}
|
||
|
|
clientStoreDrivers[name] = driver
|
||
|
|
}
|
||
|
|
|
||
|
|
// OpenClientStore returns a ClientStore specified by a configuration.
|
||
|
|
func OpenClientStore(name string, cfg *Config) (ClientStore, error) {
|
||
|
|
driver, ok := clientStoreDrivers[name]
|
||
|
|
if !ok {
|
||
|
|
return nil, fmt.Errorf(
|
||
|
|
"store: unknown driver %q (forgotten import?)",
|
||
|
|
name,
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
return driver.New(cfg)
|
||
|
|
}
|