def setUpModule(): global keyspace_env global shard_0_master global shard_1_master global lookup_master logging.debug('in setUpModule') try: environment.topo_server().setup() logging.debug('Setting up tablets') keyspace_env = keyspace_util.TestEnv() keyspace_env.launch( 'user', shards=['-80', '80-'], ddls=[ create_vt_user, create_vt_user2, create_vt_user_extra, create_vt_music, create_vt_music_extra, create_join_user, create_join_user_extra, create_join_name_info, create_twopc_user, ], rdonly_count=1, # to test SplitQuery twopc_coordinator_address='localhost:15028', # enables 2pc ) keyspace_env.launch( 'lookup', ddls=[ create_vt_user_seq, create_vt_music_seq, create_vt_main_seq, create_music_user_map, create_name_user2_map, create_main, create_twopc_lookup, ], twopc_coordinator_address='localhost:15028', # enables 2pc ) shard_0_master = keyspace_env.tablet_map['user.-80.master'] shard_1_master = keyspace_env.tablet_map['user.80-.master'] lookup_master = keyspace_env.tablet_map['lookup.0.master'] utils.apply_vschema(vschema) utils.VtGate().start( tablets=[shard_0_master, shard_1_master, lookup_master], extra_args=['-transaction_mode', 'TWOPC']) utils.vtgate.wait_for_endpoints('user.-80.master', 1) utils.vtgate.wait_for_endpoints('user.80-.master', 1) utils.vtgate.wait_for_endpoints('lookup.0.master', 1) except: tearDownModule() raise
def setUpModule(): global keyspace_env global shard_0_master global shard_1_master global lookup_master logging.debug('in setUpModule') try: environment.topo_server().setup() logging.debug('Setting up tablets') keyspace_env = keyspace_util.TestEnv() keyspace_env.launch( 'user', shards=['-80', '80-'], ddls=[ create_vt_user, create_vt_user2, create_vt_user_extra, create_vt_music, create_vt_music_extra, create_join_user, create_join_user_extra, create_join_name_info, ], rdonly_count=1, # to test SplitQuery ) keyspace_env.launch( 'lookup', ddls=[ create_vt_user_seq, create_vt_music_seq, create_vt_main_seq, create_music_user_map, create_name_user2_map, create_main, ], ) shard_0_master = keyspace_env.tablet_map['user.-80.master'] shard_1_master = keyspace_env.tablet_map['user.80-.master'] lookup_master = keyspace_env.tablet_map['lookup.0.master'] utils.apply_vschema(vschema) utils.VtGate().start( tablets=[shard_0_master, shard_1_master, lookup_master]) utils.vtgate.wait_for_endpoints('user.-80.master', 1) utils.vtgate.wait_for_endpoints('user.80-.master', 1) utils.vtgate.wait_for_endpoints('lookup.0.master', 1) except: tearDownModule() raise
def main(): parser = optparse.OptionParser(usage='usage: %prog [options]') utils.add_options(parser) (options, args) = parser.parse_args() options.debug = True utils.set_options(options) env = keyspace_util.TestEnv() try: environment.topo_server().setup() env.launch( 'user', shards=['-80', '80-'], ddls=[ 'create table user(user_id bigint, name varchar(128), ' 'primary key(user_id))', 'create table user_extra(user_id bigint, extra varchar(128), ' 'primary key(user_id))', 'create table music(user_id bigint, music_id bigint, ' 'primary key(user_id, music_id))', 'create table music_extra(music_id bigint, ' 'keyspace_id bigint unsigned, primary key(music_id))', ], ) env.launch( 'lookup', ddls=[ 'create table user_idx(user_id bigint not null auto_increment, ' 'primary key(user_id))', 'create table name_user_idx(name varchar(128), user_id bigint, ' 'primary key(name, user_id))', 'create table music_user_idx(music_id bigint not null ' 'auto_increment, user_id bigint, primary key(music_id))', ], ) utils.apply_vschema(vschema) utils.VtGate().start(cache_ttl='500s') utils.Vtctld().start() print 'vtgate:', utils.vtgate.port print 'vtctld:', utils.vtctld.port utils.pause('the cluster is up, press enter to shut it down...') finally: env.teardown() utils.kill_sub_processes() utils.remove_tmp_files() environment.topo_server().teardown()
def setUpModule(): global keyspace_env global shard_0_master global shard_1_master global lookup_master logging.debug('in setUpModule') try: environment.topo_server().setup() logging.debug('Setting up tablets') keyspace_env = keyspace_util.TestEnv() keyspace_env.launch( 'user', shards=['-80', '80-'], ddls=[ create_vt_user, create_vt_user2, create_vt_user_extra, create_vt_music, create_vt_music_extra, create_join_user, create_join_user_extra, ], ) keyspace_env.launch( 'lookup', ddls=[ create_vt_user_seq, create_vt_music_seq, create_music_user_map, create_name_user2_map, ], ) shard_0_master = keyspace_env.tablet_map['user.-80.master'] shard_1_master = keyspace_env.tablet_map['user.80-.master'] lookup_master = keyspace_env.tablet_map['lookup.0.master'] utils.apply_vschema(vschema) utils.VtGate().start() except: tearDownModule() raise
def setUpModule(): global keyspace_env global shard_0_master global shard_0_replica global shard_1_master global lookup_master logging.debug('in setUpModule') try: environment.topo_server().setup() logging.debug('Setting up tablets') keyspace_env = keyspace_util.TestEnv() keyspace_env.launch( 'user', shards=['-80', '80-'], ddls=[ create_sharded_message, ], ) keyspace_env.launch( 'lookup', ddls=[ create_unsharded_message, ], ) shard_0_master = keyspace_env.tablet_map['user.-80.master'] shard_0_replica = keyspace_env.tablet_map['user.-80.replica.0'] shard_1_master = keyspace_env.tablet_map['user.80-.master'] lookup_master = keyspace_env.tablet_map['lookup.0.master'] utils.apply_vschema(vschema) utils.VtGate().start(tablets=[ shard_0_master, shard_0_replica, shard_1_master, lookup_master ]) utils.vtgate.wait_for_endpoints('user.-80.master', 1) utils.vtgate.wait_for_endpoints('user.-80.replica', 1) utils.vtgate.wait_for_endpoints('user.80-.master', 1) utils.vtgate.wait_for_endpoints('lookup.0.master', 1) except: tearDownModule() raise
def setUpModule(): global keyspace_env global shard_0_master global shard_1_master global lookup_master global vtgate_server global vtgate_port logging.debug("in setUpModule") try: environment.topo_server().setup() logging.debug("Setting up tablets") keyspace_env = keyspace_util.TestEnv() keyspace_env.launch( "user", shards=["-80", "80-"], ddls=[ create_vt_user, create_vt_user2, create_vt_user_extra, create_vt_music, create_vt_music_extra, ], ) keyspace_env.launch( "lookup", ddls=[ create_vt_user_idx, create_music_user_map, create_name_user2_map, ], ) shard_0_master = keyspace_env.tablet_map["user.-80.master"] shard_1_master = keyspace_env.tablet_map["user.80-.master"] lookup_master = keyspace_env.tablet_map["lookup.0.master"] utils.apply_vschema(schema) vtgate_server, vtgate_port = utils.vtgate_start() except: tearDownModule() raise
def setUpModule(): global keyspace_env global shard_0_master try: environment.topo_server().setup() keyspace_env = keyspace_util.TestEnv() keyspace_env.launch('user', ddls=[ create_vt_user, create_main, ]) shard_0_master = keyspace_env.tablet_map['user.0.master'] utils.VtGate().start( tablets=[shard_0_master], extra_args=['-vschema_ddl_authorized_users', '%'], ) utils.vtgate.wait_for_endpoints('user.0.master', 1) except: tearDownModule() raise
def main(): parser = optparse.OptionParser(usage="usage: %prog [options]") utils.add_options(parser) (options, args) = parser.parse_args() options.debug = True utils.set_options(options) env = keyspace_util.TestEnv() vtgate_server = None try: environment.topo_server().setup() env.launch( "user", shards=["-80", "80-"], ddls=[ 'create table user(user_id bigint, name varchar(128), primary key(user_id))', 'create table user_extra(user_id bigint, extra varchar(128), primary key(user_id))', 'create table music(user_id bigint, music_id bigint, primary key(user_id, music_id))', 'create table music_extra(music_id bigint, keyspace_id bigint unsigned, primary key(music_id))', ], ) env.launch( "lookup", ddls=[ 'create table user_idx(user_id bigint not null auto_increment, primary key(user_id))', 'create table name_user_idx(name varchar(128), user_id bigint, primary key(name, user_id))', 'create table music_user_idx(music_id bigint not null auto_increment, user_id bigint, primary key(music_id))', ], ) utils.apply_vschema(vschema) vtgate_server, vtgate_port = utils.vtgate_start(cache_ttl='500s') utils.Vtctld().start() print "vtgate:", vtgate_port print "vtctld:", utils.vtctld.port utils.pause("the cluster is up, press enter to shut it down...") finally: utils.vtgate_kill(vtgate_server) env.teardown() utils.kill_sub_processes() utils.remove_tmp_files() environment.topo_server().teardown()