def updateESDocStatus(url_update,
url_search,
uuid,
status,
info='',
reason=''):
for i in range(RETRY_COUNT):
updateDocStatus(url_update, uuid, status, info, reason)
#leave this for test
openRequest = retrieveRequestById(url_search, uuid)
if openRequest != None:
new_status = openRequest['status']
if new_status == status:
return True
time.sleep(1)
logger.error("ElasticSearch failed " + uuid)
continue
#one more last try
for i in range(RETRY_COUNT):
updateDocStatus(url_update, uuid, status)
openRequest = retrieveRequestById(url_search, uuid)
if openRequest != None:
new_status = openRequest['status']
if new_status == status:
return True
time.sleep(1)
logger.error("ElasticSearch failed " + uuid)
continue
return False
def reserveJob(url_update, url_search, uuid, status):
for i in range(RETRY_COUNT):
if status == REQUEST_STATE.INITIAL.value:
updatedStatus = REQUEST_STATE.PREPROCESS_INPROGRESS.value
updateDocStatus(url_update, uuid, updatedStatus)
elif status == REQUEST_STATE.PREPROCESS_COMPLETED.value:
updatedStatus = REQUEST_STATE.PREPROCESS_INPROGRESS.value
updateDocStatus(url_update, uuid, updatedStatus)
elif status == REQUEST_STATE.PREPROCESS_INPROGRESS.value:
updatedStatus = REQUEST_STATE.PREPROCESS_INPROGRESS.value
updateDocStatus(url_update, uuid, updatedStatus)
else:
#leave this for test
updatedStatus = status
updateDocStatus(url_update, uuid, updatedStatus)
openRequest = retrieveRequestById(url_search, uuid)
if openRequest == None:
logger.error("failed to find uuid " + uuid)
continue
new_status = openRequest['status']
if new_status == updatedStatus:
return updatedStatus
else:
if i == 2:
logger.error("failed to update uuid " + uuid + " from " +
status + " to " + updatedStatus)
continue
return status
def postprocess(url_update, uuid):
return updateDocStatus(url_update, uuid, REQUEST_STATE.POSTPROCESS.value)
def preprocess(url_update, uuid):
return updateDocStatus(url_update, uuid, REQUEST_STATE.PREPROCESS.value)