Caleb Gardner
28654e237e
Added MongoDB instances of DB tables Updated some DB interfaces Added logging to count cleanup
84 lines
2.1 KiB
Go
84 lines
2.1 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/CalebQ42/darkstorm-server/internal/backend"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
)
|
|
|
|
type MongoTable[T backend.IDStruct] struct {
|
|
col *mongo.Collection
|
|
}
|
|
|
|
func NewMongoTable[T backend.IDStruct](col *mongo.Collection) *MongoTable[T] {
|
|
return &MongoTable[T]{
|
|
col: col,
|
|
}
|
|
}
|
|
|
|
func (m *MongoTable[T]) Get(ID string) (data *T, err error) {
|
|
res := m.col.FindOne(context.Background(), bson.M{"_id": ID})
|
|
if res.Err() == mongo.ErrNoDocuments {
|
|
return nil, backend.ErrNotFound
|
|
} else if res.Err() != nil {
|
|
return nil, res.Err()
|
|
}
|
|
var out T
|
|
err = res.Decode(&out)
|
|
return &out, err
|
|
}
|
|
|
|
func (m *MongoTable[T]) Find(values map[string]any) ([]T, error) {
|
|
res, err := m.col.Find(context.Background(), values)
|
|
if err == mongo.ErrNoDocuments {
|
|
return nil, backend.ErrNotFound
|
|
} else if err != nil {
|
|
return nil, err
|
|
}
|
|
var out []T
|
|
err = res.All(context.Background(), &out)
|
|
return out, err
|
|
}
|
|
|
|
func (m *MongoTable[T]) Insert(data T) error {
|
|
_, err := m.col.InsertOne(context.Background(), data)
|
|
return err
|
|
}
|
|
|
|
func (m *MongoTable[T]) Remove(ID string) error {
|
|
res := m.col.FindOneAndDelete(context.Background(), bson.M{"_id": ID})
|
|
return res.Err()
|
|
}
|
|
|
|
func (m *MongoTable[T]) FullUpdate(ID string, data T) error {
|
|
res := m.col.FindOneAndReplace(context.Background(), bson.M{"_id": ID}, data)
|
|
if res.Err() == mongo.ErrNoDocuments {
|
|
return backend.ErrNotFound
|
|
}
|
|
return res.Err()
|
|
}
|
|
|
|
func (m *MongoTable[T]) PartUpdate(ID string, update map[string]any) error {
|
|
res := m.col.FindOneAndUpdate(context.Background(), bson.M{"_id": ID}, update)
|
|
if res.Err() == mongo.ErrNoDocuments {
|
|
return backend.ErrNotFound
|
|
}
|
|
return res.Err()
|
|
}
|
|
|
|
func (m *MongoTable[CountLog]) RemoveOldLogs(date int) {
|
|
m.col.DeleteMany(context.Background(), bson.M{"date": bson.M{"$lt": date}})
|
|
}
|
|
func (m *MongoTable[CountLog]) Count(platform string) (int, error) {
|
|
var filter bson.M
|
|
if platform == "" || platform == "all" {
|
|
filter = bson.M{}
|
|
} else {
|
|
filter = bson.M{"platform": platform}
|
|
}
|
|
out, err := m.col.CountDocuments(context.Background(), filter)
|
|
return int(out), err
|
|
}
|