blob: e31315e84d710507e20fcec214874eea1a365aa3 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
// Copyright 2023 The Pigweed Authors
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may not
// use this file except in compliance with the License. You may obtain a copy of
// the License at
//
// https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
// License for the specific language governing permissions and limitations under
// the License.
#pragma once
#include "pw_rpc/client_server.h"
#include "pw_rpc/packet_meta.h"
#include "pw_rpc_transport/rpc_transport.h"
#include "pw_span/span.h"
#include "pw_status/status.h"
namespace pw::rpc {
// An RpcPacketProcessor implementation that uses an incoming RPC packet
// metadata to find its target service and sends the packet to that service for
// processing.
class ServiceRegistry : public RpcPacketProcessor {
public:
explicit ServiceRegistry(span<Channel> channels) : client_server_(channels) {}
ClientServer& client_server() { return client_server_; }
template <typename Service>
typename Service::Client CreateClient(uint32_t channel_id) {
return typename Service::Client(client_server_.client(), channel_id);
}
void RegisterService(Service& service) {
client_server_.server().RegisterService(service);
}
Status ProcessRpcPacket(ConstByteSpan rpc_packet) override {
PW_TRY_ASSIGN(const auto meta, PacketMeta::FromBuffer(rpc_packet));
if (meta.destination_is_client()) {
return client_server_.client().ProcessPacket(rpc_packet);
}
if (meta.destination_is_server()) {
return client_server_.server().ProcessPacket(rpc_packet);
}
return Status::DataLoss();
}
private:
ClientServer client_server_;
};
} // namespace pw::rpc
|
