github/Discord.Net
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improved GetMessages with cache enabled

Browse Source
This commit is contained in:
RogueException
2016-10-06 05:37:45 -03:00
parent bb1fc2c42a
commit 0b47d5bd2e
1 changed files with 32 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
src/Discord.Net.WebSocket/Entities/Channels/SocketChannelHelper.cs
View File

@@ -11,25 +11,42 @@ namespace Discord.WebSocket
public static IAsyncEnumerable<IReadOnlyCollection<IMessage>> GetMessagesAsync(SocketChannel channel, DiscordSocketClient discord, MessageCache messages, public static IAsyncEnumerable<IReadOnlyCollection<IMessage>> GetMessagesAsync(SocketChannel channel, DiscordSocketClient discord, MessageCache messages,
ulong? fromMessageId, Direction dir, int limit, CacheMode mode, RequestOptions options) ulong? fromMessageId, Direction dir, int limit, CacheMode mode, RequestOptions options)
{ {
IReadOnlyCollection<SocketMessage> cachedMessages; if (dir == Direction.Around)
IAsyncEnumerable<IReadOnlyCollection<IMessage>> result; throw new NotImplementedException(); //TODO: Impl
if (messages != null) //Cache enabled IReadOnlyCollection<SocketMessage> cachedMessages = null;
cachedMessages = messages.GetMany(fromMessageId, dir, limit); IAsyncEnumerable<IReadOnlyCollection<IMessage>> result = null;
else
cachedMessages = ImmutableArray.Create<SocketMessage>();
result = ImmutableArray.Create(cachedMessages).ToAsyncEnumerable<IReadOnlyCollection<IMessage>>(); if (dir == Direction.After && fromMessageId == null)
limit -= cachedMessages.Count; return AsyncEnumerable.Empty<IReadOnlyCollection<IMessage>>();
if (mode == CacheMode.CacheOnly || limit <= 0)
return result; if (dir == Direction.Before || mode == CacheMode.CacheOnly)
{
if (messages != null) //Cache enabled
cachedMessages = messages.GetMany(fromMessageId, dir, limit);
else
cachedMessages = ImmutableArray.Create<SocketMessage>();
result = ImmutableArray.Create(cachedMessages).ToAsyncEnumerable<IReadOnlyCollection<IMessage>>();
}
if (dir == Direction.Before) if (dir == Direction.Before)
fromMessageId = cachedMessages.Min(x => x.Id); {
limit -= cachedMessages.Count;
if (mode == CacheMode.CacheOnly || limit <= 0)
return result;
//Download remaining messages
var downloadedMessages = ChannelHelper.GetMessagesAsync(channel, discord, cachedMessages.Min(x => x.Id), dir, limit, options);
return result.Concat(downloadedMessages);
}
else else
fromMessageId = cachedMessages.Max(x => x.Id); {
var downloadedMessages = ChannelHelper.GetMessagesAsync(channel, discord, fromMessageId, dir, limit, options); if (mode == CacheMode.CacheOnly)
return result.Concat(downloadedMessages); return result;
//Dont use cache in this case
return ChannelHelper.GetMessagesAsync(channel, discord, fromMessageId, dir, limit, options);
}
} }
public static IReadOnlyCollection<SocketMessage> GetCachedMessages(SocketChannel channel, DiscordSocketClient discord, MessageCache messages, public static IReadOnlyCollection<SocketMessage> GetCachedMessages(SocketChannel channel, DiscordSocketClient discord, MessageCache messages,
ulong? fromMessageId, Direction dir, int limit) ulong? fromMessageId, Direction dir, int limit)