# copyright 2014 LOGILAB S.A. (Paris, FRANCE), all rights reserved.
# contact http://www.logilab.fr/ -- mailto:contact@logilab.fr
#
# This file is part of CubicWeb.
#
# CubicWeb is free software: you can redistribute it and/or modify it under the
# terms of the GNU Lesser General Public License as published by the Free
# Software Foundation, either version 2.1 of the License, or (at your option)
# any later version.
#
# CubicWeb 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 Lesser General Public License for more
# details.
#
# You should have received a copy of the GNU Lesser General Public License along
# with CubicWeb.  If not, see <http://www.gnu.org/licenses/>.

"""looping task for dumping instance's stats in a file
"""

__docformat__ = "restructuredtext en"

from datetime import datetime
import json

from cubicweb.server import hook

class LogStatsStartHook(hook.Hook):
    """register task to regularly dump instance's stats in a file

    data are stored as one json entry per row
    """
    __regid__ = 'cubicweb.hook.logstats.start'
    events = ('server_startup',)

    def __call__(self):
        interval = self.repo.config.get('logstat-interval', 0)
        if interval <= 0:
            return            

        def dump_stats(repo):
            statsfile = repo.config.get('logstat-file')
            with repo.internal_cnx() as cnx:
                stats = cnx.call_service('repo_stats')
                gcstats = cnx.call_service('repo_gc_stats', nmax=5)
                
            allstats = {'resources': stats,
                        'memory': gcstats,
                        'timestamp': datetime.utcnow().isoformat(),
                       }
            try:
                with open(statsfile, 'ab') as ofile:
                    json.dump(allstats, ofile)
                    ofile.write('\n')
            except IOError:
                repo.warning('Cannot open stats file for writing: %s', statsfile)
                    
        self.repo.looping_task(interval, dump_stats, self.repo)