import gc
import resource
import socket
import uuid

from pysandesh.sandesh_base import *
from sandesh_common.vns.constants import INSTANCE_ID_DEFAULT
from vnc_cfg_api_server.sandesh.traces.ttypes import IfmapTrace
from cfgm_common.uve.vnc_api.ttypes import VncApiCommon, VncApiConfigLog

import gevent
from gevent import monkey
monkey.patch_all()


def main():
    _sandesh = Sandesh()
    module_name = 'test_module'
    node_type_name = 'test_node_type_name'
    hostname = socket.gethostname()
    _sandesh.init_generator(module_name, hostname, node_type_name, INSTANCE_ID_DEFAULT,
            ['127.0.0.1:8086'], 'test_sandesh', -1, ['cfgm_common'])

##### Test logging
    _sandesh.set_logging_params(
        enable_local_log=True,
        category='',
        level=SandeshLevel.SYS_DEBUG,
        file='/tmp/sandesh.log',
        enable_syslog=False,
        syslog_facility=Sandesh._DEFAULT_SYSLOG_FACILITY)


##### Test trace buffer
#    _sandesh.trace_buffer_create(name="TestSandeshTrace", size=1000)
#    req_id = 'req-%s' %(str(uuid.uuid4()))
#    test_trace = IfmapTrace(request_id=req_id)
#    test_trace.operation = 'test_oper'
#    test_trace.body = 'test_body'
#    test_trace.trace_msg(name='test_sandesh', sandesh=_sandesh)


    prev = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss
    print 'Memory usage: %s (kb)' % prev
    gevent.sleep(5)
    now = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss
    print 'Memory usage: %s (kb)' % now
    for i in range(200):
        apiConfig = VncApiCommon()
        apiConfig.object_type='yo_obj_type'
        apiConfig.identifier_name='yo_fq_name_str'
        apiConfig.identifier_uuid = 'yo_uuid'
        apiConfig.operation = 'delete'
        log = VncApiConfigLog(api_log=apiConfig, sandesh=_sandesh)
        log.send(sandesh=_sandesh)

        gevent.sleep(1)
        gc.collect()

        now = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss
        if now > prev:
            print '(%i) Memory increase from %i to %i (kb) + %i' % (i, prev, now, now - prev)
        if now < prev:
            print '(%i) Memory decrease from %i to %i (kb) - %i' % (i, prev, now, prev - now)
        prev = now

    print 'Memory usage: %s (kb) -- END' % resource.getrusage(resource.RUSAGE_SELF).ru_maxrss
    gc.collect()
    print 'Memory usage: %s (kb) -- END' % resource.getrusage(resource.RUSAGE_SELF).ru_maxrss

if __name__ == "__main__":
    main()