def cluster_status(self):
"""
Returns a ThriftClusterStatus
"""
cs = self.client.getClusterStatus(self.request_context)
fixup_enums(cs, {"state": JobTrackerState})
return cs
def cluster_status(self):
"""
Returns a ThriftClusterStatus
"""
cs = self.client.getClusterStatus(self.thread_local.request_context)
fixup_enums(cs, {"state": JobTrackerState})
return cs
def test_fixup_enums(self):
enum = TestEnum()
struct1 = TestStruct()
self.assertTrue(hasattr(enum, "_VALUES_TO_NAMES"))
struct1.myenum = 0
thrift_util.fixup_enums(struct1, {"myenum": TestEnum})
self.assertTrue(hasattr(struct1, "myenumAsString"))
self.assertEquals(struct1.myenumAsString, 'ENUM_ONE')
def _fixup_tasktracker(self, tracker):
"""
Adds string representations of enums to a ThirftTaskTrackerStatus
"""
if not tracker.taskReports:
return tracker
for tts in tracker.taskReports:
fixup_enums(tts.taskID.taskID, {"taskType": ThriftTaskType})
self._fixup_taskstatus(tts)
def _fixup_taskstatus(self, taskstatus):
"""
Adds string representations of enums to a ThriftTaskInProgress
"""
fixup_enums(taskstatus, {
"state": ThriftTaskState,
"phase": ThriftTaskPhase
})
fixup_enums(taskstatus.taskID.taskID, {"taskType": ThriftTaskType})
def _fixup_tasktracker(self, tracker):
"""
Adds string representations of enums to a ThirftTaskTrackerStatus
"""
if not tracker.taskReports:
return tracker
for tts in tracker.taskReports:
fixup_enums(tts.taskID.taskID, {"taskType": ThriftTaskType})
self._fixup_taskstatus(tts)
def _fixup_job(self, job):
"""
Run through all the enums in a job and add their string variant
"""
fixup_enums(job.status, {"runState": ThriftJobState, "priority": ThriftJobPriority})
fixup_enums(job, {"priority": ThriftJobPriority})
if job.tasks is not None:
for tip in job.tasks.tasks:
fixup_enums(tip.taskID, {"taskType": ThriftTaskType})
self._fixup_task_state(tip)
for taskstatus in tip.taskStatuses.values():
self._fixup_taskstatus(taskstatus)
def _fixup_job(self, job):
"""
Run through all the enums in a job and add their string variant
"""
fixup_enums(job.status, {
"runState": ThriftJobState,
"priority": ThriftJobPriority
})
fixup_enums(job, {"priority": ThriftJobPriority})
if job.tasks is not None:
for tip in job.tasks.tasks:
fixup_enums(tip.taskID, {"taskType": ThriftTaskType})
self._fixup_task_state(tip)
for taskstatus in tip.taskStatuses.values():
self._fixup_taskstatus(taskstatus)
def _fixup_taskstatus(self, taskstatus):
"""
Adds string representations of enums to a ThriftTaskInProgress
"""
fixup_enums(taskstatus, {"state": ThriftTaskState, "phase": ThriftTaskPhase})
fixup_enums(taskstatus.taskID.taskID, {"taskType": ThriftTaskType})
def _fixup_task_in_progress(self, tip):
fixup_enums(tip.taskID, {"taskType": ThriftTaskType})
self._fixup_task_state(tip)
for taskstatus in tip.taskStatuses.values():
self._fixup_taskstatus(taskstatus)
def _fixup_task_in_progress(self, tip):
fixup_enums(tip.taskID, {"taskType": ThriftTaskType})
self._fixup_task_state(tip)
for taskstatus in tip.taskStatuses.values():
self._fixup_taskstatus(taskstatus)