def onfailure(futurekey):
futureobj = GetFutureAndCheckReady(futurekey)
lrun = lrunKey.get()
if lrun and lrun.status in [
_TEST_RUN_STATUS_NOT_READY, _TEST_RUN_STATUS_READY
]:
lrun.status = _TEST_RUN_STATUS_FAIL
lrun.progress = 100
try:
futureobj.get_result()
except Exception, ex:
lrun.final_message = str(ex)
lrun.final_runtime_usec = calc_final_runtime_usec(lrun.started)
lrun.put()
def OnPageMapSuccess(pagemapfuturekey):
pagemapfuture = GetFutureAndCheckReady(pagemapfuturekey)
placeholderfuture = GetFutureAndCheckReady(placeholderfuturekey)
future = GetFutureAndCheckReady(futurekey)
result = pagemapfuture.get_result()
placeholderfuture.set_success(result)
future.set_success(result)
def onsuccess(futurekey):
futureobj = GetFutureAndCheckReady(futurekey)
lrun = lrunKey.get()
if lrun and lrun.status in [
_TEST_RUN_STATUS_NOT_READY, _TEST_RUN_STATUS_READY
]:
lrun.status = _TEST_RUN_STATUS_PASS
lrun.progress = 100
lresult = futureobj.get_result()
if lresult:
lrun.final_message = str(lresult)
lrun.final_runtime_usec = calc_final_runtime_usec(lrun.started)
lrun.put()
def OnCountSuccess(countfuturekey):
countfuture = GetFutureAndCheckReady(countfuturekey)
futureobj = GetFutureAndCheckReady(futurekey)
count = countfuture.get_result()
placeholderfuture = placeholderfuturekey.get()
if placeholderfuture:
placeholderfuture.set_weight(count * 2)
futureobj.set_weight(count * 2)
futurendbshardedpagemap(pagemapf,
ndbquery,
pagesize,
onsuccessf=OnPageMapSuccess,
weight=count,
parentkey=placeholderfuturekey,
**taskkwargs)
# now that the second pass is actually constructed and running, we can let the placeholder accept a result.
placeholderfuture.set_readyforesult()