mirror of
https://github.com/cadmium-im/zirconium-sharp.git
synced 2024-11-08 11:41:04 +00:00
Make linking multiple events when saving the event
This commit is contained in:
parent
6ea903ebb9
commit
8b49ea7cd3
@ -1,4 +1,5 @@
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Threading.Tasks;
|
||||
using ChatSubsystem.Storage.Interfaces;
|
||||
using ChatSubsystem.Storage.Models;
|
||||
@ -23,7 +24,14 @@ namespace ChatSubsystem.Storage
|
||||
events = database.GetCollection<Event>(EventsCollectionName);
|
||||
}
|
||||
|
||||
public IList<Event> GetEventsForUser(EntityID user, EntityID token, int limit)
|
||||
public async Task<IList<Event>> GetEventsForUser(EntityID user, EntityID since, int limit)
|
||||
{
|
||||
var chats = await _chatStorageManager.GetChatsForUser(user);
|
||||
|
||||
return new List<Event>();
|
||||
}
|
||||
|
||||
private async Task<IList<Event>> GetEventsForChat(EntityID since, int limit)
|
||||
{
|
||||
return new List<Event>();
|
||||
}
|
||||
@ -35,16 +43,21 @@ namespace ChatSubsystem.Storage
|
||||
var startWith = (AggregateExpressionDefinition<Event, string>)"$_id";
|
||||
var @as = (FieldDefinition<EventWithChildren, IEnumerable<Event>>)"Children";
|
||||
|
||||
// link to previous global events
|
||||
var res = await events.Aggregate()
|
||||
.GraphLookup(events, connectFromField, connectToField, startWith, @as)
|
||||
.Match("{ Children: { $size: 0 } }").FirstOrDefaultAsync();
|
||||
if (res == null)
|
||||
.Match("{ Children: { $size: 0 } }").ToListAsync();
|
||||
if (res.Count == 0)
|
||||
{
|
||||
await events.InsertOneAsync(e);
|
||||
return;
|
||||
}
|
||||
e.PrevID = res.Id;
|
||||
res.ForEach(x =>
|
||||
{
|
||||
e.PrevID.Append(x.Id);
|
||||
});
|
||||
|
||||
// link to previous events in chat
|
||||
connectToField = (FieldDefinition<Event, EntityID>)"PrevEvents";
|
||||
var opts = new AggregateGraphLookupOptions<Event, Event, EventWithChildren>()
|
||||
{
|
||||
@ -52,14 +65,17 @@ namespace ChatSubsystem.Storage
|
||||
};
|
||||
res = await events.Aggregate()
|
||||
.GraphLookup(events, connectFromField, connectToField, startWith, @as, opts)
|
||||
.Match("{ Children: { $size: 0 } }").FirstOrDefaultAsync();
|
||||
if (res == null)
|
||||
.Match("{ Children: { $size: 0 } }").ToListAsync();
|
||||
if (res.Count == 0)
|
||||
{
|
||||
await events.InsertOneAsync(e);
|
||||
return;
|
||||
}
|
||||
res.ForEach(x =>
|
||||
{
|
||||
e.PrevEvents.Append(x.EventID);
|
||||
});
|
||||
|
||||
e.PrevEvent = res.EventID;
|
||||
await events.InsertOneAsync(e);
|
||||
}
|
||||
|
||||
|
@ -7,7 +7,7 @@ namespace ChatSubsystem.Storage.Interfaces
|
||||
{
|
||||
public interface IEventStorageManager
|
||||
{
|
||||
IList<Event> GetEventsForUser(EntityID user, EntityID token, int limit);
|
||||
Task<IList<Event>> GetEventsForUser(EntityID user, EntityID since, int limit);
|
||||
Task<Event> GetEventById(EntityID id);
|
||||
Task SaveEvent(Event e);
|
||||
}
|
||||
|
@ -1,3 +1,4 @@
|
||||
using System.Collections.Generic;
|
||||
using MongoDB.Bson;
|
||||
using MongoDB.Bson.Serialization.Attributes;
|
||||
using Zirconium.Core.Models;
|
||||
@ -13,8 +14,8 @@ namespace ChatSubsystem.Storage.Models
|
||||
public EntityID ChatId { get; set; }
|
||||
public string Type { get; set; }
|
||||
public long Timestamp { get; set; }
|
||||
public EntityID PrevEvent { get; set; }
|
||||
public ObjectId PrevID { get; set; }
|
||||
public IList<EntityID> PrevEvents { get; set; }
|
||||
public IList<ObjectId> PrevID { get; set; }
|
||||
public EntityID OriginServer { get; set; }
|
||||
public EventContent Content { get; set; }
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user