wlinator/nadekobot
wlinator/nadekobot
1
Fork 0
mirror of https://gitlab.com/Kwoth/nadekobot.git synced 2024-10-03 04:23:13 +00:00
Code Issues Releases Wiki Activity

Fixed some NRT warnings in pub/sub mechanism

Browse source
This commit is contained in:
Kwoth 2021-12-29 06:59:41 +01:00
parent 82000c97a4
commit 6eee161b6b
5 changed files with 44 additions and 34 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/NadekoBot/Common/PubSub/EventPubSub.cs
View file

@ -2,12 +2,13 @@ namespace NadekoBot.Common;
public class EventPubSub : IPubSub
{
private readonly Dictionary<string, Dictionary<Delegate, List<Func<object?, ValueTask>>>> _actions = new();
private readonly Dictionary<string, Dictionary<Delegate, List<Func<object, ValueTask>>>> _actions = new();
private readonly object _locker = new();
public Task Sub<TData>(in TypedKey<TData> key, Func<TData?, ValueTask> action)
public Task Sub<TData>(in TypedKey<TData> key, Func<TData, ValueTask> action)
where TData: notnull
{
Func<object?, ValueTask> localAction = obj => action((TData?)obj);
Func<object, ValueTask> localAction = obj => action((TData)obj);
lock (_locker)
{
if (!_actions.TryGetValue(key.Key, out var keyActions))
@ -29,6 +30,7 @@ public class EventPubSub : IPubSub
}
public Task Pub<TData>(in TypedKey<TData> key, TData data)
where TData: notnull
{
lock (_locker)
{

6
src/NadekoBot/Common/PubSub/IPubSub.cs
View file

@ -2,6 +2,8 @@ namespace NadekoBot.Common;
public interface IPubSub
{
public Task Pub<TData>(in TypedKey<TData> key, TData data);
public Task Sub<TData>(in TypedKey<TData> key, Func<TData?, ValueTask> action);
public Task Pub<TData>(in TypedKey<TData> key, TData data)
where TData: notnull;
public Task Sub<TData>(in TypedKey<TData> key, Func<TData, ValueTask> action)
where TData: notnull;
}

10
src/NadekoBot/Common/PubSub/RedisPubSub.cs
View file

@ -16,13 +16,15 @@ public sealed class RedisPubSub : IPubSub
}
public Task Pub<TData>(in TypedKey<TData> key, TData data)
where TData : notnull
{
var serialized = _serializer.Serialize(data);
return _multi.GetSubscriber()
.PublishAsync($"{_creds.RedisKey()}:{key.Key}", serialized, CommandFlags.FireAndForget);
}
public Task Sub<TData>(in TypedKey<TData> key, Func<TData?, ValueTask> action)
public Task Sub<TData>(in TypedKey<TData> key, Func<TData, ValueTask> action)
where TData : notnull
{
var eventName = key.Key;
@ -31,7 +33,11 @@ public sealed class RedisPubSub : IPubSub
try
{
var dataObj = _serializer.Deserialize<TData>(data);
await action(dataObj);
if(dataObj is not null)
await action(dataObj);
else
Log.Warning("Publishing event {EventName} with a null value. This is not allowed",
eventName);
}
catch (Exception ex)
{

38
src/NadekoBot/Program.cs
View file

@ -2,25 +2,25 @@ var pid = Environment.ProcessId;
var shardId = 0;
int? totalShards = null; // 0 to read from creds.yml
// if (args.Length > 0)
// {
// if (!int.TryParse(args[0], out shardId))
// {
// Console.Error.WriteLine("Invalid first argument (shard id): {0}", args[0]);
// return;
// }
//
// if (args.Length > 1)
// {
// if (!int.TryParse(args[1], out var shardCount))
// {
// Console.Error.WriteLine("Invalid second argument (total shards): {0}", args[1]);
// return;
// }
//
// totalShards = shardCount;
// }
// }
if (args.Length > 0 && args[0] != "run")
{
if (!int.TryParse(args[0], out shardId))
{
Console.Error.WriteLine("Invalid first argument (shard id): {0}", args[0]);
return;
}
if (args.Length > 1)
{
if (!int.TryParse(args[1], out var shardCount))
{
Console.Error.WriteLine("Invalid second argument (total shards): {0}", args[1]);
return;
}
totalShards = shardCount;
}
}
LogSetup.SetupLogger(shardId);
Log.Information("Pid: {ProcessId}", pid);

16
src/NadekoBot/Services/Settings/ConfigServiceBase.cs
View file

@ -27,7 +27,7 @@ public abstract class ConfigServiceBase<TSettings> : IConfigService
private readonly Dictionary<string, Func<TSettings, string, bool>> _propSetters = new();
private readonly Dictionary<string, Func<object>> _propSelectors = new();
private readonly Dictionary<string, Func<object, string>> _propPrinters = new();
private readonly Dictionary<string, string> _propComments = new();
private readonly Dictionary<string, string?> _propComments = new();
/// <summary>
/// Initialized an instance of <see cref="ConfigServiceBase{TSettings}" />
@ -106,12 +106,12 @@ public abstract class ConfigServiceBase<TSettings> : IConfigService
Expression<Func<TSettings, TProp>> selector,
SettingParser<TProp> parser,
Func<TProp, string> printer,
Func<TProp, bool> checker = null)
Func<TProp, bool>? checker = null)
{
checker ??= _ => true;
key = key.ToLowerInvariant();
_propPrinters[key] = obj => printer((TProp)obj);
_propSelectors[key] = () => selector.Compile()(data);
_propSelectors[key] = () => selector.Compile()(data)!;
_propSetters[key] = Magic(selector, parser, checker);
_propComments[key] = ((MemberExpression)selector.Body).Member.GetCustomAttribute<CommentAttribute>()?.Comment;
}
@ -146,26 +146,26 @@ public abstract class ConfigServiceBase<TSettings> : IConfigService
foreach (var memberExpression in expressions.AsEnumerable().Reverse())
{
var localProp = (PropertyInfo)memberExpression.Member;
targetObject = localProp.GetValue(targetObject);
targetObject = localProp.GetValue(targetObject)!;
}
prop!.SetValue(targetObject, value, null);
prop.SetValue(targetObject, value, null);
return true;
};
public IReadOnlyList<string> GetSettableProps()
=> _propSetters.Keys.ToList();
public string GetSetting(string prop)
public string? GetSetting(string prop)
{
prop = prop.ToLowerInvariant();
if (!_propSelectors.TryGetValue(prop, out var selector) || !_propPrinters.TryGetValue(prop, out var printer))
return default;
return null;
return printer(selector());
}
public string GetComment(string prop)
public string? GetComment(string prop)
{
if (_propComments.TryGetValue(prop, out var comment))
return comment;