def __init__( self, configFile, instanceId ):
# Setting the signal handler to trigger the stop event which
# is interpreted by each actor implementation
global timeToStopEvent
gevent.signal( signal.SIGQUIT, _stopAllActors )
gevent.signal( signal.SIGINT, _stopAllActors )
self._initLogging()
self.instanceId = instanceId
self.log( "Initializing" )
self.stopEvent = timeToStopEvent
self.actors = {}
self.py_beach_dir = None
self.configFilePath = configFile
self.configFile = None
with open( self.configFilePath, 'r' ) as f:
self.configFile = yaml.load( f )
self.py_beach_dir = os.path.dirname( os.path.abspath( __file__ ) )
os.chdir( os.path.dirname( os.path.abspath( self.configFilePath ) ) )
self.codeDirectory = os.path.abspath( self.configFile.get( 'code_directory', './' ) )
self.opsSocket = _ZMREP( 'ipc:///tmp/py_beach_instance_%s' % instanceId, isBind = True )
self.log( "Listening for ops on %s" % ( 'ipc:///tmp/py_beach_instance_%s' % instanceId, ) )
self.hostOpsPort = self.configFile.get( 'ops_port', 4999 )
self.hostOpsSocket = _ZMREP( 'tcp://127.0.0.1:%d' % self.hostOpsPort, isBind = False )
ActorHandle._setHostDirInfo( self.configFile.get( 'directory_port',
'ipc:///tmp/py_beach_directory_port' ) )
gevent.spawn( self.svc_receiveTasks )
gevent.spawn( self.svc_monitorActors )
self.log( "Now open to actors" )
timeToStopEvent.wait()
self.log( "Exiting, stopping all actors." )
for actor in self.actors.values():
actor.stop()
gevent.joinall( self.actors.values() )
self.log( "All Actors exiting, exiting." )
def __init__( self, configFile, logging_level, logging_dest, iface = None ):
# Setting the signal handler to trigger the stop event
global timeToStopEvent
gevent.signal( signal.SIGQUIT, _stop )
gevent.signal( signal.SIGINT, _stop )
self._logger = None
self._log_level = logging_level
self._log_dest = logging_dest
self._initLogging( logging_level, logging_dest )
self.stopEvent = timeToStopEvent
self.py_beach_dir = None
self.configFilePath = os.path.abspath( configFile )
self.configFile = None
self.directory = {}
self.reverseDir = {}
self.tombstones = {}
self.actorInfo = {}
self.ports_available = Set()
self.nProcesses = 0
self.processes = []
self.initialProcesses = False
self.seedNodes = []
self.directoryPort = None
self.opsPort = 0
self.opsSocket = None
self.port_range = ( 0, 0 )
self.interface = None
self.ifaceIp4 = None
self.nodes = {}
self.peer_keepalive_seconds = 0
self.instance_keepalive_seconds = 0
self.tombstone_culling_seconds = 0
self.isActorChanged = gevent.event.Event()
self.isInstanceChanged = gevent.event.Event()
self.isTombstoneChanged = gevent.event.Event()
self.dirLock = RWLock()
# Cleanup potentially old sockets
os.system( 'rm /tmp/py_beach*' )
# Load default configs
with open( self.configFilePath, 'r' ) as f:
self.configFile = yaml.load( f )
self.py_beach_dir = os.path.dirname( os.path.abspath( __file__ ) )
os.chdir( os.path.dirname( os.path.abspath( self.configFilePath ) ) )
self.private_key = self.configFile.get( 'private_key', None )
if self.private_key is not None:
with open( self.private_key, 'r' ) as f:
key_path = self.private_key
self.private_key = f.read()
self._log( "Using shared key: %s" % key_path )
self.admin_token = self.configFile.get( 'admin_token', None )
self.nProcesses = self.configFile.get( 'n_processes', 0 )
if self.nProcesses == 0:
self.nProcesses = multiprocessing.cpu_count()
self._log( "Using %d instances per node" % self.nProcesses )
if iface is not None:
self.interface = iface
self.ifaceIp4 = _getIpv4ForIface( self.interface )
if self.ifaceIp4 is None:
self._logCritical( "Could not use iface %s (from cli)." % self.interface )
sys.exit( -1 )
else:
self.interface = self.configFile.get( 'interface', None )
if self.interface is not None:
self.ifaceIp4 = _getIpv4ForIface( self.interface )
if self.ifaceIp4 is None:
self._logCritical( "Could not use iface %s (from config)." % self.interface )
sys.exit( -1 )
# Building a list of interfaces to auto-detect
defaultInterfaces = _getPublicInterfaces()
while self.ifaceIp4 is None and 0 != len( defaultInterfaces ):
self.interface = defaultInterfaces.pop()
self.ifaceIp4 = _getIpv4ForIface( self.interface )
if self.ifaceIp4 is None:
self._log( "Failed to use interface %s." % self.interface )
if self.ifaceIp4 is None:
self._logCritical( "Could not find an interface to use." )
sys.exit( -1 )
self.seedNodes = self.configFile.get( 'seed_nodes', [] )
if 0 == len( self.seedNodes ):
self.seedNodes.append( self.ifaceIp4 )
for s in self.seedNodes:
self._log( "Using seed node: %s" % s )
self.directoryPort = _ZMREP( self.configFile.get( 'directory_port',
'ipc:///tmp/py_beach_directory_port' ),
isBind = True,
private_key = self.private_key )
self.opsPort = self.configFile.get( 'ops_port', 4999 )
self.opsSocket = _ZMREP( 'tcp://%s:%d' % ( self.ifaceIp4, self.opsPort ),
isBind = True,
private_key = self.private_key )
self._log( "Listening for ops on %s:%d" % ( self.ifaceIp4, self.opsPort ) )
self.port_range = ( self.configFile.get( 'port_range_start', 5000 ), self.configFile.get( 'port_range_end', 6000 ) )
self.ports_available.update( xrange( self.port_range[ 0 ], self.port_range[ 1 ] + 1 ) )
self.peer_keepalive_seconds = self.configFile.get( 'peer_keepalive_seconds', 60 )
self.instance_keepalive_seconds = self.configFile.get( 'instance_keepalive_seconds', 600 )
self.directory_sync_seconds = self.configFile.get( 'directory_sync_seconds', 60 )
self.tombstone_culling_seconds = self.configFile.get( 'tombstone_culling_seconds', 3600 )
self.instance_strategy = self.configFile.get( 'instance_strategy', 'random' )
# Bootstrap the seeds
for s in self.seedNodes:
self._connectToNode( s )
# Start services
self._log( "Starting services" )
gevent.spawn_later( random.randint( 0, 3 ), self._svc_directory_requests )
gevent.spawn_later( random.randint( 0, 3 ), self._svc_instance_keepalive )
gevent.spawn_later( random.randint( 0, 3 ), self._svc_host_keepalive )
gevent.spawn_later( random.randint( 0, 3 ), self._svc_directory_sync )
gevent.spawn_later( random.randint( 0, 3 ), self._svc_cullTombstones )
gevent.spawn_later( random.randint( 0, 3 ), self._svc_applyTombstones )
gevent.spawn_later( random.randint( 0, 3 ), self._svc_cleanupCats )
for _ in range( 20 ):
gevent.spawn( self._svc_receiveOpsTasks )
gevent.spawn( self._svc_pushDirChanges )
# Start the instances
for n in range( self.nProcesses ):
self._startInstance( isIsolated = False )
# Wait to be signaled to exit
self._log( "Up and running" )
timeToStopEvent.wait()
# Any teardown required
for proc in self.processes:
self._sendQuitToInstance( proc )
self._log( "Exiting." )
def __init__(self, configFile, instanceId, logging_level, logging_dest,
interface):
# Setting the signal handler to trigger the stop event which
# is interpreted by each actor implementation
global timeToStopEvent
gevent.signal(signal.SIGQUIT, _stopAllActors)
gevent.signal(signal.SIGINT, _stopAllActors)
gevent.signal(signal.SIGTERM, _stopAllActors)
self.instanceId = instanceId
self._log_level = logging_level
self._log_dest = logging_dest
self._initLogging(logging_level, logging_dest)
self.log("Initializing")
self.stopEvent = timeToStopEvent
self.isOpen = True
self.actors = {}
self.py_beach_dir = None
self.configFilePath = configFile
self.configFile = None
self.interface = interface
self.ifaceIp4 = _getIpv4ForIface(self.interface)
with open(self.configFilePath, 'r') as f:
self.configFile = yaml.load(f)
self.py_beach_dir = os.path.dirname(os.path.abspath(__file__))
os.chdir(os.path.dirname(os.path.abspath(self.configFilePath)))
self.private_key = self.configFile.get('private_key', None)
if self.private_key is not None:
with open(self.private_key, 'r') as f:
key_path = self.private_key
self.private_key = f.read()
self.log("Using shared key: %s" % key_path)
self.codeDirectory = self.configFile.get('code_directory', './')
if '://' not in self.codeDirectory:
self.codeDirectory = os.path.abspath(self.codeDirectory)
Actor._code_directory_root = self.codeDirectory
self.opsSocket = _ZMREP('ipc:///tmp/py_beach_instance_%s' % instanceId,
isBind=True)
#self.log( "Listening for ops on %s" % ( 'ipc:///tmp/py_beach_instance_%s' % instanceId, ) )
self.hostOpsPort = self.configFile.get('ops_port', 4999)
self.hostOpsSocket = _ZMREP('tcp://%s:%d' %
(self.ifaceIp4, self.hostOpsPort),
isBind=False,
private_key=self.private_key)
ActorHandle._setHostDirInfo(
self.configFile.get('directory_port',
'ipc:///tmp/py_beach_directory_port'),
self.private_key)
ActorHandleGroup._setHostDirInfo(
'tcp://%s:%d' % (self.ifaceIp4, self.hostOpsPort),
self.private_key)
for _ in range(20):
gevent.spawn(self.svc_receiveTasks)
gevent.spawn(self.svc_monitorActors)
gevent.spawn(self.svc_reportUsage)
#self.log( "Now open to actors" )
timeToStopEvent.wait()
self.log("Exiting, stopping all actors.")
for actor in self.actors.values():
actor.stop()
gevent.joinall(self.actors.values())
self.log("All Actors exiting, exiting.")
def __init__( self, configFile, iface = None ):
# Setting the signal handler to trigger the stop event
global timeToStopEvent
gevent.signal( signal.SIGQUIT, _stop )
gevent.signal( signal.SIGINT, _stop )
self._logger = None
self._initLogging()
self.stopEvent = timeToStopEvent
self.py_beach_dir = None
self.configFilePath = os.path.abspath( configFile )
self.configFile = None
self.directory = {}
self.tombstones = {}
self.actorInfo = {}
self.ports_available = Set()
self.nProcesses = 0
self.processes = []
self.initialProcesses = False
self.seedNodes = []
self.directoryPort = None
self.opsPort = 0
self.opsSocket = None
self.port_range = ( 0, 0 )
self.interface = None
self.ifaceIp4 = None
self.nodes = {}
self.peer_keepalive_seconds = 0
self.instance_keepalive_seconds = 0
self.tombstone_culling_seconds = 0
self.isActorChanged = gevent.event.Event()
self.isInstanceChanged = gevent.event.Event()
# Load default configs
with open( self.configFilePath, 'r' ) as f:
self.configFile = yaml.load( f )
self.py_beach_dir = os.path.dirname( os.path.abspath( __file__ ) )
os.chdir( os.path.dirname( os.path.abspath( self.configFilePath ) ) )
self.nProcesses = self.configFile.get( 'n_processes', 0 )
if self.nProcesses == 0:
self.nProcesses = multiprocessing.cpu_count()
self._log( "Using %d instances per node" % self.nProcesses )
if iface is not None:
self.interface = iface
else:
self.interface = self.configFile.get( 'interface', 'eth0' )
self.ifaceIp4 = _getIpv4ForIface( self.interface )
self.seedNodes = self.configFile.get( 'seed_nodes', [] )
if 0 == len( self.seedNodes ):
self.seedNodes.append( self.ifaceIp4 )
for s in self.seedNodes:
self._log( "Using seed node: %s" % s )
self.directoryPort = _ZMREP( self.configFile.get( 'directory_port',
'ipc:///tmp/py_beach_directory_port' ),
isBind = True )
self.opsPort = self.configFile.get( 'ops_port', 4999 )
self.opsSocket = _ZMREP( 'tcp://%s:%d' % ( self.ifaceIp4, self.opsPort ), isBind = True )
self._log( "Listening for ops on %s:%d" % ( self.ifaceIp4, self.opsPort ) )
self.port_range = ( self.configFile.get( 'port_range_start', 5000 ), self.configFile.get( 'port_range_end', 6000 ) )
self.ports_available.update( xrange( self.port_range[ 0 ], self.port_range[ 1 ] + 1 ) )
self.peer_keepalive_seconds = self.configFile.get( 'peer_keepalive_seconds', 60 )
self.instance_keepalive_seconds = self.configFile.get( 'instance_keepalive_seconds', 60 )
self.directory_sync_seconds = self.configFile.get( 'directory_sync_seconds', 60 )
self.tombstone_culling_seconds = self.configFile.get( 'tombstone_culling_seconds', 3600 )
self.instance_strategy = self.configFile.get( 'instance_strategy', 'random' )
# Bootstrap the seeds
for s in self.seedNodes:
self._connectToNode( s )
# Start services
self._log( "Starting services" )
gevent.spawn( self._svc_directory_requests )
gevent.spawn( self._svc_instance_keepalive )
gevent.spawn( self._svc_host_keepalive )
gevent.spawn( self._svc_directory_sync )
gevent.spawn( self._svc_cullTombstones )
gevent.spawn( self._svc_receiveOpsTasks )
gevent.spawn( self._svc_pushDirChanges )
# Start the instances
for n in range( self.nProcesses ):
self._startInstance( isIsolated = False )
# Wait to be signaled to exit
self._log( "Up and running" )
timeToStopEvent.wait()
# Any teardown required
for proc in self.processes:
self._sendQuitToInstance( proc )
self._log( "Exiting." )
def __init__(self,
host,
realm,
ip,
port,
uid,
log_level,
log_dest,
beach_config_path,
parameters={},
resources={},
ident=None,
trusted=[],
n_concurrent=1,
private_key=None):
gevent.Greenlet.__init__(self)
self.name = uid
self._log_level = log_level
self._log_dest = log_dest
self._beach_config_path = beach_config_path
self._initLogging(log_level, log_dest)
self.stopEvent = gevent.event.Event()
self._realm = realm
self._ip = ip
self._port = port
self._host = host
self._parameters = parameters
self._resources = resources
self._ident = ident
self._trusted = trusted
self._n_initial_concurrent = n_concurrent
self._n_concurrent = 0
self._private_key = private_key
self._exception = None
self._n_free_handlers = 0
self._is_initted = False
self._qps = 0.0
self._q_counter = 0
self._last_qps_count = 0
self._q_total_time = 0.0
self._q_avg = 0.0
# We have some special magic parameters
self._trace_enabled = self._parameters.get('beach_trace_enabled',
False)
# We keep track of all the handlers for the user per message request type
self._handlers = {}
# All user generated threads
self._threads = gevent.pool.Group()
# This socket receives all taskings for the actor and dispatch
# the messages as requested by user
self._opsSocket = _ZMREP('tcp://%s:%d' % (self._ip, self._port),
isBind=True,
private_key=self._private_key)
self._vHandles = []
self.schedule(10, self._generateQpsCount)
def __init__( self,
host,
realm,
ip,
port,
uid,
log_level,
log_dest,
beach_config_path,
parameters = {},
resources = {},
ident = None,
trusted = [],
n_concurrent = 1,
private_key = None ):
gevent.Greenlet.__init__( self )
self.name = uid
self._log_level = log_level
self._log_dest = log_dest
self._beach_config_path = beach_config_path
self._initLogging( log_level, log_dest )
self.stopEvent = gevent.event.Event()
self._realm = realm
self._ip = ip
self._port = port
self._host = host
self._parameters = parameters
self._resources = resources
self._ident = ident
self._trusted = trusted
self._n_initial_concurrent = n_concurrent
self._n_concurrent = 0
self._private_key = private_key
self._exception = None
self._n_free_handlers = 0
self._is_initted = False
self._qps = 0.0
self._q_counter = 0
self._last_qps_count = 0
self._q_total_time = 0.0
self._q_avg = 0.0
# We have some special magic parameters
self._trace_enabled = self._parameters.get( 'beach_trace_enabled', False )
# We keep track of all the handlers for the user per message request type
self._handlers = {}
# All user generated threads
self._threads = gevent.pool.Group()
# This socket receives all taskings for the actor and dispatch
# the messages as requested by user
self._opsSocket = _ZMREP( 'tcp://%s:%d' % ( self._ip, self._port ),
isBind = True,
private_key = self._private_key )
self._vHandles = []
self.schedule( 10, self._generateQpsCount )
def __init__( self, configFile, instanceId, logging_level, logging_dest, interface ):
# Setting the signal handler to trigger the stop event which
# is interpreted by each actor implementation
global timeToStopEvent
gevent.signal( signal.SIGQUIT, _stopAllActors )
gevent.signal( signal.SIGINT, _stopAllActors )
self.instanceId = instanceId
self._log_level = logging_level
self._log_dest = logging_dest
self._initLogging( logging_level, logging_dest )
self.log( "Initializing" )
self.stopEvent = timeToStopEvent
self.actors = {}
self.py_beach_dir = None
self.configFilePath = configFile
self.configFile = None
self.interface = interface
self.ifaceIp4 = _getIpv4ForIface( self.interface )
with open( self.configFilePath, 'r' ) as f:
self.configFile = yaml.load( f )
self.py_beach_dir = os.path.dirname( os.path.abspath( __file__ ) )
os.chdir( os.path.dirname( os.path.abspath( self.configFilePath ) ) )
self.private_key = self.configFile.get( 'private_key', None )
if self.private_key is not None:
with open( self.private_key, 'r' ) as f:
key_path = self.private_key
self.private_key = f.read()
self.log( "Using shared key: %s" % key_path )
self.codeDirectory = self.configFile.get( 'code_directory', './' )
if '://' not in self.codeDirectory:
self.codeDirectory = os.path.abspath( self.codeDirectory )
Actor._code_directory_root = self.codeDirectory
self.opsSocket = _ZMREP( 'ipc:///tmp/py_beach_instance_%s' % instanceId,
isBind = True )
self.log( "Listening for ops on %s" % ( 'ipc:///tmp/py_beach_instance_%s' % instanceId, ) )
self.hostOpsPort = self.configFile.get( 'ops_port', 4999 )
self.hostOpsSocket = _ZMREP( 'tcp://%s:%d' % ( self.ifaceIp4, self.hostOpsPort ),
isBind = False,
private_key = self.private_key )
ActorHandle._setHostDirInfo( self.configFile.get( 'directory_port',
'ipc:///tmp/py_beach_directory_port' ),
self.private_key )
ActorHandleGroup._setHostDirInfo( 'tcp://%s:%d' % ( self.ifaceIp4, self.hostOpsPort ),
self.private_key )
for _ in range( 20 ):
gevent.spawn( self.svc_receiveTasks )
gevent.spawn( self.svc_monitorActors )
self.log( "Now open to actors" )
timeToStopEvent.wait()
self.log( "Exiting, stopping all actors." )
for actor in self.actors.values():
actor.stop()
gevent.joinall( self.actors.values() )
self.log( "All Actors exiting, exiting." )
