#service #tower-service #pub-sub #subscription #front-end #back-end #request

linera-alloy-pubsub

Ethereum JSON-RPC publish-subscribe tower service and type definitions

1 unstable release

0.1.0 Jun 1, 2024

#42 in #tower-service


Used in 6 crates

MIT/Apache

145KB
2.5K SLoC

linera-alloy-pubsub

Ethereum JSON-RPC publish-subscribe tower service and type definitions.

Overview

PubSub services, unlike regular RPC services, are long-lived and bidirectional. They are used to subscribe to events on the server, and receive notifications when those events occur.

The PubSub system here consists of 3 logical parts:

  • The frontend is the part of the system that the user interacts with. It exposes a simple API that allows the user to issue requests and manage subscriptions.
  • The service is an intermediate layer that manages request/response mappings, subscription aliasing, and backend lifecycle events. Running PubSubConnect::into_service will spawn a long-lived service task. The service exists to manage the lifecycle of requests and subscriptions over reconnections, and to serve any number of frontends.
  • The backend is an actively running connection to the server. Users should NEVER instantiate a backend directly. Instead, they should use PubSubConnect::into_service for some connection object. Backends are responsible for managing the connection to the server,accepting user requests from the service and forwarding server responses to the service.

This crate provides the following:

  • PubSubConnect: A trait for instantiating a PubSub service by connecting to some backend. Implementors of this trait are responsible for the precise connection details, and for spawning the backend task. Users should ALWAYS call PubSubConnect::into_service to get a running service with a running backend.
  • ConnectionHandle: A handle to a running backend. This type is returned by PubSubConnect::connect, and owned by the service. Dropping the handle will shut down the backend.
  • ConnectionInterface: The reciprocal of [ConnectionHandle]. This type is owned by the backend, and is used to communicate with the service. Dropping the interface will notify the service of a terminal error.
  • PubSubFrontend: The frontend. A handle to a running PubSub service. It is used to issue requests and subscription lifecycle instructions to the service.
  • RawSubscription: A handle to a subscription. This type is yielded by the service when a user issues a get_subscription() request. It is a tokio::broadcast channel which receives notifications from the service when the server sends a notification for the subscription.
  • Subscription: A handle to a subscription expecting a specific response type. A wrapper around RawSubscription that deserializes notifications into the expected type, and allows the user to accept or discard unexpected responses.
  • SubscriptionItem: An item in a typed Subscription. This type is yielded by the subscription via the recv_any() API a notification is received and contains the deserialized item. If deserialization fails, it contains the raw JSON value.

On Handling Subscriptions

For a normal request, the user sends a request to the frontend, and later receives a response via a tokio oneshot channel. This is straightforward and easy to reason about. Subscriptions, however, are side-effects of other requests, and are long-lived. They are managed by the service and identified by a U256 id. The service uses this id to manage the subscription lifecycle, and to dispatch notifications to the correct subscribers.

Server & Local IDs

When a user issues a subscription request, the frontend sends a subscription request to the service. The service dispatches it to the RPC server via the backend. The service then intercepts the RPC server response containing the serve id, and assigns a local_id to the subscription. This local_id is used to identify the subscription in the service and in tasks consuming the subscription, while the server_id is used to identify the subscription to the RPC server, and to associate notifications with specific active subscriptions.

This allows us to use long-lived local_id values to manage subscriptions over multiple reconnections, without having to notify frontend users of the ID change when the server connection is lost. It also prevents race conditions when unsubscribing during or immediately after a reconnection.

What is a subscription request?

The service uses the is_subscription() method in the request to determine whether a given RPC request is a subscription. In general, subscription requests use the eth_subscribe method. However, other methods may also be used to create subscriptions, such as admin_peerEvents. To allow custom subscriptions on unknown methods, the Request, SerializedRequest and RpcCall expose set_is_subscription(), which can be used to mark any given request as a subscription.

When marking a request as a subscription, the service will intercept the RPC response, which MUST be a U256 value. Subscription requests that return anything other than a U256 value will not function.

Subscription Lifecycle

Regular Request Lifecycle

  1. The user issues a request to the frontend.
  2. The frontend sends the request to the service, with a oneshot channel to receive the response.
  3. The service stores the oneshot channel in its RequestManager.
  4. The service sends the request to the backend.
  5. The backend sends the request to the RPC server.
  6. The RPC server responds with a JSON RPC response.
  7. The backend sends the response to the service.
  8. The service sends the response to the waiting task via the oneshot.

Subscription Request Lifecycle:

  1. The user issues a subscription request to the frontend.
  2. The frontend sends the request to the service, with a oneshot channel to receive the response.
  3. The service stores the oneshot channel in its RequestManager.
  4. The service sends the request to the backend.
  5. The backend sends the request to the RPC server.
  6. The RPC server responds with a U256 value (the server_id).
  7. The backend sends the response to the service.
  8. The service assigns a local_id to the subscription, creates a subscription broadcast channel, and stores the relevant information in its SubscriptionManager.
  9. The service overwrites the JSON RPC response with the local_id.
  10. The service sends the response to the waiting task via the oneshot.

Subscription Notification Lifecycle

  1. The RPC server sends a notification to the backend.
  2. The backend sends the notification to the service.
  3. The service looks up the local_id i1n its SubscriptionManager.
  4. If present, the service sends the notification to the relevant channel.
    1. Otherwise, the service ignores the notification.

Dependencies

~19–30MB
~503K SLoC