-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #25 from CirclesUBI/dev
Dev
- Loading branch information
Showing
20 changed files
with
971 additions
and
79 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,130 @@ | ||
| using Circles.Index.Common; | ||
| using Circles.Index.Query; | ||
| using Nethermind.Core; | ||
| using Nethermind.JsonRpc; | ||
| using Nethermind.JsonRpc.Modules.Subscribe; | ||
|
|
||
| namespace Circles.Index.Rpc; | ||
|
|
||
| public class NotifyEventArgs(CirclesEvent[] events) : EventArgs | ||
| { | ||
| public CirclesEvent[] Events { get; } = events; | ||
| } | ||
|
|
||
| public class CirclesSubscription : Subscription | ||
| { | ||
| public override string Type => "circles"; | ||
|
|
||
| private readonly CirclesSubscriptionParams _param; | ||
|
|
||
| public static long SubscriberCount => _subscriberCount; | ||
| private static long _subscriberCount; | ||
|
|
||
| public CirclesSubscription(IJsonRpcDuplexClient jsonRpcDuplexClient, Context context, | ||
| CirclesSubscriptionParams param) : base( | ||
| jsonRpcDuplexClient) | ||
| { | ||
| Notification += OnNotification; | ||
| _param = param; | ||
|
|
||
| if (param.Address == Address.Zero) | ||
| { | ||
| throw new Exception("The zero address cannot be subscribed to."); | ||
| } | ||
|
|
||
| if (param.Address != null) | ||
| { | ||
| var select = new Select("V_Crc", "Avatars", ["avatar"], [ | ||
| new FilterPredicate("avatar", FilterType.Equals, param.Address?.ToString(true, false)) | ||
| ], [], 1); | ||
|
|
||
| var parameterizedSql = select.ToSql(context.Database); | ||
| var avatarInfo = context.Database.Select(parameterizedSql); | ||
| if (!avatarInfo.Rows.Any()) | ||
| { | ||
| throw new Exception($"The address {param.Address} is not a circles avatar."); | ||
| } | ||
| } | ||
|
|
||
| Interlocked.Increment(ref _subscriberCount); | ||
| } | ||
|
|
||
| public static event EventHandler<NotifyEventArgs>? Notification; | ||
|
|
||
| public static void Notify(Context context, Range<long> importedRange) | ||
| { | ||
| if (_subscriberCount == 0) | ||
| { | ||
| return; | ||
| } | ||
|
|
||
| var queryEvents = new QueryEvents(context); | ||
| var events = queryEvents.CirclesEvents(null, importedRange.Min, importedRange.Max); | ||
|
|
||
| if (events.Length == 0) | ||
| { | ||
| return; | ||
| } | ||
|
|
||
| Notification?.Invoke(null, new NotifyEventArgs(events)); | ||
| } | ||
|
|
||
| private void OnNotification(object? sender, NotifyEventArgs e) | ||
| { | ||
| ScheduleAction(async () => | ||
| { | ||
| CirclesEvent[] events; | ||
| if (_param.Address != null) | ||
| { | ||
| var filterAddress = _param.Address!.ToString(true, false); | ||
| events = FilterForAffectedAddress(e, filterAddress); | ||
| } | ||
| else | ||
| { | ||
| events = e.Events; | ||
| } | ||
| JsonRpcResult result = CreateSubscriptionMessage(events); | ||
| await JsonRpcDuplexClient.SendJsonRpcResult(result); | ||
| }); | ||
| } | ||
|
|
||
| private CirclesEvent[] FilterForAffectedAddress(NotifyEventArgs e, string filterAddress) | ||
| { | ||
| var filteredForAddress = new List<CirclesEvent>(); | ||
| var addressesInEvent = new HashSet<string>(); | ||
|
|
||
| foreach (var circlesEvent in e.Events) | ||
| { | ||
| addressesInEvent.Clear(); | ||
|
|
||
| foreach (var circlesEventValue in circlesEvent.Values) | ||
| { | ||
| if (!QueryEvents.AddressColumns.Contains(circlesEventValue.Key)) | ||
| { | ||
| continue; | ||
| } | ||
|
|
||
| if (circlesEventValue.Value is string address) | ||
| { | ||
| addressesInEvent.Add(address); | ||
| } | ||
| } | ||
|
|
||
| if (addressesInEvent.Contains(filterAddress)) | ||
| { | ||
| filteredForAddress.Add(circlesEvent); | ||
| } | ||
| } | ||
|
|
||
| return filteredForAddress.ToArray(); | ||
| } | ||
|
|
||
| public override void Dispose() | ||
| { | ||
| base.Dispose(); | ||
|
|
||
| Interlocked.Decrement(ref _subscriberCount); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,32 @@ | ||
| using System.Text.Json; | ||
| using Nethermind.Core; | ||
| using Nethermind.JsonRpc; | ||
|
|
||
| namespace Circles.Index.Rpc; | ||
|
|
||
| public class CirclesSubscriptionParams : IJsonRpcParam | ||
| { | ||
| public Address? Address { get; private set; } | ||
|
|
||
| public void ReadJson(JsonElement element, JsonSerializerOptions options) | ||
| { | ||
| JsonDocument? doc = null; | ||
| try | ||
| { | ||
| if (element.ValueKind == JsonValueKind.String) | ||
| { | ||
| doc = JsonDocument.Parse(element.GetString() ?? "{}"); | ||
| element = doc.RootElement; | ||
| } | ||
|
|
||
| if (element.TryGetProperty("address", out JsonElement addressElement)) | ||
| { | ||
| Address = Address.TryParse(addressElement.GetString(), out Address? address) ? address : null; | ||
| } | ||
| } | ||
| finally | ||
| { | ||
| doc?.Dispose(); | ||
| } | ||
| } | ||
| } |
Oops, something went wrong.