metaproxy_1::filter::Limit::Impl Class Reference

Collaboration diagram for metaproxy_1::filter::Limit::Impl:

Public Member Functions
	Impl ()
	~Impl ()
void	process (metaproxy_1::Package &package)
void	configure (const xmlNode *ptr)

Private Attributes
boost::mutex	m_session_mutex
std::map< mp::Session, Limit::Ses * >	m_sessions
int	m_bw_max
int	m_pdu_max
int	m_search_max
int	m_max_record_retrieve

Detailed Description

Definition at line 44 of file filter_limit.cpp.

Constructor & Destructor Documentation

Impl()

mp::filter::Limit::Impl::Impl

(

)

Definition at line 85 of file filter_limit.cpp.

                  : m_bw_max(0), m_pdu_max(0), m_search_max(0),
                          m_max_record_retrieve(0)
{
}

~Impl()

mp::filter::Limit::Impl::~Impl

(

)

Definition at line 90 of file filter_limit.cpp.

{
}

Member Function Documentation

configure()

void mp::filter::Limit::Impl::configure

(

const xmlNode *

ptr

)

Definition at line 94 of file filter_limit.cpp.

{
    for (ptr = ptr->children; ptr; ptr = ptr->next)
    {
        if (ptr->type != XML_ELEMENT_NODE)
            continue;
        if (!strcmp((const char *) ptr->name, "limit"))
        {
            const struct _xmlAttr *attr;
            for (attr = ptr->properties; attr; attr = attr->next)
            {
                if (!strcmp((const char *) attr->name, "bandwidth"))
                    m_bw_max = mp::xml::get_int(attr->children, 0);
                else if (!strcmp((const char *) attr->name, "pdu"))
                    m_pdu_max = mp::xml::get_int(attr->children, 0);
                else if (!strcmp((const char *) attr->name, "search"))
                    m_search_max = mp::xml::get_int(attr->children, 0);
                else if (!strcmp((const char *) attr->name, "retrieve"))
                    m_max_record_retrieve =
                        mp::xml::get_int(attr->children, 0);
                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));
        }
    }
}

process()

void mp::filter::Limit::Impl::process

(

metaproxy_1::Package &

package

)

Definition at line 129 of file filter_limit.cpp.

{
    int sz = 0;
    {
        boost::mutex::scoped_lock scoped_lock(m_session_mutex);
 
        yf::Limit::Ses *ses = 0;
 
        std::map<mp::Session,yf::Limit::Ses *>::iterator it =
            m_sessions.find(package.session());
        if (it != m_sessions.end())
            ses = it->second;
        else
        {
            ses = new yf::Limit::Ses;
            m_sessions[package.session()] = ses;
        }
 
 
        Z_GDU *gdu = package.request().get();
        if (gdu && gdu->which == Z_GDU_Z3950)
        {
            sz += package.request().get_size();
            // we're getting a Z39.50 package
            Z_APDU *apdu = gdu->u.z3950;
            if (apdu->which == Z_APDU_searchRequest)
                ses->search_stat.add_bytes(1);
            if (m_max_record_retrieve)
            {
                if (apdu->which == Z_APDU_presentRequest)
                {
                    Z_PresentRequest *pr = apdu->u.presentRequest;
                    if (pr->numberOfRecordsRequested &&
                        *pr->numberOfRecordsRequested > m_max_record_retrieve)
                        *pr->numberOfRecordsRequested = m_max_record_retrieve;
                }
            }
        }
    }
    package.move();
    int reduce = 0;
    {
        boost::mutex::scoped_lock scoped_lock(m_session_mutex);
 
        yf::Limit::Ses *ses = 0;
 
        std::map<mp::Session,yf::Limit::Ses *>::iterator it =
            m_sessions.find(package.session());
        if (it != m_sessions.end())
            ses = it->second;
        else
        {
            ses = new yf::Limit::Ses;
            m_sessions[package.session()] = ses;
        }
 
        sz += package.response().get_size();
 
        ses->bw_stat.add_bytes(sz);
        ses->pdu_stat.add_bytes(1);
 
        int bw_total = ses->bw_stat.get_total();
        int pdu_total = ses->pdu_stat.get_total();
        int search_total = ses->search_stat.get_total();
 
        if (m_search_max)
            reduce += search_total / m_search_max;
        if (m_bw_max)
            reduce += (bw_total/m_bw_max);
        if (m_pdu_max)
        {
            if (pdu_total > m_pdu_max)
            {
                int nreduce = (m_pdu_max >= 60) ? 1 : 60/m_pdu_max;
                reduce = (reduce > nreduce) ? reduce : nreduce;
            }
        }
        if (package.session().is_closed())
        {
            m_sessions.erase(package.session());
            delete ses;
        }
    }
    if (reduce)
    {
        std::ostringstream os;
        os << "S" << " "
           << package
           << " sleeping "
           << reduce
           << " seconds";
        yaz_log(YLOG_LOG, "%s", os.str().c_str());
#ifdef WIN32
        Sleep(reduce * 1000);
#else
        sleep(reduce);
#endif
    }
}

Member Data Documentation

m_bw_max

int metaproxy_1::filter::Limit::Impl::m_bw_max

private

Definition at line 53 of file filter_limit.cpp.

m_max_record_retrieve

int metaproxy_1::filter::Limit::Impl::m_max_record_retrieve

private

Definition at line 56 of file filter_limit.cpp.

m_pdu_max

int metaproxy_1::filter::Limit::Impl::m_pdu_max

private

Definition at line 54 of file filter_limit.cpp.

m_search_max

int metaproxy_1::filter::Limit::Impl::m_search_max

private

Definition at line 55 of file filter_limit.cpp.

m_session_mutex

boost::mutex metaproxy_1::filter::Limit::Impl::m_session_mutex

private

Definition at line 51 of file filter_limit.cpp.

m_sessions

std::map<mp::Session,Limit::Ses *> metaproxy_1::filter::Limit::Impl::m_sessions

private

Definition at line 52 of file filter_limit.cpp.

---

The documentation for this class was generated from the following file:

• filter_limit.cpp