81 lines
2 KiB
Go
81 lines
2 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 mock implements the models interface for a BitTorrent tracker's
|
|
// backend models. It can be used in production, but isn't recommended.
|
|
// Stored values will not persist if the tracker is restarted.
|
|
package mock
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/chihaya/chihaya/config"
|
|
"github.com/chihaya/chihaya/drivers/backend"
|
|
"github.com/chihaya/chihaya/models"
|
|
)
|
|
|
|
type driver struct{}
|
|
|
|
// Mock is a concrete implementation of the backend.Conn interface (plus some
|
|
// debugging methods) that stores deltas in memory.
|
|
type Mock struct {
|
|
deltaHistory []*models.AnnounceDelta
|
|
deltaHistoryM sync.RWMutex
|
|
}
|
|
|
|
func (d *driver) New(conf *config.DriverConfig) (backend.Conn, error) {
|
|
return &Mock{}, nil
|
|
}
|
|
|
|
// Close returns nil.
|
|
func (m *Mock) Close() error {
|
|
return nil
|
|
}
|
|
|
|
// RecordAnnounce adds a delta to the history.
|
|
func (m *Mock) RecordAnnounce(delta *models.AnnounceDelta) error {
|
|
m.deltaHistoryM.Lock()
|
|
defer m.deltaHistoryM.Unlock()
|
|
|
|
m.deltaHistory = append(m.deltaHistory, delta)
|
|
|
|
return nil
|
|
}
|
|
|
|
// DeltaHistory safely copies and returns the history of recorded deltas.
|
|
func (m *Mock) DeltaHistory() []models.AnnounceDelta {
|
|
m.deltaHistoryM.Lock()
|
|
defer m.deltaHistoryM.Unlock()
|
|
|
|
cp := make([]models.AnnounceDelta, len(m.deltaHistory))
|
|
for index, delta := range m.deltaHistory {
|
|
cp[index] = *delta
|
|
}
|
|
|
|
return cp
|
|
}
|
|
|
|
// LoadTorrents returns (nil, nil).
|
|
func (m *Mock) LoadTorrents(ids []uint64) ([]*models.Torrent, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
// LoadAllTorrents returns (nil, nil).
|
|
func (m *Mock) LoadAllTorrents() ([]*models.Torrent, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
// LoadUsers returns (nil, nil).
|
|
func (m *Mock) LoadUsers(ids []uint64) ([]*models.User, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
// LoadAllUsers returns (nil, nil).
|
|
func (m *Mock) LoadAllUsers(ids []uint64) ([]*models.User, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func init() {
|
|
backend.Register("mock", &driver{})
|
|
}
|