You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
51 lines
2.2 KiB
51 lines
2.2 KiB
4 months ago
|
DOWN_SQL = """
|
||
|
|
ALTER TABLE jobs ADD COLUMN synchronizing tinyint(1) default NULL;
|
||
|
|
ALTER TABLE autotests ADD COLUMN synch_type smallint(6) NOT NULL;
|
||
|
|
UPDATE autotests SET synch_type = 1;
|
||
|
|
UPDATE autotests SET synch_type = 2 WHERE sync_count > 1;
|
||
|
|
ALTER TABLE jobs ADD COLUMN synch_type int(11) default NULL;
|
||
|
|
UPDATE jobs SET synch_type = 1;
|
||
|
|
UPDATE jobs SET synch_type = 2 WHERE synch_count > 1;
|
||
|
|
ALTER TABLE host_queue_entries DROP COLUMN `execution_subdir`;
|
||
|
|
"""
|
||
|
|
|
||
|
|
def migrate_up(manager):
|
||
|
|
# add execution_subdir field
|
||
|
|
manager.execute("""ALTER TABLE host_queue_entries ADD COLUMN
|
||
|
|
`execution_subdir` varchar(255) NOT NULL""")
|
||
|
|
|
||
|
|
# fill in execution_subdir field for running/complete entries
|
||
|
|
rows = manager.execute("""
|
||
|
|
SELECT jobs.id, jobs.synch_type, COUNT(1) FROM jobs
|
||
|
|
INNER JOIN host_queue_entries AS hqe ON jobs.id = hqe.job_id
|
||
|
|
GROUP BY jobs.id""")
|
||
|
|
job_hqe_count = dict((row[0], row[2]) for row in rows)
|
||
|
|
synch_jobs = set(row[0] for row in rows if row[1] == 2)
|
||
|
|
hqes = manager.execute("""
|
||
|
|
SELECT hqe.id, hqe.job_id, hqe.status, hqe.complete, hosts.hostname
|
||
|
|
FROM host_queue_entries AS hqe
|
||
|
|
INNER JOIN hosts ON hqe.host_id = hosts.id
|
||
|
|
WHERE hqe.status IN ('Starting', 'Running') OR complete""")
|
||
|
|
for id, job_id, status, complete, hostname in hqes:
|
||
|
|
if job_id in synch_jobs or job_hqe_count[job_id] == 1:
|
||
|
|
execution_subdir = ''
|
||
|
|
else:
|
||
|
|
execution_subdir = hostname
|
||
|
|
manager.execute(
|
||
|
|
'UPDATE host_queue_entries SET execution_subdir = %s WHERE id = %s',
|
||
|
|
execution_subdir, id)
|
||
|
|
|
||
|
|
# ensure synch_type information doesn't get lost if we need to migrate down
|
||
|
|
manager.execute('UPDATE jobs SET synch_count = 1 WHERE synch_type = 1')
|
||
|
|
manager.execute('UPDATE jobs SET synch_count = 2 '
|
||
|
|
'WHERE synch_type = 2 AND synch_count = 1')
|
||
|
|
# drop the old synch_type fields
|
||
|
|
manager.execute('ALTER TABLE jobs DROP COLUMN synch_type')
|
||
|
|
manager.execute('ALTER TABLE autotests DROP COLUMN synch_type')
|
||
|
|
# drop deprecated synchronizing field
|
||
|
|
manager.execute('ALTER TABLE jobs DROP COLUMN synchronizing')
|
||
|
|
|
||
|
|
|
||
|
|
def migrate_down(manager):
|
||
|
|
manager.execute_script(DOWN_SQL)
|