159 lines
5.1 KiB
C++
159 lines
5.1 KiB
C++
/***************************************************************************
|
|
* Copyright (C) gempa GmbH *
|
|
* All rights reserved. *
|
|
* Contact: gempa GmbH (seiscomp-dev@gempa.de) *
|
|
* *
|
|
* GNU Affero General Public License Usage *
|
|
* This file may be used under the terms of the GNU Affero *
|
|
* Public License version 3.0 as published by the Free Software Foundation *
|
|
* and appearing in the file LICENSE included in the packaging of this *
|
|
* file. Please review the following information to ensure the GNU Affero *
|
|
* Public License version 3.0 requirements will be met: *
|
|
* https://www.gnu.org/licenses/agpl-3.0.html. *
|
|
* *
|
|
* Other Usage *
|
|
* Alternatively, this file may be used in accordance with the terms and *
|
|
* conditions contained in a signed written agreement between you and *
|
|
* gempa GmbH. *
|
|
***************************************************************************/
|
|
|
|
|
|
#ifndef SEISCOMP_IO_RECORDSTREAM_WS_H
|
|
#define SEISCOMP_IO_RECORDSTREAM_WS_H
|
|
|
|
#include <string>
|
|
#include <set>
|
|
#include <sstream>
|
|
|
|
#include <seiscomp/core.h>
|
|
#include <seiscomp/core/datetime.h>
|
|
#include <seiscomp/io/recordstream.h>
|
|
#include <seiscomp/io/socket.h>
|
|
#include <seiscomp/io/httpsocket.h>
|
|
#include <seiscomp/io/recordstream/streamidx.h>
|
|
|
|
|
|
struct _bson_t;
|
|
typedef bson_t _bson_t;
|
|
|
|
|
|
namespace Seiscomp {
|
|
namespace RecordStream {
|
|
|
|
|
|
class SC_SYSTEM_CORE_API HMBQueue {
|
|
public:
|
|
//! C'tor
|
|
HMBQueue();
|
|
|
|
//! Destructor
|
|
virtual ~HMBQueue();
|
|
|
|
//! Adds the given stream
|
|
void addStream(std::string loc, std::string cha,
|
|
const OPT(Core::Time) &stime,
|
|
const OPT(Core::Time) &etime);
|
|
|
|
//! Sets the sequence number
|
|
void setSequenceNumber(int64_t seq);
|
|
|
|
//! Removes all entries
|
|
void clear();
|
|
|
|
//! Returns a BSON document
|
|
bson_t* toBSON() const;
|
|
|
|
private:
|
|
OPT(Core::Time) _stime;
|
|
OPT(Core::Time) _etime;
|
|
int64_t _seq;
|
|
std::set<std::string> _topics;
|
|
};
|
|
|
|
|
|
template<typename SocketType>
|
|
class SC_SYSTEM_CORE_API HMBConnection : public Seiscomp::IO::RecordStream {
|
|
//DECLARE_SC_CLASS(HMBConnection<SocketType>);
|
|
|
|
public:
|
|
//! C'tor
|
|
HMBConnection();
|
|
|
|
//! Initializing Constructor
|
|
HMBConnection(std::string serverloc);
|
|
|
|
//! Destructor
|
|
~HMBConnection() override;
|
|
|
|
//! The recordtype cannot be selected when using an HMB
|
|
//! connection. It will always create MiniSeed records
|
|
bool setRecordType(const char *type) override;
|
|
|
|
//! Initialize the HMB connection.
|
|
bool setSource(const std::string &source) override;
|
|
|
|
//! Supply user credentials
|
|
//! Adds the given stream to the server connection description
|
|
bool addStream(const std::string &networkCode,
|
|
const std::string &stationCode,
|
|
const std::string &locationCode,
|
|
const std::string &channelCode) override;
|
|
|
|
//! Adds the given stream to the server connection description
|
|
bool addStream(const std::string &networkCode,
|
|
const std::string &stationCode,
|
|
const std::string &locationCode,
|
|
const std::string &channelCode,
|
|
const OPT(Core::Time) &startTime,
|
|
const OPT(Core::Time) &endTime) override;
|
|
|
|
//! Adds the given start time to the server connection description
|
|
bool setStartTime(const OPT(Core::Time) &stime) override;
|
|
|
|
//! Adds the given end time to the server connection description
|
|
bool setEndTime(const OPT(Core::Time) &etime) override;
|
|
|
|
//! Sets timeout
|
|
bool setTimeout(int seconds) override;
|
|
|
|
//! Terminates the HMB connection.
|
|
void close() override;
|
|
|
|
Record *next() override;
|
|
|
|
|
|
private:
|
|
//! Removes all stream list, time window, etc. -entries from the connection description object.
|
|
bool clear();
|
|
|
|
//! Reconnects a terminated HMB connection.
|
|
bool reconnect();
|
|
|
|
|
|
private:
|
|
IO::HttpSocket<SocketType> _sock;
|
|
std::string _serverHost;
|
|
std::string _serverPath;
|
|
std::string _user;
|
|
std::string _password;
|
|
std::set<Seiscomp::RecordStream::StreamIdx> _streams;
|
|
OPT(Core::Time) _stime;
|
|
OPT(Core::Time) _etime;
|
|
std::map<std::string, HMBQueue> _queues;
|
|
std::string _sid;
|
|
std::string _cid;
|
|
bool _readingData;
|
|
|
|
std::string bsonGetString(const bson_t *bson, const char *key);
|
|
int64_t bsonGetInt(const bson_t *bson, const char *key);
|
|
void bsonGetBlob(const bson_t *bson, const char *key, const void **data, int *data_len);
|
|
void initSession();
|
|
std::string receive();
|
|
};
|
|
|
|
}
|
|
}
|
|
|
|
#endif
|
|
|