Пример #1
0
def job_url_update(_job_key=""):
    func_name = sys._getframe().f_code.co_name
    file_name = __name__.upper()
    helpers.job_key_check(_job_key, func_name, file_name)
    file_dir_name = os.path.dirname(os.path.abspath(__file__))
    job_status_interface = weibox_macros.CC_JOB_UPDATE_INTERFACE
    job_space = helpers.job_space(_job_key)
    job_trace_log_file = job_space + "/weiflow-from-weiclient.log"
    job_status = ""
    try:
        # job submit failed
        if (not os.path.isdir(job_space)):
            job_status_update.job_status_update(_job_key, "FAILED")
        # job run failed
        elif (not os.path.isfile(job_trace_log_file)):
            job_status_update.job_status_update(_job_key, "LOST")
        # job run successed.
        else:
            yarn_app_id = read_yarn_app_id(_job_key)
            # job not fount applicationId : submitted just now or has no applicationId
            if ("NULL_YARN_APPLICATION_ID" == yarn_app_id):
                job_status_update.job_status_update(_job_key, "WAITING")
            else:
                log_url = weibox_macros.CC_JOB_LOG_URL_INTERFACE % _job_key
                job_url = yarn_operation.job_trace_url(yarn_app_id)
                # job_status = json.loads(com.weibo.weibox.yarn.tools.yarn_operation.yarn_job_query(yarn_app_id)).get("app").get("state")
                job_status = yarn_operation.yarn_job_status(yarn_app_id)
                job_status_update.job_update(_job_key=_job_key,
                                             _status=job_status,
                                             _job_url=job_url,
                                             _log_url=log_url)
    except Exception, e:
        logger.error(func_name + " : " + e.message)
