send_monitoring_event: support for default event and gomastats

infra_libs/event_mon/config.py

 # Copyright 2015 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import logging
 import socket
 
 import infra_libs
 from infra_libs.event_mon.protos.chrome_infra_log_pb2 import (
   ChromeInfraEvent, ServiceEvent)
@@ skipping 147 matching lines @@
       _router = ev_router._TextStreamRouter()
     elif run_type == 'file':
       _router = ev_router._LocalFileRouter(output_file,
                                            dry_run=dry_run)
     else:
       _router = ev_router._HttpRouter(_cache,
                                       ENDPOINTS.get(run_type),
                                       dry_run=dry_run)
 
 
+def get_default_event():
+  """Returns a copy of the default event."""
+
+  # We return a copy here to tell people not to modify the event directly.
+  ret = ChromeInfraEvent()
+  ret.CopyFrom(_cache['default_event'])
+  return ret
+
+
 def set_default_event(event):
   """Change the default ChromeInfraEvent used to compute all events.
 
   Args:
     event (ChromeInfraEvent): default event
   """
   # Here we raise an exception because failing to set the default event
   # could lead to invalid data in the database.
   if not isinstance(event, ChromeInfraEvent):
     msg = ('A ChromeInfraEvent is required as the default event. Got %s' %
            type(event))
     logging.error(msg)
     raise TypeError(msg)
 
   _cache['default_event'] = event
 
 
 def close():
   """Reset the state.
 
   Call this right before exiting the program.
 
   Returns:
     success (bool): False if an error occured
   """
   global _router, _cache
   _router = None
   _cache = {}
   return True