| // Copyright 2022 The Fuchsia Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include <fidl/fuchsia.examples.services/cpp/fidl.h> |
| #include <lib/async-loop/cpp/loop.h> |
| #include <lib/async-loop/default.h> |
| #include <lib/component/outgoing/cpp/outgoing_directory.h> |
| #include <lib/syslog/cpp/log_settings.h> |
| #include <lib/syslog/cpp/macros.h> |
| |
| #include <cstdlib> |
| #include <utility> |
| |
| struct Account { |
| // Account owner's name |
| std::string name; |
| // Account balance in cents |
| int64_t balance; |
| }; |
| |
| // Implementation of the `fuchsia.examples.services/ReadOnlyAccount` protocol |
| class ProtectedAccount : public fidl::Server<fuchsia_examples_services::ReadOnlyAccount> { |
| public: |
| explicit ProtectedAccount(Account account) : account_(std::move(account)) {} |
| |
| void GetOwner(GetOwnerCompleter::Sync& completer) override { completer.Reply({account_.name}); } |
| void GetBalance(GetBalanceCompleter::Sync& completer) override { |
| completer.Reply({account_.balance}); |
| } |
| |
| private: |
| Account account_; |
| }; |
| |
| // Implementation of the `fuchsia.examples.services/ReadWriteAccount` protocol |
| class OpenAccount : public fidl::Server<fuchsia_examples_services::ReadWriteAccount> { |
| public: |
| explicit OpenAccount(Account account) : account_(std::move(account)) {} |
| |
| void GetOwner(GetOwnerCompleter::Sync& completer) override { completer.Reply({account_.name}); } |
| |
| void GetBalance(GetBalanceCompleter::Sync& completer) override { |
| completer.Reply({account_.balance}); |
| } |
| |
| void Debit(DebitRequest& request, DebitCompleter::Sync& completer) override { |
| if (account_.balance >= request.amount()) { |
| account_.balance -= request.amount(); |
| completer.Reply({true}); |
| } else { |
| completer.Reply({false}); |
| } |
| FX_SLOG(INFO, "Account balance updated: ", FX_KV("balance", account_.balance)); |
| } |
| |
| void Credit(CreditRequest& request, CreditCompleter::Sync& completer) override { |
| account_.balance += request.amount(); |
| completer.Reply(); |
| FX_SLOG(INFO, "Account balance updated: ", FX_KV("balance", account_.balance)); |
| } |
| |
| private: |
| Account account_; |
| }; |
| |
| int main(int argc, const char* argv[], char* envp[]) { |
| async::Loop loop(&kAsyncLoopConfigAttachToCurrentThread); |
| |
| // Read program arguments and construct the account |
| if (argc < 3) { |
| FX_SLOG(ERROR, "Invalid number of arguments."); |
| return -1; |
| } |
| |
| component::OutgoingDirectory outgoing(loop.dispatcher()); |
| auto result = outgoing.ServeFromStartupInfo(); |
| if (result.is_error()) { |
| FX_SLOG(ERROR, "Failed to serve outgoing directory."); |
| return -1; |
| } |
| |
| auto name = argv[1]; |
| auto balance = atoi(argv[2]); |
| Account user_account = {.name = name, .balance = balance}; |
| FX_SLOG(INFO, "Starting bank account provider", FX_KV("name", user_account.name.c_str()), |
| FX_KV("balance", user_account.balance)); |
| |
| fidl::ServerBindingGroup<fuchsia_examples_services::ReadOnlyAccount> read_only_bindings; |
| fidl::ServerBindingGroup<fuchsia_examples_services::ReadWriteAccount> read_write_bindings; |
| |
| ProtectedAccount protected_account(user_account); |
| OpenAccount open_account(user_account); |
| |
| auto handler = fuchsia_examples_services::BankAccount::InstanceHandler({ |
| .read_only = read_only_bindings.CreateHandler(&protected_account, loop.dispatcher(), |
| fidl::kIgnoreBindingClosure), |
| .read_write = read_write_bindings.CreateHandler(&open_account, loop.dispatcher(), |
| fidl::kIgnoreBindingClosure), |
| }); |
| |
| result = outgoing.AddService<fuchsia_examples_services::BankAccount>(std::move(handler)); |
| if (result.is_error()) { |
| FX_SLOG(ERROR, "Failed to add service to outgoing."); |
| return -1; |
| } |
| |
| loop.Run(); |
| return 0; |
| } |