source: vbox/trunk/src/VBox/ValidationKit/testmanager/debug/functions.pgsql@ 76776

-- $Id: functions.pgsql 76553 2019-01-01 01:45:53Z vboxsync $
--- @file
-- ?????????????????????????
--

--
-- Copyright (C) 2012-2019 Oracle Corporation
--
-- This file is part of VirtualBox Open Source Edition (OSE), as
-- available from http://www.virtualbox.org. This file is free software;
-- you can redistribute it and/or modify it under the terms of the GNU
-- General Public License (GPL) as published by the Free Software
-- Foundation, in version 2 as it comes in the "COPYING" file of the
-- VirtualBox OSE distribution. VirtualBox OSE is distributed in the
-- hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
--
-- The contents of this file may alternatively be used under the terms
-- of the Common Development and Distribution License Version 1.0
-- (CDDL) only, as it comes in the "COPYING.CDDL" file of the
-- VirtualBox OSE distribution, in which case the provisions of the
-- CDDL are applicable instead of those of the GPL.
--
-- You may elect to license modified versions of this file under the
-- terms and conditions of either the GPL or the CDDL or both.
--

\connect testmanager;

DROP FUNCTION authenticate_testbox(inet, uuid);
DROP FUNCTION testbox_status_set(integer, TestBoxState_T);

-- Authenticate Test Box record by IP and UUID and set its state to IDLE
-- Args: IP, UUID
CREATE OR REPLACE FUNCTION authenticate_testbox(inet, uuid) RETURNS testboxes AS $$
    DECLARE
        _ip ALIAS FOR $1;
        _uuidSystem ALIAS FOR $2;
        _box TestBoxes;
    BEGIN
        -- Find Test Box record
        SELECT *
            FROM testboxes
            WHERE ip=_ip AND uuidSystem=_uuidSystem INTO _box;
        IF FOUND THEN
            -- Update Test Box status if exists
            UPDATE TestBoxStatuses SET enmState='idle' WHERE idTestBox=_box.idTestBox;
            IF NOT FOUND THEN
                -- Otherwise, add new record to TestBoxStatuses table
                INSERT
                    INTO TestBoxStatuses(idTestBox, idGenTestBox, enmState)
                    VALUES (_box.idTestBox, _box.idGenTestBox, 'idle');
            END IF;
        END IF;
        return _box;
    END;
$$ LANGUAGE plpgsql;

-- Set Test Box status and make sure if it has been set
-- Args: Test Box ID, new status
CREATE OR REPLACE FUNCTION testbox_status_set(integer, TestBoxState_T) RETURNS VOID AS $$
    DECLARE
        _box ALIAS FOR $1;
        _status ALIAS FOR $2;
    BEGIN
        -- Update Test Box status if exists
        UPDATE TestBoxStatuses SET enmState=_status WHERE idTestBox=_box;
        IF NOT FOUND THEN
            RAISE EXCEPTION 'Test Box (#%) was not found in database', _box;
        END IF;
    END;
$$ LANGUAGE plpgsql;