145 lines
4.3 KiB
Python
145 lines
4.3 KiB
Python
# Copyright 2017 AT&T Intellectual Property. All other rights reserved.
|
|
#
|
|
# 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.
|
|
"""Definitions for Drydock database tables."""
|
|
import copy
|
|
|
|
from sqlalchemy.schema import Table, Column
|
|
from sqlalchemy.types import Boolean, DateTime, String, Integer, Text
|
|
from sqlalchemy.dialects import postgresql as pg
|
|
|
|
|
|
class ExtendTable(Table):
|
|
def __new__(cls, metadata):
|
|
self = super().__new__(cls, cls.__tablename__, metadata,
|
|
*cls.__schema__)
|
|
return self
|
|
|
|
|
|
class Tasks(ExtendTable):
|
|
"""Table for persisting Tasks."""
|
|
|
|
__tablename__ = 'tasks'
|
|
|
|
__baseschema__ = [
|
|
Column('task_id', pg.BYTEA(16), primary_key=True),
|
|
Column('parent_task_id', pg.BYTEA(16)),
|
|
Column('subtask_id_list', pg.ARRAY(pg.BYTEA(16))),
|
|
Column('result_status', String(32)),
|
|
Column('result_message', String(128)),
|
|
Column('result_reason', String(128)),
|
|
Column('result_error_count', Integer),
|
|
Column('result_successes', pg.ARRAY(String(32))),
|
|
Column('result_failures', pg.ARRAY(String(32))),
|
|
Column('retry', Integer),
|
|
Column('status', String(32)),
|
|
Column('created', DateTime),
|
|
Column('created_by', String(16)),
|
|
Column('updated', DateTime),
|
|
Column('design_ref', String(128)),
|
|
Column('request_context', pg.JSON),
|
|
Column('node_filter', pg.JSON),
|
|
Column('action', String(32)),
|
|
Column('terminated', DateTime),
|
|
Column('terminated_by', String(16)),
|
|
Column('terminate', Boolean, default=False)
|
|
]
|
|
|
|
__add_result_links__ = [
|
|
Column('result_links', pg.JSON),
|
|
]
|
|
|
|
__schema__ = copy.copy(__baseschema__)
|
|
__schema__.extend(__add_result_links__)
|
|
|
|
|
|
class ResultMessage(ExtendTable):
|
|
"""Table for tracking result/status messages."""
|
|
|
|
__tablename__ = 'result_message'
|
|
|
|
__baseschema__ = [
|
|
Column('sequence', Integer, primary_key=True),
|
|
Column('task_id', pg.BYTEA(16)),
|
|
Column('message', String(1024)),
|
|
Column('error', Boolean),
|
|
Column('context', String(64)),
|
|
Column('context_type', String(16)),
|
|
Column('ts', DateTime),
|
|
Column('extra', pg.JSON)
|
|
]
|
|
|
|
__schema__ = copy.copy(__baseschema__)
|
|
|
|
|
|
class ActiveInstance(ExtendTable):
|
|
"""Table to organize multiple orchestrator instances."""
|
|
|
|
__tablename__ = 'active_instance'
|
|
|
|
__baseschema__ = [
|
|
Column('dummy_key', Integer, primary_key=True),
|
|
Column('identity', pg.BYTEA(16)),
|
|
Column('last_ping', DateTime),
|
|
]
|
|
|
|
__schema__ = copy.copy(__baseschema__)
|
|
|
|
|
|
class BootAction(ExtendTable):
|
|
"""Table persisting node build data."""
|
|
|
|
__tablename__ = 'boot_action'
|
|
|
|
__baseschema__ = [
|
|
Column('node_name', String(280), primary_key=True),
|
|
Column('task_id', pg.BYTEA(16)),
|
|
Column('identity_key', pg.BYTEA(32)),
|
|
]
|
|
|
|
__schema__ = copy.copy(__baseschema__)
|
|
|
|
|
|
class BootActionStatus(ExtendTable):
|
|
"""Table tracking status of node boot actions."""
|
|
|
|
__tablename__ = 'boot_action_status'
|
|
|
|
__baseschema__ = [
|
|
Column('node_name', String(280), index=True),
|
|
Column('action_id', pg.BYTEA(16), primary_key=True),
|
|
Column('action_name', String(64)),
|
|
Column('task_id', pg.BYTEA(16), index=True),
|
|
Column('identity_key', pg.BYTEA(32)),
|
|
Column('action_status', String(32)),
|
|
]
|
|
|
|
__schema__ = copy.copy(__baseschema__)
|
|
|
|
|
|
class BuildData(ExtendTable):
|
|
"""Table for persisting node build data."""
|
|
|
|
__tablename__ = 'build_data'
|
|
|
|
__baseschema__ = [
|
|
Column('node_name', String(32), index=True),
|
|
Column('task_id', pg.BYTEA(16), index=True),
|
|
Column('collected_date', DateTime),
|
|
Column('generator', String(256)),
|
|
Column('data_format', String(32)),
|
|
Column('data_element', Text),
|
|
]
|
|
|
|
__schema__ = copy.copy(__baseschema__)
|