Пример #2
0
def job_status(_job_key=""):
    # checke the job_key
    func_name = sys._getframe().f_code.co_name
    file_name = __name__.upper()
    helpers.job_key_check(_job_key, func_name, file_name)
    job_status_update.job_status_update(_job_key=_job_key, _status="acceptted")
    status_cmd = assembleCmd(_job_key)
    print("arena status cmd: %s " % status_cmd)
    '''
Пример #3
0
def job_query(_job_key = ""):
    func_name = sys._getframe().f_code.co_name
    file_name = __name__.upper()
    helpers.job_key_check(_job_key,func_name,file_name)
    try:
        response_info = helpers.get(utils.weibox_macros.CC_JOB_QUERY_INTERFACE % _job_key)
        response_obj = json.loads(response_info)
    except Exception,e:
        logger.error(e.message + "at %s." % file_name)
Пример #4
0
def job_update(_job_key = "", _status = "", _job_url = "", _log_url = ""):
    func_name = sys._getframe().f_code.co_name
    file_name = __name__.upper()
    helpers.job_key_check(_job_key,func_name,file_name)
    job_status_interface = utils.weibox_macros.CC_JOB_UPDATE_INTERFACE
    job_update_url = job_status_interface + "&jobUrl=%s&logUrl=%s&jobStatus=%s" % (_job_url,_log_url,_status)
    job_update_url = job_status_interface
    param_update = {"jobKey" : _job_key, "jobUrl" : _job_url, "logUrl" : _log_url, "jobStatus" : _status}
    return helpers.post(job_update_url, param_update, "basic")
Пример #5
0
def job_stdout_update(_job_key=""):
    func_name = sys._getframe().f_code.co_name
    file_name = __name__.upper()
    helpers.job_key_check(_job_key, func_name, file_name)
    file_dir_name = os.path.dirname(os.path.abspath(__file__))
    job_stdout_interface = utils.weibox_macros.CC_JOB_STDOUT_INTERFACE
    job_space = helpers.job_space(_job_key)
    job_log_file = job_space + "/weiflow-from-weiclient.log"
    param_value = {"jobKey": _job_key, "file": job_log_file}
    return helpers.post(job_stdout_interface, param_value, "stdout")
Пример #6
0
def read_yarn_app_id(_job_key = ""):
    func_name = sys._getframe().f_code.co_name
    file_name = __name__.upper()
    helpers.job_key_check(_job_key,func_name,file_name)
    file_dir_name = os.path.dirname(os.path.abspath(__file__))
    job_status_interface = weibox_macros.CC_JOB_UPDATE_INTERFACE
    job_space = helpers.job_space(_job_key)
    job_trace_log_file = job_space + weibox_macros.WEIBOX_LOG
    logger.info("JOB_TRACE_LOG_FILE : %s." % job_trace_log_file)
    try:
        job_log_content = open(job_trace_log_file).read()
        yarn_app_id = resolve_yarn_app_id([job_log_content])
    except Exception,e:
        logger.error(func_name + " : " + e.message)
Пример #7
0
def job_prepare(_job_key=""):
    # checke the job_key
    func_name = sys._getframe().f_code.co_name
    file_name = __name__.upper()
    helpers.job_key_check(_job_key, func_name, file_name)
    # update the status to cc show weibox received the job.
    job_status_update.job_status_update(_job_key=_job_key,
                                        _status="submitting")
    weibox_operation.job_seperate(_job_key)
    _key = "result.json"
    rst_json = get_args.get_args(_job_key, _key)

    # job prepared. ready
    job_status_update.job_status_update(_job_key=_job_key, _status="ready")
    return rst_json
Пример #8
0
def job_seperate(_job_key=""):
    func_name = sys._getframe().f_code.co_name
    file_name = __name__.upper()
    helpers.job_key_check(_job_key, func_name, file_name)
    #response_obj = job_status_update.job_query(_job_key)
    #code_path = helpers.get_value("result.codePath", response_obj)
    job_space = helpers.job_space(_job_key)
    if (os.path.isdir(job_space)):
        import shutil
        shutil.rmtree(job_space)
    mkdir_response_code = os.makedirs(job_space)
    print("mkdir code:%s" % mkdir_response_code)
    #rsync_response_code = 0
    #if("null" != code_path):
    #rsync_response_code = helpers.sys_run("rsync -zr --progress 10.77.29.68::backup/weiflow/%s/* %s/" % (code_path,job_space))
    return mkdir_response_code  # or rsync_response_code
Пример #9
0
def job_prepare(_job_key=""):
    # checke the job_key
    func_name = sys._getframe().f_code.co_name
    file_name = __name__.upper()
    helpers.job_key_check(_job_key, func_name, file_name)
    # update the status to cc show weibox received the job.
    job_status_update.job_status_update(_job_key=_job_key,
                                        _status="submitting")

    response_obj = job_status_update.job_query(_job_key)
    target_user_cc = weibox_macros.submit_account(
        helpers.get_value("result.submitAccount", response_obj))

    # run_command = helpers.get_value("result.runCommand", response_obj)
    _key = "result.json"
    rst_json = get_args.get_args(_job_key, _key)
    specified_submit_type = rst_json.get("job_common_submit_type")

    # job prepared. ready
    job_status_update.job_status_update(_job_key=_job_key, _status="ready")
    return rst_json
Пример #10
0
def job_submit(_job_key=""):
    # checke the job_key
    func_name = sys._getframe().f_code.co_name
    file_name = __name__.upper()
    helpers.job_key_check(_job_key, func_name, file_name)
    rst_json = job_prepare(_job_key)
    job_status_update.job_status_update(_job_key=_job_key, _status="acceptted")
    submit_cmd = assemble_cmd(rst_json, _job_key)
    print("arena submit cmd: %s " % submit_cmd)

    try:
        #sys_run_code = helpers.sys_run(submit_cmd)
        status, output = commands.getstatusoutput(submit_cmd)
        print("status code :%s" % status)
        print("output :%s" % output)
        if (0 == status):
            log_url = get_log_url(rst_json)
            print("log url:%s" % log_url)
            job_url = get_job_url(log_url)
            print("job url:%s" % job_url)
            job_log_url = job_url + "\n" + log_url
            #job_status_update.job_update(_job_key, "acceptted", log_url, "")
            job_status_update.job_update(_job_key, "acceptted", job_log_url,
                                         "")
            #job_space = helpers.job_space(_job_key)
            #log_cmd = "arena logs -f " + rst_json.get("job_name") + " >> " + job_space + "/weiflow-from-weiclient.log"
            #print("logs cmd:%s" % log_cmd)
            #status, output = commands.getstatusoutput(log_cmd)
            #job_url_update.job_url_update(_job_key)
            #job_status_update.job_status_update(_job_key = _job_key, _status = "finished")
        else:
            job_status_update.job_status_update(_job_key=_job_key,
                                                _status="failed")
            print(output)
    except Exception, e:
        logger.warn(e)
        job_status_update.job_status_update(_job_key=_job_key,
                                            _status="failed")
Пример #11
0
def job_log_and_update(_job_key=""):
    func_name = sys._getframe().f_code.co_name
    file_name = __name__.upper()
    helpers.job_key_check(_job_key, func_name, file_name)
    job_url_update(_job_key)
    job_stdout_update(_job_key)