87 lines
1.9 KiB
Go
87 lines
1.9 KiB
Go
package persistence
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"encoding/json"
|
|
"net/url"
|
|
"os"
|
|
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type out struct {
|
|
InfoHash string `json:"infoHash"`
|
|
Name string `json:"name"`
|
|
Files []File `json:"files"`
|
|
}
|
|
|
|
var notSupportedError = errors.New("This dummy database engine (\"stdout\") does not support any sort of queries")
|
|
|
|
func makeStdoutDatabase(_ *url.URL) (Database, error) {
|
|
s := new(stdout)
|
|
s.encoder = json.NewEncoder(os.Stdout)
|
|
return s, nil
|
|
}
|
|
|
|
type stdout struct {
|
|
encoder *json.Encoder
|
|
}
|
|
|
|
func (s *stdout) Engine() databaseEngine {
|
|
return Stdout
|
|
}
|
|
|
|
func (s *stdout) DoesTorrentExist(infoHash []byte) (bool, error) {
|
|
// Always say that "No the torrent does not exist" because we do not have
|
|
// a way to know if we have seen it before or not.
|
|
// TODO:
|
|
// A possible improvement would be using bloom filters (with low false positive
|
|
// probabilities) to apply some reasonable filtering.
|
|
return false, nil
|
|
}
|
|
|
|
func (s *stdout) AddNewTorrent(infoHash []byte, name string, files []File) error {
|
|
err := s.encoder.Encode(out{
|
|
InfoHash: hex.EncodeToString(infoHash),
|
|
Name: name,
|
|
Files: files,
|
|
})
|
|
if err != nil {
|
|
return errors.Wrap(err, "DB engine stdout encode error")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *stdout) Close() error {
|
|
return os.Stdout.Sync()
|
|
}
|
|
|
|
func (s *stdout) GetNumberOfTorrents() (uint, error) {
|
|
return 0, notSupportedError
|
|
}
|
|
|
|
func (s *stdout) QueryTorrents(
|
|
query string,
|
|
epoch int64,
|
|
orderBy OrderingCriteria,
|
|
ascending bool,
|
|
limit uint,
|
|
lastOrderedValue *float64,
|
|
lastID *uint64,
|
|
) ([]TorrentMetadata, error) {
|
|
return nil, notSupportedError
|
|
}
|
|
|
|
func (s *stdout) GetTorrent(infoHash []byte) (*TorrentMetadata, error) {
|
|
return nil, notSupportedError
|
|
}
|
|
|
|
func (s *stdout) GetFiles(infoHash []byte) ([]File, error) {
|
|
return nil, notSupportedError
|
|
}
|
|
|
|
func (s *stdout) GetStatistics(from string, n uint) (*Statistics, error) {
|
|
return nil, notSupportedError
|
|
}
|