sharp-chat/SharpChat.Common/Messages/Storage/MemoryMessageStorage.cs

91 lines
3.1 KiB
C#
Raw Normal View History

2022-08-30 15:05:29 +00:00
using SharpChat.Channels;
using SharpChat.Events;
using System;
using System.Collections.Generic;
using System.Linq;
namespace SharpChat.Messages.Storage {
public class MemoryMessageStorage : IMessageStorage {
private List<MemoryMessage> Messages { get; } = new();
private List<MemoryMessageChannel> Channels { get; } = new();
private readonly object Sync = new();
public void GetMessage(long messageId, Action<IMessage> callback) {
if(callback == null)
throw new ArgumentNullException(nameof(callback));
lock(Sync)
callback(Messages.FirstOrDefault(m => m.MessageId == messageId));
}
public void GetMessages(IChannel channel, Action<IEnumerable<IMessage>> callback, int amount, int offset) {
lock(Sync) {
IEnumerable<IMessage> subset = Messages.Where(m => m.Channel.Equals(channel));
int start = subset.Count() - offset - amount;
if(start < 0) {
amount += start;
start = 0;
}
callback(subset.Skip(start).Take(amount));
}
}
private void StoreMessage(MessageCreateEvent mce) {
lock(Sync) {
MemoryMessageChannel channel = Channels.FirstOrDefault(c => mce.ChannelId.Equals(mce.ChannelId));
if(channel == null)
return; // This is basically an invalid state
Messages.Add(new(channel, mce));
}
}
private void UpdateMessage(object sender, MessageUpdateEvent mue) {
lock(Sync)
Messages.FirstOrDefault(m => m.MessageId == mue.MessageId)?.HandleEvent(sender, mue);
}
private void DeleteMessage(MessageDeleteEvent mde) {
lock(Sync)
Messages.RemoveAll(m => m.MessageId == mde.MessageId);
}
private void CreateChannel(ChannelCreateEvent cce) {
lock(Sync)
Channels.Add(new(cce));
}
private void DeleteChannel(ChannelDeleteEvent cde) {
lock(Sync) {
MemoryMessageChannel channel = Channels.FirstOrDefault(c => cde.ChannelId.Equals(c.ChannelId));
if(channel == null)
return;
Channels.Remove(channel);
Messages.RemoveAll(m => m.Channel.Equals(channel));
}
}
public void HandleEvent(object sender, IEvent evt) {
switch(evt) {
case MessageCreateEvent mce:
StoreMessage(mce);
break;
case MessageUpdateEvent mue:
UpdateMessage(sender, mue);
break;
case MessageDeleteEvent mde:
DeleteMessage(mde);
break;
case ChannelCreateEvent cce:
CreateChannel(cce);
break;
case ChannelDeleteEvent cde:
DeleteChannel(cde);
break;
}
}
}
}