aboutsummaryrefslogtreecommitdiff
blob: 1cf2439319059caca4f6857c983b5c26268ff9e2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
# This file has parts from Buildbot and is modifyed by Gentoo Authors. 
# Buildbot is free software: you can redistribute it and/or modify it 
# under the terms of the GNU General Public License as published by the 
# Free Software Foundation, version 2.
#
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
# details.
#
# You should have received a copy of the GNU General Public License along with
# this program; if not, write to the Free Software Foundation, Inc., 51
# Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
#
# Copyright Buildbot Team Members
# Origins: buildbot.db.*
# Modifyed by Gentoo Authors.
# Copyright 2021 Gentoo Authors

import uuid
import sqlalchemy as sa

from twisted.internet import defer

from buildbot.db import base

class BuildsConnectorComponent(base.DBConnectorComponent):

    #@defer.inlineCallbacks
    def addBuild(self, project_build_data):
        created_at = int(self.master.reactor.seconds())
        def thd(conn, no_recurse=False):
            tbl = self.db.model.projects_builds
            # get the highest current number
            r = conn.execute(sa.select([sa.func.max(tbl.c.build_id)],
                                       whereclause=(tbl.c.project_uuid == project_build_data['project_uuid'])))
            number = r.scalar()
            new_number = 1 if number is None else number + 1
            try:
                q = tbl.insert()
                r = conn.execute(q, dict(project_uuid=project_build_data['project_uuid'],
                                         version_uuid=project_build_data['version_uuid'],
                                         status=project_build_data['status'],
                                         requested=project_build_data['requested'],
                                         created_at=created_at,
                                         buildbot_build_id=0,
                                         build_id=new_number))
            except (sa.exc.IntegrityError, sa.exc.ProgrammingError):
                id = None
                new_number = None
            else:
                id = r.inserted_primary_key[0]
            return id, new_number
        return self.db.pool.do(thd)

    @defer.inlineCallbacks
    def setStatusBuilds(self, id, status):
        updated_at = int(self.master.reactor.seconds())
        def thd(conn, no_recurse=False):
        
                tbl = self.db.model.projects_builds
                q = tbl.update()
                q = q.where(tbl.c.id == id)
                conn.execute(q, updated_at=updated_at,
                                status=status)
        yield self.db.pool.do(thd)

    @defer.inlineCallbacks
    def setBuildbotBuildIdBuilds(self, id, buildbot_build_id):
        updated_at = int(self.master.reactor.seconds())
        def thd(conn, no_recurse=False):
        
                tbl = self.db.model.projects_builds
                q = tbl.update()
                q = q.where(tbl.c.id == id)
                conn.execute(q, updated_at=updated_at,
                                buildbot_build_id=buildbot_build_id)
        yield self.db.pool.do(thd)