def testMcas(self):
   mc = GetMemcacheClient()
   key1, key2 = uuid4().hex, uuid4().hex
   mc.set_multi({key1: 'foo', key2: 'bar'})
   item1, item2 = mcas_get(mc, key1), mcas_get(mc, key2)
   self.assertTrue(mcas(mc, [(item1, 'foo2'), (item2, 'bar2')]))
   self.assertEqual(mc.get_multi([key1, key2]), {key1: 'foo2', key2: 'bar2'})
 def testMcas(self):
   mc = GetMemcacheClient()
   key1, key2 = uuid4().hex, uuid4().hex
   mc.set_multi({key1: 'foo', key2: 'bar'})
   item1, item2 = mcas_get(mc, key1), mcas_get(mc, key2)
   self.assertTrue(mcas(mc, [(item1, 'foo2'), (item2, 'bar2')]))
   self.assertEqual(mc.get_multi([key1, key2]), {key1: 'foo2', key2: 'bar2'})
 def testMcasGet(self):
   mc = GetMemcacheClient()
   key = uuid4().hex
   value = 'foo'
   self.assertTrue(mc.set(key, value))
   item = mcas_get(mc, key)
   self.assertEqual(key, item.key)
   self.assertEqual(value, item.value)
   self.assertTrue(mc.set(key, 'bar'))
   item2 = mcas_get(mc, key)
   self.assertNotEqual(item.cas_id, item2.cas_id)
 def testMcasGet(self):
     mc = GetMemcacheClient()
     key = uuid4().hex
     value = 'foo'
     self.assertTrue(mc.set(key, value))
     item = mcas_get(mc, key)
     self.assertEqual(key, item.key)
     self.assertEqual(value, item.value)
     self.assertTrue(mc.set(key, 'bar'))
     item2 = mcas_get(mc, key)
     self.assertNotEqual(item.cas_id, item2.cas_id)
def worker():
  QueueManager.register(COMMAND_QUEUE)
  m = QueueManager(address=('', CONTROL_PORT), authkey=CONTROL_KEY)
  m.connect()
  command_queue = getattr(m, COMMAND_QUEUE)()
  r = Random()
  r.seed(time.time())

  while True:
    command = command_queue.get()
    mc = GetMemcacheClient(command['servers'])
    location_keys = command['location_keys']
    num_locations_per_mcas = command['num_locations_per_mcas']
    batch_size = command['batch_size']
    print('worker received command')
    batch_time = time.time()
    num_retries = 0
    for i in range(batch_size):
      op_time = time.time()
      mcas_keys = r.sample(location_keys, num_locations_per_mcas)
      while True:
        if mcas(mc, ((item, item.value + 1)
            for item in (mcas_get(mc, key) for key in mcas_keys))):
          break
        else:
          num_retries += 1
      if TARGET_RATE:
        time.sleep(max(0, 1 / TARGET_RATE - (time.time() - op_time)))
    print('done: %.1f operations per second, %s retries' % (
         batch_size/(time.time() - batch_time), num_retries))
    command_queue.task_done()
def worker():
    QueueManager.register(COMMAND_QUEUE)
    m = QueueManager(address=('', CONTROL_PORT), authkey=CONTROL_KEY)
    m.connect()
    command_queue = getattr(m, COMMAND_QUEUE)()
    r = Random()
    r.seed(time.time())

    while True:
        command = command_queue.get()
        mc = GetMemcacheClient(command['servers'])
        location_keys = command['location_keys']
        num_locations_per_mcas = command['num_locations_per_mcas']
        batch_size = command['batch_size']
        print('worker received command')
        batch_time = time.time()
        num_retries = 0
        for i in range(batch_size):
            op_time = time.time()
            mcas_keys = r.sample(location_keys, num_locations_per_mcas)
            while True:
                if mcas(mc, ((item, item.value + 1)
                             for item in (mcas_get(mc, key)
                                          for key in mcas_keys))):
                    break
                else:
                    num_retries += 1
            if TARGET_RATE:
                time.sleep(max(0, 1 / TARGET_RATE - (time.time() - op_time)))
        print('done: %.1f operations per second, %s retries' %
              (batch_size / (time.time() - batch_time), num_retries))
        command_queue.task_done()