Commit 75ba961e authored by Max Kellermann's avatar Max Kellermann

Client: move message functions into the class

parent c2d5ce0c
...@@ -141,7 +141,7 @@ src_mpd_SOURCES = \ ...@@ -141,7 +141,7 @@ src_mpd_SOURCES = \
src/ClientRead.cxx \ src/ClientRead.cxx \
src/ClientWrite.cxx \ src/ClientWrite.cxx \
src/ClientMessage.cxx src/ClientMessage.hxx \ src/ClientMessage.cxx src/ClientMessage.hxx \
src/ClientSubscribe.cxx src/ClientSubscribe.hxx \ src/ClientSubscribe.cxx \
src/ClientFile.cxx src/ClientFile.hxx \ src/ClientFile.cxx src/ClientFile.hxx \
src/Listen.cxx src/Listen.hxx \ src/Listen.cxx src/Listen.hxx \
src/LogInit.cxx src/LogInit.hxx \ src/LogInit.cxx src/LogInit.hxx \
......
...@@ -87,11 +87,6 @@ public: ...@@ -87,11 +87,6 @@ public:
} }
gcc_pure gcc_pure
bool IsSubscribed(const char *channel_name) const {
return subscriptions.find(channel_name) != subscriptions.end();
}
gcc_pure
bool IsExpired() const { bool IsExpired() const {
return !FullyBufferedSocket::IsDefined(); return !FullyBufferedSocket::IsDefined();
} }
...@@ -132,6 +127,30 @@ public: ...@@ -132,6 +127,30 @@ public:
void IdleAdd(unsigned flags); void IdleAdd(unsigned flags);
bool IdleWait(unsigned flags); bool IdleWait(unsigned flags);
enum class SubscribeResult {
/** success */
OK,
/** invalid channel name */
INVALID,
/** already subscribed to this channel */
ALREADY,
/** too many subscriptions */
FULL,
};
gcc_pure
bool IsSubscribed(const char *channel_name) const {
return subscriptions.find(channel_name) != subscriptions.end();
}
SubscribeResult Subscribe(const char *channel);
bool Unsubscribe(const char *channel);
void UnsubscribeAll();
bool PushMessage(const ClientMessage &msg);
private: private:
/* virtual methods from class BufferedSocket */ /* virtual methods from class BufferedSocket */
virtual InputResult OnSocketInput(void *data, size_t length) override; virtual InputResult OnSocketInput(void *data, size_t length) override;
......
...@@ -18,72 +18,75 @@ ...@@ -18,72 +18,75 @@
*/ */
#include "config.h" #include "config.h"
#include "ClientSubscribe.hxx"
#include "ClientInternal.hxx" #include "ClientInternal.hxx"
#include "Idle.hxx" #include "Idle.hxx"
#include <assert.h> #include <assert.h>
#include <string.h> #include <string.h>
enum client_subscribe_result bool Unsubscribe(const char *channel);
client_subscribe(Client &client, const char *channel) void UnsubscribeAll();
bool PushMessage(const ClientMessage &msg);
Client::SubscribeResult
Client::Subscribe(const char *channel)
{ {
assert(channel != nullptr); assert(channel != nullptr);
if (!client_message_valid_channel_name(channel)) if (!client_message_valid_channel_name(channel))
return CLIENT_SUBSCRIBE_INVALID; return Client::SubscribeResult::INVALID;
if (client.num_subscriptions >= CLIENT_MAX_SUBSCRIPTIONS) if (num_subscriptions >= CLIENT_MAX_SUBSCRIPTIONS)
return CLIENT_SUBSCRIBE_FULL; return Client::SubscribeResult::FULL;
auto r = client.subscriptions.insert(channel); auto r = subscriptions.insert(channel);
if (!r.second) if (!r.second)
return CLIENT_SUBSCRIBE_ALREADY; return Client::SubscribeResult::ALREADY;
++client.num_subscriptions; ++num_subscriptions;
idle_add(IDLE_SUBSCRIPTION); idle_add(IDLE_SUBSCRIPTION);
return CLIENT_SUBSCRIBE_OK; return Client::SubscribeResult::OK;
} }
bool bool
client_unsubscribe(Client &client, const char *channel) Client::Unsubscribe(const char *channel)
{ {
const auto i = client.subscriptions.find(channel); const auto i = subscriptions.find(channel);
if (i == client.subscriptions.end()) if (i == subscriptions.end())
return false; return false;
assert(client.num_subscriptions > 0); assert(num_subscriptions > 0);
client.subscriptions.erase(i); subscriptions.erase(i);
--client.num_subscriptions; --num_subscriptions;
idle_add(IDLE_SUBSCRIPTION); idle_add(IDLE_SUBSCRIPTION);
assert((client.num_subscriptions == 0) == assert((num_subscriptions == 0) ==
client.subscriptions.empty()); subscriptions.empty());
return true; return true;
} }
void void
client_unsubscribe_all(Client &client) Client::UnsubscribeAll()
{ {
client.subscriptions.clear(); subscriptions.clear();
client.num_subscriptions = 0; num_subscriptions = 0;
} }
bool bool
client_push_message(Client &client, const ClientMessage &msg) Client::PushMessage(const ClientMessage &msg)
{ {
if (client.messages.size() >= CLIENT_MAX_MESSAGES || if (messages.size() >= CLIENT_MAX_MESSAGES ||
!client.IsSubscribed(msg.GetChannel())) !IsSubscribed(msg.GetChannel()))
return false; return false;
if (client.messages.empty()) if (messages.empty())
client.IdleAdd(IDLE_MESSAGE); IdleAdd(IDLE_MESSAGE);
client.messages.push_back(msg); messages.push_back(msg);
return true; return true;
} }
/*
* Copyright (C) 2003-2013 The Music Player Daemon Project
* http://www.musicpd.org
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along
* with this program; if not, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
#ifndef MPD_CLIENT_SUBSCRIBE_HXX
#define MPD_CLIENT_SUBSCRIBE_HXX
#include "Compiler.h"
class Client;
class ClientMessage;
enum client_subscribe_result {
/** success */
CLIENT_SUBSCRIBE_OK,
/** invalid channel name */
CLIENT_SUBSCRIBE_INVALID,
/** already subscribed to this channel */
CLIENT_SUBSCRIBE_ALREADY,
/** too many subscriptions */
CLIENT_SUBSCRIBE_FULL,
};
enum client_subscribe_result
client_subscribe(Client &client, const char *channel);
bool
client_unsubscribe(Client &client, const char *channel);
void
client_unsubscribe_all(Client &client);
bool
client_push_message(Client &client, const ClientMessage &msg);
#endif
...@@ -19,7 +19,6 @@ ...@@ -19,7 +19,6 @@
#include "config.h" #include "config.h"
#include "MessageCommands.hxx" #include "MessageCommands.hxx"
#include "ClientSubscribe.hxx"
#include "Client.hxx" #include "Client.hxx"
#include "ClientList.hxx" #include "ClientList.hxx"
#include "Instance.hxx" #include "Instance.hxx"
...@@ -37,28 +36,29 @@ handle_subscribe(Client &client, gcc_unused int argc, char *argv[]) ...@@ -37,28 +36,29 @@ handle_subscribe(Client &client, gcc_unused int argc, char *argv[])
{ {
assert(argc == 2); assert(argc == 2);
switch (client_subscribe(client, argv[1])) { switch (client.Subscribe(argv[1])) {
case CLIENT_SUBSCRIBE_OK: case Client::SubscribeResult::OK:
return COMMAND_RETURN_OK; return COMMAND_RETURN_OK;
case CLIENT_SUBSCRIBE_INVALID: case Client::SubscribeResult::INVALID:
command_error(client, ACK_ERROR_ARG, command_error(client, ACK_ERROR_ARG,
"invalid channel name"); "invalid channel name");
return COMMAND_RETURN_ERROR; return COMMAND_RETURN_ERROR;
case CLIENT_SUBSCRIBE_ALREADY: case Client::SubscribeResult::ALREADY:
command_error(client, ACK_ERROR_EXIST, command_error(client, ACK_ERROR_EXIST,
"already subscribed to this channel"); "already subscribed to this channel");
return COMMAND_RETURN_ERROR; return COMMAND_RETURN_ERROR;
case CLIENT_SUBSCRIBE_FULL: case Client::SubscribeResult::FULL:
command_error(client, ACK_ERROR_EXIST, command_error(client, ACK_ERROR_EXIST,
"subscription list is full"); "subscription list is full");
return COMMAND_RETURN_ERROR; return COMMAND_RETURN_ERROR;
} }
/* unreachable */ /* unreachable */
return COMMAND_RETURN_OK; assert(false);
gcc_unreachable();
} }
enum command_return enum command_return
...@@ -66,7 +66,7 @@ handle_unsubscribe(Client &client, gcc_unused int argc, char *argv[]) ...@@ -66,7 +66,7 @@ handle_unsubscribe(Client &client, gcc_unused int argc, char *argv[])
{ {
assert(argc == 2); assert(argc == 2);
if (client_unsubscribe(client, argv[1])) if (client.Unsubscribe(argv[1]))
return COMMAND_RETURN_OK; return COMMAND_RETURN_OK;
else { else {
command_error(client, ACK_ERROR_NO_EXIST, command_error(client, ACK_ERROR_NO_EXIST,
...@@ -124,7 +124,7 @@ handle_send_message(Client &client, ...@@ -124,7 +124,7 @@ handle_send_message(Client &client,
bool sent = false; bool sent = false;
const ClientMessage msg(argv[1], argv[2]); const ClientMessage msg(argv[1], argv[2]);
for (const auto &c : *instance->client_list) for (const auto &c : *instance->client_list)
if (client_push_message(*c, msg)) if (c->PushMessage(msg))
sent = true; sent = true;
if (sent) if (sent)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment