Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 15 additions & 0 deletions include/solana.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -1103,6 +1103,14 @@ class Connection {
/// Websocket requests
namespace subscription {

enum class LogsFilter : short { ALL, ALLWITHVOTES };

NLOHMANN_JSON_SERIALIZE_ENUM(LogsFilter,
{
{LogsFilter::ALL, "all"},
{LogsFilter::ALLWITHVOTES, "allWithVotes"},
})

/**
* Subscribe to an account to receive notifications when the lamports or data
* for a given account public key changes
Expand Down Expand Up @@ -1138,6 +1146,13 @@ class WebSocketSubscriber {
/// @brief remove the account change listener for the given id
/// @param sub_id the id for which removing subscription is needed
void removeAccountChangeListener(RequestIdType sub_id);

int onLogs(Callback callback,
const Commitment &commitment = Commitment::FINALIZED,
const LogsFilter &logFilter = LogsFilter::ALL,
Callback on_subscibe = nullptr, Callback on_unsubscribe = nullptr);

void removeOnLogsListener(RequestIdType sub_id);
};
} // namespace subscription
} // namespace rpc
Expand Down
25 changes: 25 additions & 0 deletions lib/solana.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -1137,6 +1137,31 @@ int WebSocketSubscriber::onAccountChange(const solana::PublicKey &pub_key,
void WebSocketSubscriber::removeAccountChangeListener(RequestIdType sub_id) {
sess->unsubscribe(sub_id);
}

/*Function to subscribe to logs*/
int WebSocketSubscriber::onLogs(Callback callback, const Commitment &commitment,
const LogsFilter &logFilter,
Callback on_subscibe, Callback on_unsubscribe) {
// create parameters using the user provided input
json param = {logFilter, {{"commitment", commitment}}};

// create a new request content
RequestContent req(curr_id, "logsSubscribe", "logsUnsubscribe", callback,
std::move(param), on_subscibe, on_unsubscribe);

// subscribe the new request content
sess->subscribe(req);

// increase the curr_id so that it can be used for the next request content
curr_id += 2;

return req.id;
}

void WebSocketSubscriber::removeOnLogsListener(RequestIdType sub_id) {
sess->unsubscribe(sub_id);
}

} // namespace subscription
} // namespace rpc
} // namespace solana
30 changes: 30 additions & 0 deletions tests/main.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -984,6 +984,35 @@ TEST_CASE("getBlocks") {
CHECK_LE(Blocks[Blocks.size() - 1], latestslot);
}

TEST_CASE("logs subscribe and unsubscribe") {
solana::Keypair keyPair = solana::Keypair::fromFile(KEY_PAIR_FILE);
const auto connection = solana::rpc::Connection(solana::DEVNET);
solana::rpc::subscription::WebSocketSubscriber sub("api.devnet.solana.com",
"80");
bool subscribe_called = false;
auto call_on_subscribe = [&subscribe_called](const json& data) {
subscribe_called = true;
};

// solana::rpc::subscription::WebSocketSubscriber
// subscribe for account change
int sub_id = sub.onLogs(call_on_subscribe);
// change account data
connection.requestAirdrop(keyPair.publicKey, 50);
// wait for 40 seconds for transaction to process
sleep(40);
// assure that callback has been called
CHECK(subscribe_called);
// stop listening to websocket
sub.removeOnLogsListener(sub_id);
// change account data
connection.requestAirdrop(keyPair.publicKey, 50);
// wait for 40 seconds for transaction to process
sleep(40);
// ensure that callback wasn't called
CHECK(subscribe_called);
}

TEST_CASE("getTokenSupply") {
const auto connection = solana::rpc::Connection(solana::MAINNET_BETA);
const auto TokenSupply =
Expand Down Expand Up @@ -1020,3 +1049,4 @@ TEST_CASE("getTokenAccountsByOwner") {
solana::TokenAccountsByOwnerConfig{{}, "jsonParsed"});
CHECK_GT(TokenAccountsByOwner.value.size(), 0);
}