tracker/storage/storage.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(userID, torrentID uint64) error
	Unprune(torrentID uint64) error

	// Peers
	NewLeecher(torrent *Torrent, p *Peer) error
	RmLeecher(torrentID uint64, peerID string) error

	NewSeeder(torrent *Torrent, p *Peer) error
	RmSeeder(torrentID uint64, peerID string) error

	// Users
	DecrementSlots(userID uint64) error
}