src/tashi/accounting/accounting.py

#!/usr/bin/python

# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
# 
#   http://www.apache.org/licenses/LICENSE-2.0
# 
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.    

import sys
import signal
import logging.config

from tashi.rpycservices import rpycservices
from rpyc.utils.server import ThreadedServer
#from rpyc.utils.authenticators import TlsliteVdbAuthenticator

#from tashi.rpycservices.rpyctypes import *
from tashi.util import getConfig, createClient, instantiateImplementation, boolean, debugConsole, signalHandler
import tashi

class Accounting(object):
	def __init__(self, config, cmclient):
		self.config = config
		self.cm = cmclient
		self.hooks = []
		self.log = logging.getLogger(__file__)

		items = self.config.items("Accounting")
		items.sort()
		for item in items:
			(name, value) = item
			name = name.lower()
			if (name.startswith("hook")):
				try:
					self.hooks.append(instantiateImplementation(value, config, cmclient, False))
				except:
					self.log.exception("Failed to load hook %s" % (value))
					
	def initAccountingServer(self):
		service = instantiateImplementation(self.config.get("Accounting", "service"), self.config)

		#if boolean(self.config.get("Security", "authAndEncrypt")):
		if False:
			pass
		else:
			t = ThreadedServer(service=rpycservices.ManagerService, hostname='0.0.0.0', port=int(self.config.get('AccountingService', 'port')), auto_register=False)

		t.logger.setLevel(logging.ERROR)
		t.service.service = service
		t.service._type = 'AccountingService'

		debugConsole(globals())

		try:
			t.start()
		except KeyboardInterrupt:
			self.handleSIGTERM(signal.SIGTERM, None)

	@signalHandler(signal.SIGTERM)
	def handleSIGTERM(self, signalNumber, stackFrame):
		self.log.info('Exiting cluster manager after receiving a SIGINT signal')
		sys.exit(0)

def main():
	(config, configFiles) = getConfig(["Accounting"])
	publisher = instantiateImplementation(config.get("Accounting", "publisher"), config)
	tashi.publisher = publisher
	cmclient = createClient(config)
	logging.config.fileConfig(configFiles)
	accounting = Accounting(config, cmclient)

	accounting.initAccountingServer()

if __name__ == "__main__":
	main()