105 lines
3.8 KiB
Python
105 lines
3.8 KiB
Python
# Copyright 2014 Rackspace Australia
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import logging
|
|
|
|
import alembic
|
|
import alembic.config
|
|
import sqlalchemy as sa
|
|
import voluptuous as v
|
|
|
|
from zuul.connection import BaseConnection
|
|
|
|
BUILDSET_TABLE = 'zuul_buildset'
|
|
BUILD_TABLE = 'zuul_build'
|
|
|
|
|
|
class SQLConnection(BaseConnection):
|
|
driver_name = 'sql'
|
|
log = logging.getLogger("connection.sql")
|
|
|
|
def __init__(self, connection_name, connection_config):
|
|
|
|
super(SQLConnection, self).__init__(connection_name, connection_config)
|
|
|
|
self.dburi = None
|
|
self.engine = None
|
|
self.connection = None
|
|
self.tables_established = False
|
|
try:
|
|
self.dburi = self.connection_config.get('dburi')
|
|
self.engine = sa.create_engine(self.dburi)
|
|
self._migrate()
|
|
self._setup_tables()
|
|
self.tables_established = True
|
|
except sa.exc.NoSuchModuleError:
|
|
self.log.exception(
|
|
"The required module for the dburi dialect isn't available. "
|
|
"SQL connection %s will be unavailable." % connection_name)
|
|
except sa.exc.OperationalError:
|
|
self.log.exception(
|
|
"Unable to connect to the database or establish the required "
|
|
"tables. Reporter %s is disabled" % self)
|
|
|
|
def _migrate(self):
|
|
"""Perform the alembic migrations for this connection"""
|
|
with self.engine.begin() as conn:
|
|
context = alembic.migration.MigrationContext.configure(conn)
|
|
current_rev = context.get_current_revision()
|
|
self.log.debug('Current migration revision: %s' % current_rev)
|
|
|
|
config = alembic.config.Config()
|
|
config.set_main_option("script_location",
|
|
"zuul:alembic/sql_reporter")
|
|
config.set_main_option("sqlalchemy.url",
|
|
self.connection_config.get('dburi'))
|
|
|
|
alembic.command.upgrade(config, 'head')
|
|
|
|
def _setup_tables(self):
|
|
metadata = sa.MetaData()
|
|
|
|
self.zuul_buildset_table = sa.Table(
|
|
BUILDSET_TABLE, metadata,
|
|
sa.Column('id', sa.Integer, primary_key=True),
|
|
sa.Column('zuul_ref', sa.String(255)),
|
|
sa.Column('pipeline', sa.String(255)),
|
|
sa.Column('project', sa.String(255)),
|
|
sa.Column('change', sa.Integer, nullable=True),
|
|
sa.Column('patchset', sa.Integer, nullable=True),
|
|
sa.Column('ref', sa.String(255)),
|
|
sa.Column('score', sa.Integer),
|
|
sa.Column('message', sa.TEXT()),
|
|
)
|
|
|
|
self.zuul_build_table = sa.Table(
|
|
BUILD_TABLE, metadata,
|
|
sa.Column('id', sa.Integer, primary_key=True),
|
|
sa.Column('buildset_id', sa.Integer,
|
|
sa.ForeignKey(BUILDSET_TABLE + ".id")),
|
|
sa.Column('uuid', sa.String(36)),
|
|
sa.Column('job_name', sa.String(255)),
|
|
sa.Column('result', sa.String(255)),
|
|
sa.Column('start_time', sa.DateTime()),
|
|
sa.Column('end_time', sa.DateTime()),
|
|
sa.Column('voting', sa.Boolean),
|
|
sa.Column('log_url', sa.String(255)),
|
|
sa.Column('node_name', sa.String(255)),
|
|
)
|
|
|
|
|
|
def getSchema():
|
|
sql_connection = v.Any(str, v.Schema({}, extra=True))
|
|
return sql_connection
|