91 lines
2.3 KiB
Go
91 lines
2.3 KiB
Go
// Copyright 2013 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 storage provides a generic interface for manipulating a
|
|
// BitTorrent tracker's data store.
|
|
package storage
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/pushrax/chihaya/config"
|
|
)
|
|
|
|
var (
|
|
drivers = make(map[string]Driver)
|
|
ErrTxDone = errors.New("storage: Transaction has already been committed or rolled back")
|
|
)
|
|
|
|
type Driver interface {
|
|
New(*config.Storage) DS
|
|
}
|
|
|
|
// Register makes a database driver available by the provided name.
|
|
// If Register is called twice with the same name or if driver is nil,
|
|
// it panics.
|
|
func Register(name string, driver Driver) {
|
|
if driver == nil {
|
|
panic("storage: Register driver is nil")
|
|
}
|
|
if _, dup := drivers[name]; dup {
|
|
panic("storage: Register called twice for driver " + name)
|
|
}
|
|
drivers[name] = driver
|
|
}
|
|
|
|
// Open opens a data store specified by a storage configuration.
|
|
func Open(conf *config.Storage) (DS, error) {
|
|
driver, ok := drivers[conf.Driver]
|
|
if !ok {
|
|
return nil, fmt.Errorf(
|
|
"storage: unknown driver %q (forgotten import?)",
|
|
conf.Driver,
|
|
)
|
|
}
|
|
pool := driver.New(conf)
|
|
return pool, nil
|
|
}
|
|
|
|
// DS represents a data store handle. It's expected to be safe for concurrent
|
|
// use by multiple goroutines.
|
|
//
|
|
// A pool of connections or a database/sql.DB is a great concrete type to
|
|
// implement the DS interface.
|
|
type DS interface {
|
|
Close() error
|
|
|
|
Begin() (Tx, error)
|
|
|
|
FindUser(passkey string) (*User, bool, error)
|
|
FindTorrent(infohash string) (*Torrent, bool, error)
|
|
ClientWhitelisted(peerID string) (bool, error)
|
|
}
|
|
|
|
// Tx represents an in-progress data store transaction.
|
|
// A transaction must end with a call to Commit or Rollback.
|
|
//
|
|
// After a call to Commit or Rollback, all operations on the
|
|
// transaction must fail with ErrTxDone.
|
|
type Tx interface {
|
|
Commit() error
|
|
Rollback() error
|
|
|
|
// Torrents
|
|
Snatch(u *User, t *Torrent) error
|
|
Unprune(t *Torrent) error
|
|
|
|
// Peers
|
|
NewLeecher(t *Torrent, p *Peer) error
|
|
SetLeecher(t *Torrent, p *Peer) error
|
|
RmLeecher(t *Torrent, p *Peer) error
|
|
|
|
NewSeeder(t *Torrent, p *Peer) error
|
|
SetSeeder(t *Torrent, p *Peer) error
|
|
RmSeeder(t *Torrent, p *Peer) error
|
|
|
|
// Users
|
|
IncrementSlots(u *User) error
|
|
DecrementSlots(u *User) error
|
|
}
|