-/* $Id: filter_log.cpp,v 1.11 2006-01-04 11:55:31 adam Exp $
- Copyright (c) 2005, Index Data.
+/* $Id: filter_log.cpp,v 1.24 2006-08-30 08:35:47 marc Exp $
+ Copyright (c) 2005-2006, Index Data.
-%LICENSE%
+ See the LICENSE file for details
*/
#include "config.hpp"
-
-#include "filter.hpp"
-#include "router.hpp"
#include "package.hpp"
#include <string>
+#include <sstream>
#include <boost/thread/mutex.hpp>
#include <boost/date_time/posix_time/posix_time.hpp>
+#include "gduutil.hpp"
#include "util.hpp"
+#include "xmlutil.hpp"
#include "filter_log.hpp"
+#include <fstream>
#include <yaz/zgdu.h>
+#include <yaz/wrbuf.h>
+#include <yaz/querytowrbuf.h>
+
-namespace yf = yp2::filter;
+namespace mp = metaproxy_1;
+namespace yf = metaproxy_1::filter;
-namespace yp2 {
+namespace metaproxy_1 {
namespace filter {
+ class Log::LFile {
+ public:
+ boost::mutex m_mutex;
+ std::string m_fname;
+ std::ofstream fout;
+ std::ostream &out;
+ LFile(std::string fname);
+ LFile(std::string fname, std::ostream &use_this);
+ };
+ typedef boost::shared_ptr<Log::LFile> LFilePtr;
class Log::Rep {
friend class Log;
- static boost::mutex m_log_mutex;
- std::string m_msg;
+ Rep();
+ void openfile(const std::string &fname);
+ std::string m_msg_config;
+ LFilePtr m_file;
+ bool m_access;
+ bool m_req_apdu;
+ bool m_res_apdu;
+ bool m_req_session;
+ bool m_res_session;
+ bool m_init_options;
};
+ // Only used during configure stage (no threading)
+ static std::list<LFilePtr> filter_log_files;
}
}
-boost::mutex yf::Log::Rep::m_log_mutex;
+yf::Log::Rep::Rep()
+{
+ m_req_apdu = true;
+ m_res_apdu = true;
+ m_req_session = true;
+ m_res_session = true;
+ m_init_options = false;
+ openfile("");
+}
yf::Log::Log(const std::string &x) : m_p(new Rep)
{
- m_p->m_msg = x;
+ m_p->m_msg_config = x;
}
yf::Log::Log() : m_p(new Rep)
yf::Log::~Log() {}
-void yf::Log::process(Package &package) const
+void stream_write(ODR o, void *handle, int type, const char *buf, int len)
+{
+ yf::Log::LFile *lfile = (yf::Log::LFile*) handle;
+ lfile->out.write(buf, len);
+}
+
+void option_write(const char *name, void *handle)
+{
+ yf::Log::LFile *lfile = (yf::Log::LFile*) handle;
+ lfile->out << " " << name;
+}
+
+void yf::Log::process(mp::Package &package) const
{
Z_GDU *gdu;
boost::posix_time::ptime receive_time
= boost::posix_time::microsec_clock::local_time();
+
+ //std::ostringstream msg_request;
+ //std::ostringstream msg_request_2;
+ //std::ostringstream msg_response;
+ //std::ostringstream msg_response_2;
+
// scope for locking Ostream
{
- boost::mutex::scoped_lock scoped_lock(Rep::m_log_mutex);
- std::cout << receive_time << " " << m_p->m_msg;
- std::cout << " request id=" << package.session().id();
- std::cout << " close="
- << (package.session().is_closed() ? "yes" : "no")
- << "\n";
- gdu = package.request().get();
- if (gdu)
+ boost::mutex::scoped_lock scoped_lock(m_p->m_file->m_mutex);
+
+
+ if (m_p->m_access)
{
- yp2::odr odr(ODR_PRINT);
- z_GDU(odr, &gdu, 0, 0);
+ gdu = package.request().get();
+ if (gdu)
+ {
+ m_p->m_file->out
+ << m_p->m_msg_config << " "
+ << receive_time << " "
+ << package << " "
+ << "00:00:00.000000" << " "
+ << *gdu
+ << "\n";
+ }
}
- }
+ if (m_p->m_req_session)
+ {
+ m_p->m_file->out << receive_time << " " << m_p->m_msg_config;
+ m_p->m_file->out << " request id=" << package.session().id();
+ m_p->m_file->out << " close="
+ << (package.session().is_closed() ? "yes" : "no")
+ << "\n";
+ }
+
+ if (m_p->m_init_options)
+ {
+ gdu = package.request().get();
+ if (gdu && gdu->which == Z_GDU_Z3950 &&
+ gdu->u.z3950->which == Z_APDU_initRequest)
+ {
+ m_p->m_file->out << receive_time << " " << m_p->m_msg_config;
+ m_p->m_file->out << " init options:";
+ yaz_init_opt_decode(gdu->u.z3950->u.initRequest->options,
+ option_write, m_p->m_file.get());
+ m_p->m_file->out << "\n";
+ }
+ }
+
+ if (m_p->m_req_apdu)
+ {
+ gdu = package.request().get();
+ if (gdu)
+ {
+ mp::odr odr(ODR_PRINT);
+ odr_set_stream(odr, m_p->m_file.get(), stream_write, 0);
+ z_GDU(odr, &gdu, 0, 0);
+ }
+ }
+ m_p->m_file->out.flush();
+ }
+
// unlocked during move
package.move();
// scope for locking Ostream
{
- boost::mutex::scoped_lock scoped_lock(Rep::m_log_mutex);
- std::cout << send_time << " " << m_p->m_msg;
- std::cout << " response id=" << package.session().id();
- std::cout << " close="
- << (package.session().is_closed() ? "yes " : "no ")
- << "duration=" << duration
- << "\n";
- //<< "duration=" << duration.total_seconds()
- // << "." << duration.fractional_seconds()
- // << "\n";
- gdu = package.response().get();
- if (gdu)
+ boost::mutex::scoped_lock scoped_lock(m_p->m_file->m_mutex);
+
+ if (m_p->m_access)
+ {
+ gdu = package.response().get();
+ if (gdu)
+ {
+ m_p->m_file->out
+ << m_p->m_msg_config << " "
+ << send_time << " "
+ << package << " "
+ << duration << " "
+ << *gdu
+ << "\n";
+ }
+ }
+
+ if (m_p->m_res_session)
+ {
+ m_p->m_file->out << send_time << " " << m_p->m_msg_config;
+ m_p->m_file->out << " response id=" << package.session().id();
+ m_p->m_file->out << " close="
+ << (package.session().is_closed() ? "yes " : "no ")
+ << "duration=" << duration
+ << "\n";
+ }
+
+ if (m_p->m_init_options)
+ {
+ gdu = package.response().get();
+ if (gdu && gdu->which == Z_GDU_Z3950 &&
+ gdu->u.z3950->which == Z_APDU_initResponse)
+ {
+ m_p->m_file->out << receive_time << " " << m_p->m_msg_config;
+ m_p->m_file->out << " init options:";
+ yaz_init_opt_decode(gdu->u.z3950->u.initResponse->options,
+ option_write, m_p->m_file.get());
+ m_p->m_file->out << "\n";
+ }
+ }
+
+ if (m_p->m_res_apdu)
{
- yp2::odr odr(ODR_PRINT);
- z_GDU(odr, &gdu, 0, 0);
+ gdu = package.response().get();
+ if (gdu)
+ {
+ mp::odr odr(ODR_PRINT);
+ odr_set_stream(odr, m_p->m_file.get(), stream_write, 0);
+ z_GDU(odr, &gdu, 0, 0);
+ }
}
+
+ m_p->m_file->out.flush();
}
}
-static yp2::filter::Base* filter_creator()
+yf::Log::LFile::LFile(std::string fname) :
+ m_fname(fname), fout(fname.c_str()), out(fout)
+{
+}
+
+yf::Log::LFile::LFile(std::string fname, std::ostream &use_this) :
+ m_fname(fname), out(use_this)
{
- return new yp2::filter::Log;
+}
+
+void yf::Log::Rep::openfile(const std::string &fname)
+{
+ std::list<LFilePtr>::const_iterator it = filter_log_files.begin();
+ for (; it != filter_log_files.end(); it++)
+ {
+ if ((*it)->m_fname == fname)
+ {
+ m_file = *it;
+ return;
+ }
+ }
+ // open stdout for empty file
+ LFilePtr newfile(fname.length() == 0
+ ? new LFile(fname, std::cout)
+ : new LFile(fname));
+ filter_log_files.push_back(newfile);
+ m_file = newfile;
+}
+
+void yf::Log::configure(const xmlNode *ptr)
+{
+ for (ptr = ptr->children; ptr; ptr = ptr->next)
+ {
+ if (ptr->type != XML_ELEMENT_NODE)
+ continue;
+ if (!strcmp((const char *) ptr->name, "message"))
+ m_p->m_msg_config = mp::xml::get_text(ptr);
+ else if (!strcmp((const char *) ptr->name, "filename"))
+ {
+ std::string fname = mp::xml::get_text(ptr);
+ m_p->openfile(fname);
+ }
+ else if (!strcmp((const char *) ptr->name, "category"))
+ {
+ const struct _xmlAttr *attr;
+ for (attr = ptr->properties; attr; attr = attr->next)
+ {
+ if (!strcmp((const char *) attr->name,
+ "access"))
+ m_p->m_access =
+ mp::xml::get_bool(attr->children, true);
+ else if (!strcmp((const char *) attr->name, "request-apdu"))
+ m_p->m_req_apdu = mp::xml::get_bool(attr->children, true);
+ else if (!strcmp((const char *) attr->name, "response-apdu"))
+ m_p->m_res_apdu = mp::xml::get_bool(attr->children, true);
+ else if (!strcmp((const char *) attr->name, "apdu"))
+ {
+ m_p->m_req_apdu = mp::xml::get_bool(attr->children, true);
+ m_p->m_res_apdu = m_p->m_req_apdu;
+ }
+ else if (!strcmp((const char *) attr->name,
+ "request-session"))
+ m_p->m_req_session =
+ mp::xml::get_bool(attr->children, true);
+ else if (!strcmp((const char *) attr->name,
+ "response-session"))
+ m_p->m_res_session =
+ mp::xml::get_bool(attr->children, true);
+ else if (!strcmp((const char *) attr->name,
+ "session"))
+ {
+ m_p->m_req_session =
+ mp::xml::get_bool(attr->children, true);
+ m_p->m_res_session = m_p->m_req_session;
+ }
+ else if (!strcmp((const char *) attr->name,
+ "init-options"))
+ m_p->m_init_options =
+ mp::xml::get_bool(attr->children, true);
+ else
+ throw mp::filter::FilterException(
+ "Bad attribute " + std::string((const char *)
+ attr->name));
+ }
+ }
+ else
+ {
+ throw mp::filter::FilterException("Bad element "
+ + std::string((const char *)
+ ptr->name));
+ }
+ }
+}
+
+static mp::filter::Base* filter_creator()
+{
+ return new mp::filter::Log;
}
extern "C" {
- struct yp2_filter_struct yp2_filter_log = {
+ struct metaproxy_1_filter_struct metaproxy_1_filter_log = {
0,
"log",
filter_creator
};
}
-
/*
* Local variables:
* c-basic-offset: 4