Beispiel #1
0
    def Load(self, conf):
        """
        load config and init module
        """
        self.config = yaml.load(open(conf, 'r').read(), Loader=yaml.FullLoader)

        # overwrite config from env
        if 'OPENEDGE_SERVICE_INSTANCE_NAME' in os.environ:
            self.config['name'] = os.environ['OPENEDGE_SERVICE_INSTANCE_NAME']
        elif 'OPENEDGE_SERVICE_NAME' in os.environ:  # deprecated
            self.config['name'] = os.environ['OPENEDGE_SERVICE_NAME']

        if 'OPENEDGE_SERVICE_INSTANCE_ADDRESS' in os.environ:
            if 'server' not in self.config:
                self.config['server'] = {}
            self.config['server']['address'] = os.environ['OPENEDGE_SERVICE_INSTANCE_ADDRESS']
        elif 'OPENEDGE_SERVICE_ADDRESS' in os.environ:  # deprecated
            if 'server' not in self.config:
                self.config['server'] = {}
            self.config['server']['address'] = os.environ['OPENEDGE_SERVICE_ADDRESS']

        if 'name' not in self.config:
            raise Exception('config invalid, missing name')
        if 'server' not in self.config:
            raise Exception('config invalid, missing server')
        if 'address' not in self.config['server']:
            raise Exception('config invalid, missing server address')
        if 'functions' not in self.config:
            raise Exception('config invalid, missing functions')

        self.log = get_logger(self.config)
        self.functions = get_functions(self.config['functions'])
        self.server = get_grpc_server(self.config['server'])
        function_pb2_grpc.add_FunctionServicer_to_server(self, self.server)
Beispiel #2
0
    def Load(self):
        """
        load config and init module
        """
        if 'SERVICE_NAME' in os.environ:
            self.name = os.environ['SERVICE_NAME']

        if 'SERVICE_CONF' in os.environ:
            self.conf_path = os.environ['SERVICE_CONF']

        if 'SERVICE_CODE' in os.environ:
            self.code_path = os.environ['SERVICE_CODE']

        if 'SERVICE_ADDRESS' in os.environ:
            self.server_address = os.environ['SERVICE_ADDRESS']

        self.config = {'server': {'address': self.server_address}}

        if os.path.exists(self.conf_path):
            self.config = yaml.load(open(self.config, 'r').read(),
                                    Loader=yaml.FullLoader)

        self.log = get_logger(self)
        self.functions = get_functions(self.code_path)
        self.server = get_grpc_server(self.config['server'])
        function_pb2_grpc.add_FunctionServicer_to_server(self, self.server)