libra/bin/client.py
David Shrewsbury ca5445b3c5 Remove CREATE action from worker msg protocol.
Fixes bug #1078774

CREATE is now replaced by UPDATE, which did the same thing anyway.

Change-Id: Ic446a349e053c7787209bdc227bd93b02d67dd12
2012-11-14 12:36:20 -05:00

73 lines
2.0 KiB
Python
Executable File

#!/usr/bin/env python
import json
import socket
from gearman import GearmanClient, DataEncoder
class JSONDataEncoder(DataEncoder):
@classmethod
def encode(cls, encodable_object):
s = json.dumps(encodable_object)
print("Encoding JSON object to string: %s" % s)
return s
@classmethod
def decode(cls, decodable_string):
s = json.loads(decodable_string)
print("Decoding string (%s) to JSON object" % s)
return s
class JSONGearmanClient(GearmanClient):
data_encoder = JSONDataEncoder
def check_request_status(job_request):
if job_request.complete:
print "Job %s finished! Result: %s -\n%s" % (job_request.job.unique,
job_request.state,
json.dumps(
job_request.result,
indent=2
))
elif job_request.timed_out:
print "Job %s timed out!" % job_request.unique
elif job_request.state == JOB_UNKNOWN:
print "Job %s connection failed!" % job_request.unique
def main():
hostname = socket.gethostname()
task = "lbaas-1.0-%s" % hostname
client = JSONGearmanClient(['localhost:4730'])
data = """
{
"hpcs_action": "update",
"loadbalancers": [
{
"name": "a-new-loadbalancer",
"protocol": "http",
"nodes": [
{
"address": "10.1.1.1",
"port": "80"
},
{
"address": "10.1.1.2",
"port": "81"
}
]
}
]
}
"""
# Worker class expects the data as a JSON object, not string
json_data = json.loads(data)
request = client.submit_job(task, json_data)
check_request_status(request)
if __name__ == "__main__":
main()