Caleb Gardner
78 lines
2.2 KiB
Go
78 lines
2.2 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"github.com/CalebQ42/darkstorm-server/internal/backend"
|
|
"github.com/google/uuid"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
)
|
|
|
|
type MongoCrashTable struct {
|
|
*MongoTable[backend.CrashReport]
|
|
archiveCol *mongo.Collection
|
|
}
|
|
|
|
func NewMongoCrashTable(crashCol *mongo.Collection, archiveCol *mongo.Collection) *MongoCrashTable {
|
|
return &MongoCrashTable{
|
|
MongoTable: NewMongoTable[backend.CrashReport](crashCol),
|
|
archiveCol: archiveCol,
|
|
}
|
|
}
|
|
|
|
func (m *MongoCrashTable) Archive(ctx context.Context, toArchive backend.ArchivedCrash) error {
|
|
if toArchive.Platform == "" {
|
|
toArchive.Platform = "all"
|
|
}
|
|
_, err := m.archiveCol.InsertOne(ctx, toArchive)
|
|
return err
|
|
}
|
|
|
|
func (m *MongoCrashTable) IsArchived(ctx context.Context, ind backend.IndividualCrash) bool {
|
|
res := m.archiveCol.FindOne(ctx,
|
|
bson.M{"error": ind.Error, "stack": ind.Stack, "platform": bson.M{"$in": []string{ind.Platform, "all"}}},
|
|
)
|
|
return res.Err() == nil
|
|
}
|
|
|
|
func (m *MongoCrashTable) InsertCrash(ctx context.Context, ind backend.IndividualCrash) error {
|
|
first, _, _ := strings.Cut(ind.Stack, "\n")
|
|
res, err := m.col.UpdateOne(ctx,
|
|
bson.M{"error": ind.Error, "firstLine": first, //filter main report
|
|
"individual": bson.M{"$elemMatch": bson.M{"stack": ind.Stack, "platform": ind.Platform}}}, //filter individual
|
|
bson.M{"$inc": bson.M{"individual.$.count": 1}}, //increment count
|
|
)
|
|
if err != nil && err != mongo.ErrNoDocuments {
|
|
return err
|
|
}
|
|
if err == mongo.ErrNoDocuments || res.MatchedCount == 0 {
|
|
ind.Count = 1
|
|
res, err = m.col.UpdateMany(ctx,
|
|
bson.M{"error": ind.Error, "firstLine": first}, //filter
|
|
bson.M{"$push": bson.M{"individual": ind}}, //Add new individual report
|
|
)
|
|
if err != nil && err != mongo.ErrNoDocuments {
|
|
return err
|
|
}
|
|
if err == mongo.ErrNoDocuments || res.MatchedCount == 0 {
|
|
var id uuid.UUID
|
|
id, err = uuid.NewV7()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
ind.Count = 1
|
|
_, err = m.col.InsertOne(ctx,
|
|
backend.CrashReport{
|
|
ID: id.String(),
|
|
Error: ind.Error,
|
|
FirstLine: first,
|
|
Individual: []backend.IndividualCrash{ind},
|
|
},
|
|
)
|
|
}
|
|
}
|
|
return err
|
|
}
|