mirror of
https://github.com/zeek/zeek.git
synced 2025-10-09 18:18:19 +00:00
3415b5fcbe
* add support for &type_column for reading * add basic tests for logging & input * clean up a bit * add support for tables for reading (untested)
59 lines
1.2 KiB
C++
59 lines
1.2 KiB
C++
// See the file "COPYING" in the main distribution directory for copyright.
|
|
//
|
|
|
|
#ifndef INPUT_READERS_POSTGRES_H
|
|
#define INPUT_READERS_POSTGRES_H
|
|
|
|
#include "config.h"
|
|
|
|
#ifdef USE_SQLITE
|
|
|
|
#include <iostream>
|
|
#include <vector>
|
|
|
|
#include "../ReaderBackend.h"
|
|
#include "../../threading/AsciiInputOutput.h"
|
|
|
|
#include "sqlite3.h"
|
|
|
|
|
|
namespace input { namespace reader {
|
|
|
|
class SQLite : public ReaderBackend {
|
|
public:
|
|
SQLite(ReaderFrontend* frontend);
|
|
~SQLite();
|
|
|
|
static ReaderBackend* Instantiate(ReaderFrontend* frontend) { return new SQLite(frontend); }
|
|
|
|
protected:
|
|
virtual bool DoInit(const ReaderInfo& info, int arg_num_fields, const threading::Field* const* fields);
|
|
|
|
virtual void DoClose();
|
|
|
|
virtual bool DoUpdate();
|
|
|
|
virtual bool DoHeartbeat(double network_time, double current_time) { return true; }
|
|
private:
|
|
bool checkError(int code);
|
|
|
|
threading::Value* EntryToVal(sqlite3_stmt *st, const threading::Field *field, int pos, int subpos);
|
|
|
|
const threading::Field* const * fields; // raw mapping
|
|
unsigned int num_fields;
|
|
int mode;
|
|
bool started;
|
|
string query;
|
|
sqlite3 *db;
|
|
sqlite3_stmt *st;
|
|
AsciiInputOutput* io;
|
|
};
|
|
|
|
|
|
}
|
|
}
|
|
|
|
#endif /* USE_SQLITE */
|
|
|
|
#endif /* INPUT_READERS_POSTGRES_H */
|
|
|