46 lines
1.4 KiB
Go
46 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(cfg *Config) (ClientStore, error) {
|
|
driver, ok := clientStoreDrivers[cfg.ClientStore]
|
|
if !ok {
|
|
return nil, fmt.Errorf("store: unknown driver %q (forgotten import?)", cfg.ClientStore)
|
|
}
|
|
|
|
return driver.New(cfg)
|
|
}
|