APT_Utility.pm.in 2.08 KB
Newer Older
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 79 80 81 82 83 84 85 86 87 88
#!/usr/bin/perl -wT
#
# Copyright (c) 2007-2016 University of Utah and the Flux Group.
# 
# {{{EMULAB-LICENSE
# 
# This file is part of the Emulab network testbed software.
# 
# This file is free software: you can redistribute it and/or modify it
# under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or (at
# your option) any later version.
# 
# This file 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 Affero General Public
# License for more details.
# 
# You should have received a copy of the GNU Affero General Public License
# along with this file.  If not, see <http://www.gnu.org/licenses/>.
# 
# }}}
#
#
# Stuff that has no where else to go.
#
package APT_Utility;

use strict;
use English;
use Data::Dumper;
use Carp;
use Exporter;
use vars qw(@ISA @EXPORT);

@ISA    = "Exporter";
@EXPORT = qw ( );

# Must come after package declaration!
use emdb;
use libtestbed;
use APT_Instance;
use Project;
use Group;

# Configure variables
my $TB		  = "@prefix@";
my $MAINSITE      = @TBMAINSITE@;
my $TBOPS         = "@TBOPSEMAIL@";

#
# Find all of the instances a user has (should have) an account on, and
# mark those instances for update.
#
sub UpdateInstancesByUser($)
{
    my ($user) = @_;
    my @projects  = ();
    my %instances = ();

    if ($user->ProjectMembershipList(\@projects)) {
	return -1;
    }
    return 0
	if (!@projects);

    foreach my $project (@projects) {
	my $pid_idx = $project->pid_idx();

	my $query_result =
	    DBQueryWarn("select uuid from apt_instances ".
			"where pid_idx='$pid_idx'");
	return -1
	    if (!$query_result);

	while (my ($uuid) = $query_result->fetchrow_array()) {
	    my $instance = APT_Instance->Lookup($uuid);
	    next
		if (!defined($instance));
	    $instances{$uuid} = $instance;
	}
    }
    # Update each instance only once. 
    foreach my $instance (values(%instances)) {
	$instance->Update({"needupdate" => 1});
    }
    return 0;
}