class McdClient(): def __init__(self, host='127.0.0.1', port=11211): self.conn = Connection(host, port) self.conn.connect() def stats(self, type = ''): op = Stats(type) self.conn.queue_operation(op) return op def set(self, key, value, vbucket, flags, exp): op = Set(key, value, vbucket, flags, exp) self.conn.queue_operation(op) return op def delete(self, key, vbucket): op = Delete(key, vbucket) self.conn.queue_operation(op) return op def flush(self): op = Flush() self.conn.queue_operation(op) return op def shutdown(self): self.conn.close()
class UprClient(): def __init__(self, host='127.0.0.1', port=11211): self.conn = Connection(host, port) self.conn.connect() def set_proxy(self, client): self.conn.proxy = client.conn.socket def open_consumer(self, name): op = OpenConnection(FLAG_OPEN_CONSUMER, name) self.conn.queue_operation(op) return op def open_producer(self, name): op = OpenConnection(FLAG_OPEN_PRODUCER, name) self.conn.queue_operation(op) return op def add_stream(self, vbucket, flags): op = AddStream(vbucket, flags) self.conn.queue_operation(op) return op def close_stream(self, vbucket): op = CloseStream(vbucket) self.conn.queue_operation(op) return op def get_failover_log(self, vbucket): op = GetFailoverLog(vbucket) self.conn.queue_operation(op) return op def stream_req(self, vb, flags, start_seqno, end_seqno, vb_uuid, hi_seqno): op = StreamRequest(vb, flags, start_seqno, end_seqno, vb_uuid, hi_seqno) self.conn.queue_operation(op) return op def shutdown(self): self.conn.close()