def test_100_basic(self):
handler = SnowflakeServiceHandler(0, 0)
ret1 = handler.get_id('me')
ret2 = handler.get_id('me')
self.assertGreater(ret1, 0)
self.assertGreater(ret2, 0)
self.assertGreater(ret2, ret1)
def test_102_long_run(self):
handler = SnowflakeServiceHandler(0, 0)
ret = set()
total_runs = 20000
for i in xrange(total_runs):
ret.add(handler.get_id('me'))
self.assertFalse(None in ret)
self.assertEquals(len(ret), total_runs)
def test_101_multiple(self):
handler1 = SnowflakeServiceHandler(0, 0)
handler2 = SnowflakeServiceHandler(1, 0)
ret1 = handler1.get_id('me')
ret2 = handler2.get_id('me')
self.assertGreater(ret1, 0)
self.assertGreater(ret2, 0)
self.assertNotEqual(ret2, ret1)
def main():
if len(sys.argv) < 2:
print 'Usage: %s config_file' % sys.argv[0]
sys.exit()
log.startLogging(sys.stdout)
Config.init(sys.argv[1])
if Config.debug:
log.startLogging(sys.stdout)
else:
log.startLogging(DailyLogFile.fromFullPath(Config.get('log.file')))
handler = SnowflakeServiceHandler(Config.getint('worker.id'),
Config.getint('datacenter.id'))
processor = SnowflakeService.Processor(handler)
server = TTwisted.ThriftServerFactory(
processor=processor,
iprot_factory=TBinaryProtocol.TBinaryProtocolFactory())
reactor.listenTCP(Config.getint('port', default=9999),
server,
interface=Config.get('listen', default="0.0.0.0"))
reactor.run()
