socketMerge.py

#!/usr/local/bin/python
#
#                               Copyright 2006
#                                     by
#                        The Board of Trustees of the
#                     Leland Stanford Junior University.
#                            All rights reserved.
#

__facility__ = "Online"
__abstract__ = "Program to merge data from network sockets"
__author__   = "R. Claus <Claus@SLAC.Stanford.edu> SLAC - GLAST LAT I&T/Online"
__date__     = "3/30/2006"
__updated__  = ("$Date: 2006/07/14 22:34:29 $").split(' ')[1]
__version__  = "$Revision: 1.1 $"
__release__  = "$Name: R04-12-00 $"
__credits__  = "SLAC"

import LATTE.copyright_SLAC

import logging         as     log
import sys
import time
import struct

import LDF

from   LATTE.merger.geb import Geb

from   ancillarySocket  import AncillaryContributor
from   ldfSocket        import LdfContributor


class Merger(Geb):
  def __init__(self, contributors, filename):
    Geb.__init__(self, contributors)
    self.__file = file(filename, 'wb')

  def dispose(self, (statuses, buffer)):
    """!\brief Method called by the base class to dispose of an event.
    """
    self.__file.write(buffer)

  def close(self):
    """!\brief Method called by the base class to close the output file.
    """
    self.__file.close()


def getHostPort(option):
  opts = option.split(':')
  host = opts[0]
  if len(opts) == 1:
    port = 2345
  else:
    port = int(opts[1])
  return host, port


def main(options):
  ancillaryContributor = AncillaryContributor(getHostPort(options.ancillaryHost))
  ldfContributor       = LdfContributor(getHostPort(options.ldfHost))
  merger = Merger([ancillaryContributor, ldfContributor], options.outFile)
  #merger = Merger([ancillaryContributor], options.outFile)
  #merger = Merger([ldfContributor], options.outFile)

  try:
    merger.start()

    # Wait until an error occurs or the user kills us
    while(True):  time.sleep(24*3600)

  except KeyboardInterrupt:
    pass
  except Exception, e:
    log.exception(sys.argv[0] + ":\n" + str(e))

  # Shutdown the builder
  merger.shutdown()


def usage():
  """!\brief Return a program usage message
  """
  return "\n\n%s merges, or builds, events from sockets\n" % (sys.argv[0])


if __name__ == "__main__":
  # Options, first list mandatory, second list optional
  from LATTE.tools.gOptions import Options

  options = Options(["ancillaryHost", "ldfHost", "outFile"])
  try:
   options.parse()
  except Exception, msg:
   options.usage(usage())
   raise Exception, msg

  main(options)

---