|
||||
0001 # 0002 # Licensed to the Apache Software Foundation (ASF) under one or more 0003 # contributor license agreements. See the NOTICE file distributed with 0004 # this work for additional information regarding copyright ownership. 0005 # The ASF licenses this file to You under the Apache License, Version 2.0 0006 # (the "License"); you may not use this file except in compliance with 0007 # the License. You may obtain a copy of the License at 0008 # 0009 # http://www.apache.org/licenses/LICENSE-2.0 0010 # 0011 # Unless required by applicable law or agreed to in writing, software 0012 # distributed under the License is distributed on an "AS IS" BASIS, 0013 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 0014 # See the License for the specific language governing permissions and 0015 # limitations under the License. 0016 # 0017 0018 import os 0019 import imp 0020 0021 0022 # This is a hack to always refer the main code rather than built zip. 0023 main_code_dir = os.path.dirname(os.path.dirname(os.path.realpath(__file__))) 0024 daemon = imp.load_source("daemon", "%s/pyspark/daemon.py" % main_code_dir) 0025 0026 if "COVERAGE_PROCESS_START" in os.environ: 0027 worker = imp.load_source("worker", "%s/pyspark/worker.py" % main_code_dir) 0028 0029 def _cov_wrapped(*args, **kwargs): 0030 import coverage 0031 cov = coverage.coverage( 0032 config_file=os.environ["COVERAGE_PROCESS_START"]) 0033 cov.start() 0034 try: 0035 worker.main(*args, **kwargs) 0036 finally: 0037 cov.stop() 0038 cov.save() 0039 daemon.worker_main = _cov_wrapped 0040 else: 0041 raise RuntimeError("COVERAGE_PROCESS_START environment variable is not set, exiting.") 0042 0043 0044 if __name__ == '__main__': 0045 daemon.manager()
[ Source navigation ] | [ Diff markup ] | [ Identifier search ] | [ general search ] |
This page was automatically generated by the 2.1.0 LXR engine. The LXR team |