snapcast/server/serverSession.h
2015-08-12 00:22:56 +02:00

103 lines
2.3 KiB
C++

/***
This file is part of snapcast
Copyright (C) 2015 Johannes Pohl
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 3 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, see <http://www.gnu.org/licenses/>.
***/
#ifndef SERVER_CONNECTION_H
#define SERVER_CONNECTION_H
#include <string>
#include <thread>
#include <atomic>
#include <mutex>
#include <memory>
#include <boost/asio.hpp>
#include <condition_variable>
#include <set>
#include "message/message.h"
#include "common/queue.h"
using boost::asio::ip::tcp;
class ServerSession;
/// Interface: callback for a received message.
class MessageReceiver
{
public:
virtual void onMessageReceived(ServerSession* connection, const msg::BaseMessage& baseMessage, char* buffer) = 0;
};
/// Endpoint for a connected client.
/**
* Endpoint for a connected client.
* Messages are sent to the client with the "send" method.
* Received messages from the client are passed to the MessageReceiver callback
*/
class ServerSession
{
public:
ServerSession(MessageReceiver* receiver, std::shared_ptr<tcp::socket> socket);
~ServerSession();
void start();
void stop();
bool send(const msg::BaseMessage* message) const;
void add(const std::shared_ptr<const msg::BaseMessage>& message);
bool active() const
{
return active_;
}
void setStreamActive(bool active)
{
streamActive_ = active;
}
void setBufferMs(size_t bufferMs)
{
bufferMs_ = bufferMs;
}
protected:
void socketRead(void* _to, size_t _bytes);
void getNextMessage();
void reader();
void writer();
std::atomic<bool> active_;
std::atomic<bool> streamActive_;
mutable std::mutex mutex_;
std::thread* readerThread_;
std::thread* writerThread_;
std::shared_ptr<tcp::socket> socket_;
MessageReceiver* messageReceiver_;
Queue<std::shared_ptr<const msg::BaseMessage>> messages_;
size_t bufferMs_;
};
#endif