зеркало из https://github.com/github/vitess-gh.git
106 строки
3.4 KiB
Python
106 строки
3.4 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# Copyright 2015, Google Inc. All rights reserved.
|
|
# Use of this source code is governed by a BSD-style license that can
|
|
# be found in the LICENSE file.
|
|
"""This module allows you to bring up and tear down keyspaces.
|
|
"""
|
|
|
|
import os
|
|
|
|
import environment
|
|
import tablet
|
|
import utils
|
|
|
|
|
|
class TestEnv(object):
|
|
"""Main class for this module."""
|
|
|
|
def __init__(self):
|
|
self.tablet_map = {}
|
|
|
|
def launch(
|
|
self, keyspace, shards=None, replica_count=1, rdonly_count=0, ddls=None):
|
|
"""Launch test environment."""
|
|
|
|
if replica_count < 1:
|
|
raise Exception('replica_count=%d < 1; tests now use semi-sync'
|
|
' and must have at least one replica' % replica_count)
|
|
self.tablets = []
|
|
self.master_tablets = []
|
|
utils.run_vtctl(['CreateKeyspace', keyspace])
|
|
if not shards or shards[0] == '0':
|
|
shards = ['0']
|
|
else:
|
|
utils.run_vtctl(
|
|
['SetKeyspaceShardingInfo', '-force', keyspace, 'keyspace_id',
|
|
'uint64'])
|
|
|
|
# Create tablets and start mysqld.
|
|
procs = []
|
|
for shard in shards:
|
|
procs.append(self._new_tablet(keyspace, shard, 'master', None))
|
|
for i in xrange(replica_count):
|
|
procs.append(self._new_tablet(keyspace, shard, 'replica', i))
|
|
for i in xrange(rdonly_count):
|
|
procs.append(self._new_tablet(keyspace, shard, 'rdonly', i))
|
|
utils.wait_procs(procs)
|
|
|
|
# Start tablets.
|
|
for shard in shards:
|
|
self._start_tablet(keyspace, shard, 'master', None)
|
|
for i in xrange(replica_count):
|
|
self._start_tablet(keyspace, shard, 'replica', i)
|
|
for i in xrange(rdonly_count):
|
|
self._start_tablet(keyspace, shard, 'rdonly', i)
|
|
|
|
# Start replication.
|
|
for t in self.tablets:
|
|
t.wait_for_vttablet_state('NOT_SERVING')
|
|
for t in self.master_tablets:
|
|
utils.run_vtctl(['InitShardMaster', '-force', keyspace+'/'+t.shard,
|
|
t.tablet_alias], auto_log=True)
|
|
for t in self.tablets:
|
|
t.wait_for_vttablet_state('SERVING')
|
|
|
|
for ddl in ddls:
|
|
fname = os.path.join(environment.tmproot, 'ddl.sql')
|
|
with open(fname, 'w') as f:
|
|
f.write(ddl)
|
|
utils.run_vtctl(['ApplySchema', '-sql-file', fname, keyspace])
|
|
|
|
def teardown(self):
|
|
all_tablets = self.tablet_map.values()
|
|
tablet.kill_tablets(all_tablets)
|
|
teardown_procs = [t.teardown_mysql() for t in all_tablets]
|
|
utils.wait_procs(teardown_procs, raise_on_error=False)
|
|
for t in all_tablets:
|
|
t.remove_tree()
|
|
|
|
def _new_tablet(self, keyspace, shard, tablet_type, index):
|
|
"""Create a tablet and start mysqld."""
|
|
t = tablet.Tablet()
|
|
self.tablets.append(t)
|
|
if tablet_type == 'master':
|
|
self.master_tablets.append(t)
|
|
key = '%s.%s.%s' %(keyspace, shard, tablet_type)
|
|
else:
|
|
key = '%s.%s.%s.%s' % (keyspace, shard, tablet_type, index)
|
|
self.tablet_map[key] = t
|
|
return t.init_mysql()
|
|
|
|
def _start_tablet(self, keyspace, shard, tablet_type, index):
|
|
"""Start a tablet."""
|
|
target_tablet_type = tablet_type
|
|
if tablet_type == 'master':
|
|
target_tablet_type = 'replica'
|
|
key = '%s.%s.%s' %(keyspace, shard, tablet_type)
|
|
else:
|
|
key = '%s.%s.%s.%s' % (keyspace, shard, tablet_type, index)
|
|
t = self.tablet_map[key]
|
|
return t.start_vttablet(
|
|
wait_for_state=None, target_tablet_type=target_tablet_type,
|
|
init_keyspace=keyspace, init_shard=shard,
|
|
extra_args=['-queryserver-config-schema-reload-time', '1'])
|