655 lines
19 KiB
Perl
655 lines
19 KiB
Perl
package Spectre::Workflow;
|
|
|
|
=head1 LEGAL
|
|
|
|
-------------------------------------------------------------------
|
|
WebGUI is Copyright 2001-2009 Plain Black Corporation.
|
|
-------------------------------------------------------------------
|
|
Please read the legal notices (docs/legal.txt) and the license
|
|
(docs/license.txt) that came with this distribution before using
|
|
this software.
|
|
-------------------------------------------------------------------
|
|
http://www.plainblack.com info@plainblack.com
|
|
-------------------------------------------------------------------
|
|
|
|
=cut
|
|
|
|
use strict;
|
|
use HTTP::Request::Common;
|
|
use HTTP::Cookies;
|
|
use POE qw(Component::Client::HTTP);
|
|
use POE::Queue::Array;
|
|
use Tie::IxHash;
|
|
use JSON qw/ encode_json /;
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 _start ( )
|
|
|
|
Initializes the workflow manager.
|
|
|
|
=cut
|
|
|
|
sub _start {
|
|
my ( $kernel, $self, $publicEvents) = @_[ KERNEL, OBJECT, ARG0 ];
|
|
$self->debug("Starting workflow manager.");
|
|
my $serviceName = "workflow";
|
|
$kernel->alias_set($serviceName);
|
|
$kernel->call( IKC => publish => $serviceName, $publicEvents );
|
|
$kernel->yield("checkInstances");
|
|
}
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 _stop ( )
|
|
|
|
Gracefully shuts down the workflow manager.
|
|
|
|
=cut
|
|
|
|
sub _stop {
|
|
my ($kernel, $self) = @_[KERNEL, OBJECT];
|
|
$self->debug("Stopping workflow manager.");
|
|
undef $self;
|
|
}
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 addInstance ( params )
|
|
|
|
Adds a workflow instance to the workflow processing queue.
|
|
|
|
=head3 params
|
|
|
|
A hash reference containing important information about the workflow instance to add to the queue.
|
|
|
|
=head4 sitename
|
|
|
|
The host and domain of the site this instance belongs to.
|
|
|
|
=head4 instanceId
|
|
|
|
The unqiue id for this workflow instance.
|
|
|
|
=head4 priority
|
|
|
|
The priority (1,2, or 3) that this instance should be run at.
|
|
|
|
=cut
|
|
|
|
sub addInstance {
|
|
my ($self, $instance) = @_[OBJECT, ARG0];
|
|
if ($instance->{priority} < 1 || $instance->{instanceId} eq "" || $instance->{sitename} eq "") {
|
|
$self->error("Can't add workflow instance with missing data: ". $instance->{sitename}." - ".$instance->{instanceId});
|
|
} else {
|
|
my $priority = ($instance->{priority} -1) * 10;
|
|
$instance->{lastState} = "never run";
|
|
$self->debug("Adding workflow instance ".$instance->{instanceId}." from ".$instance->{sitename}." to queue at priority ".$priority.".");
|
|
$self->getWaitingQueue->enqueue($priority, $instance);
|
|
}
|
|
}
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 checkInstances ( )
|
|
|
|
Checks to see if there are any open instance slots available, and if there are assigns a new instance to be run to fill it.
|
|
|
|
=cut
|
|
|
|
sub checkInstances {
|
|
my ($kernel, $self) = @_[KERNEL, OBJECT];
|
|
$self->debug("Checking to see if we can run anymore instances right now.");
|
|
if ($self->countRunningInstances < $self->config->get("maxWorkers")) {
|
|
$self->debug("Total workflows waiting to run: ".$self->getWaitingQueue->get_item_count);
|
|
my $instance = $self->getNextInstance;
|
|
if (defined $instance) {
|
|
$kernel->yield("runWorker",$instance);
|
|
}
|
|
}
|
|
$kernel->delay_set("checkInstances",$self->config->get("timeBetweenRunningWorkflows"));
|
|
}
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 config ( )
|
|
|
|
Returns a reference to the config object.
|
|
|
|
=cut
|
|
|
|
sub config {
|
|
my $self = shift;
|
|
return $self->{_config};
|
|
}
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 countRunningInstances ( )
|
|
|
|
Returns an integer representing the number of running instances.
|
|
|
|
=cut
|
|
|
|
sub countRunningInstances {
|
|
my $self = shift;
|
|
my $instanceCount = $self->getRunningQueue->get_item_count;
|
|
$self->debug("There are $instanceCount running instances.");
|
|
return $instanceCount;
|
|
}
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 debug ( output )
|
|
|
|
Prints out debug information if debug is enabled.
|
|
|
|
=head3 output
|
|
|
|
The debug message to be printed if debug is enabled.
|
|
|
|
=cut
|
|
|
|
sub debug {
|
|
my $self = shift;
|
|
my $output = shift;
|
|
if ($self->{_debug}) {
|
|
print "WORKFLOW: ".$output."\n";
|
|
}
|
|
$self->getLogger->debug("WORKFLOW: ".$output);
|
|
}
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 deleteInstance ( instanceId )
|
|
|
|
Removes a workflow instance from the processing queue.
|
|
|
|
=cut
|
|
|
|
sub deleteInstance {
|
|
my ($self, $instanceId,$kernel, $session ) = @_[OBJECT, ARG0, KERNEL, SESSION];
|
|
$self->debug("Deleting workflow instance $instanceId from queue.");
|
|
$self->getWaitingQueue->remove_items(
|
|
sub {
|
|
my $instance = shift;
|
|
return 1 if ($instance->{instanceId} eq $instanceId);
|
|
return 0;
|
|
}
|
|
);
|
|
$self->removeInstanceFromRunningQueue($instanceId);
|
|
}
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 editWorkflowPriority ( href )
|
|
|
|
Updates the priority of a given workflow instance.
|
|
|
|
=head3 href
|
|
|
|
Contains information about the instance and the new priority.
|
|
|
|
=head4 instanceId
|
|
|
|
The id of the instance to update.
|
|
|
|
=head4 newPriority
|
|
|
|
The new priority value.
|
|
|
|
=cut
|
|
|
|
sub editWorkflowPriority {
|
|
my ($self, $request, $kernel, $session ) = @_[OBJECT, ARG0, KERNEL, SESSION];
|
|
my ($argsHref, $rsvp) = @$request;
|
|
|
|
my $instanceId = $argsHref->{instanceId};
|
|
my $newPriority = $argsHref->{newPriority};
|
|
|
|
$self->debug("Updating the priority of $instanceId to $newPriority.");
|
|
|
|
# I'm guessing that the payload can't change queues on us
|
|
my $found = 0;
|
|
my $filterCref = sub { shift->{instanceId} eq $instanceId };
|
|
for my $getQueueMethod (map "get${_}Queue", qw( Suspended Waiting Running )) {
|
|
my $q = $self->$getQueueMethod;
|
|
my($itemAref) = $q->peek_items($filterCref); # there should be only one
|
|
|
|
next unless (ref $itemAref eq 'ARRAY' and @$itemAref);
|
|
|
|
my($priority, $id, $payload) = @$itemAref;
|
|
my $ackPriority = $q->set_priority($id, $filterCref, $newPriority);
|
|
if ($ackPriority != $newPriority) {
|
|
# return an error
|
|
my $error = 'edit priority setting error';
|
|
$kernel->call(IKC=>post=>$rsvp, encode_json({message => $error}));
|
|
}
|
|
$found = 1;
|
|
last;
|
|
}
|
|
|
|
if (! $found) {
|
|
# return an error message
|
|
my $error = 'edit priority instance not found error';
|
|
$kernel->call(IKC=>post=>$rsvp, encode_json({message => $error}));
|
|
}
|
|
else {
|
|
# return success message
|
|
$kernel->call(IKC=>post=>$rsvp, encode_json({message => 'edit priority success'}));
|
|
}
|
|
}
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 error ( output )
|
|
|
|
Prints out error information if debug is enabled.
|
|
|
|
=head3 output
|
|
|
|
The error message to be printed if debug is enabled.
|
|
|
|
=cut
|
|
|
|
sub error {
|
|
my $self = shift;
|
|
my $output = shift;
|
|
if ($self->{_debug}) {
|
|
print "WORKFLOW: [Error] ".$output."\n";
|
|
}
|
|
$self->getLogger->error("WORKFLOW: ".$output);
|
|
}
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 getJsonStatus ( )
|
|
|
|
Returns JSON report about the workflow engine.
|
|
|
|
=cut
|
|
|
|
sub getJsonStatus {
|
|
my ($kernel, $request, $self) = @_[KERNEL,ARG0,OBJECT];
|
|
my ($sitename, $rsvp) = @$request;
|
|
|
|
my %queues = ();
|
|
tie %queues, 'Tie::IxHash';
|
|
%queues = (
|
|
Suspended => $self->getSuspendedQueue,
|
|
Waiting => $self->getWaitingQueue,
|
|
Running => $self->getRunningQueue,
|
|
);
|
|
|
|
my %output = ();
|
|
foreach my $queueName (keys %queues) {
|
|
|
|
# get the queue name, and how many items it has
|
|
my $queue = $queues{$queueName};
|
|
my $count = $queue->get_item_count;
|
|
|
|
# list of instances to be added to the %output structure in the event
|
|
# that a site name is provided
|
|
my @instances;
|
|
|
|
# and if there are items in that queue, add them to our data structure
|
|
if ($count > 0) {
|
|
|
|
# if a site name is provided, only process data for that site name,
|
|
# and only construct data for that site name
|
|
if($sitename ne '') {
|
|
foreach my $itemAref ($queue->peek_items(sub { shift()->{sitename} eq $sitename })) {
|
|
push @instances, $itemAref;
|
|
}
|
|
$output{$queueName} = \@instances;
|
|
}
|
|
|
|
# otherwise, process data for all sites.
|
|
else {
|
|
foreach my $queueItem ($queue->peek_items(sub {1})) {
|
|
my($priority, $id, $instance) = @{$queueItem};
|
|
|
|
# The site's name in the list of %output keys isn't a hashref;
|
|
# we haven't seen it yet
|
|
if(ref $output{$instance->{sitename}} ne 'HASH') {
|
|
$output{$instance->{sitename}} = {};
|
|
}
|
|
|
|
# The queue name in the $output{sitename} hashref isn't an
|
|
# arrayref; we haven't seen it yet
|
|
if(ref $output{$instance->{sitename}}{$queueName} ne 'ARRAY') {
|
|
$output{$instance->{sitename}}{$queueName} = [];
|
|
}
|
|
|
|
# calculate originalPriority separately
|
|
$instance->{originalPriority} = ($instance->{priority} - 1) * 10;
|
|
|
|
# finally, add the instance to the returned data structure
|
|
push @{$output{$instance->{sitename}}{$queueName}}, $instance;
|
|
}
|
|
}
|
|
}
|
|
# there's no items in this queue, but the version of this call that
|
|
# accepts a sitename expects an empty array ref anyway. Give it one.
|
|
else {
|
|
if($sitename) {
|
|
$output{$queueName} = \@instances;
|
|
}
|
|
}
|
|
}
|
|
|
|
$kernel->call(IKC=>post=>$rsvp, encode_json(\%output));
|
|
}
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 getLogger ( )
|
|
|
|
Returns a reference to the logger.
|
|
|
|
=cut
|
|
|
|
sub getLogger {
|
|
my $self = shift;
|
|
return $self->{_logger};
|
|
}
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 getNextInstance ( )
|
|
|
|
Returns the next available instance.
|
|
|
|
=cut
|
|
|
|
sub getNextInstance {
|
|
my $self = shift;
|
|
$self->debug("Looking for a workflow instance to run.");
|
|
my $waiting = $self->getWaitingQueue;
|
|
if ($waiting->get_item_count > 0) {
|
|
my ($priority, $id, $instance) = $waiting->dequeue_next;
|
|
$instance->{workingPriority} = $priority;
|
|
$self->getRunningQueue->enqueue($priority, $instance);
|
|
$self->debug("Looks like ".$instance->{instanceId}." at priority $priority would be a good workflow instance to run.");
|
|
return $instance;
|
|
}
|
|
$self->debug("Didn't see any workflow instances to run.");
|
|
return undef;
|
|
}
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 getRunningQueue ( )
|
|
|
|
Returns a reference to the queue of workflow instances that are running now.
|
|
|
|
=cut
|
|
|
|
sub getRunningQueue {
|
|
my $self = shift;
|
|
return $self->{_runningQueue};
|
|
}
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 getStatus ( )
|
|
|
|
Returns a formatted text status report about the workflow engine.
|
|
|
|
=cut
|
|
|
|
sub getStatus {
|
|
my ($kernel, $request, $self) = @_[KERNEL,ARG0,OBJECT];
|
|
my $pattern = "\t%8.8s %-30.30s %-22.22s %-15.15s %-20.20s\n";
|
|
my $summaryPattern = "%19.19s %4d\n";
|
|
my %queues = ();
|
|
tie %queues, 'Tie::IxHash';
|
|
%queues = (
|
|
"Suspended" => $self->getSuspendedQueue,
|
|
"Waiting" => $self->getWaitingQueue,
|
|
"Running" => $self->getRunningQueue,
|
|
);
|
|
my $total = 0;
|
|
my $output = "";
|
|
foreach my $queueName (keys %queues) {
|
|
my $queue = $queues{$queueName};
|
|
my $count = $queue->get_item_count;
|
|
$output .= sprintf $summaryPattern, $queueName." Workflows", $count;
|
|
if ($count > 0) {
|
|
$output .= sprintf $pattern, "Priority", "Sitename", "Instance Id", "Last State", "Last Run Time";
|
|
foreach my $item ($queue->peek_items(sub {1})) {
|
|
my ($priority, $id, $instance) = @{$item};
|
|
my $originalPriority = ($instance->{priority} - 1) * 10;
|
|
$output .= sprintf $pattern, $priority."/".$originalPriority, $instance->{sitename}, $instance->{instanceId}, $instance->{lastState}, $instance->{lastRunTime};
|
|
}
|
|
$output .= "\n";
|
|
}
|
|
$total += $count;
|
|
}
|
|
$output .= sprintf $summaryPattern, "Total Workflows", $total;
|
|
my ($data, $rsvp) = @$request;
|
|
$kernel->call(IKC=>post=>$rsvp,$output);
|
|
}
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 getSuspendedQueue ( )
|
|
|
|
Returns a reference to the queue of workflow instances that have been suspended due to error or wait timeouts.
|
|
|
|
=cut
|
|
|
|
sub getSuspendedQueue {
|
|
my $self = shift;
|
|
return $self->{_suspendedQueue};
|
|
}
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 getWaitingQueue ( )
|
|
|
|
Returns a reference to the queue of workflow instances waiting to run.
|
|
|
|
=cut
|
|
|
|
sub getWaitingQueue {
|
|
my $self = shift;
|
|
return $self->{_waitingQueue};
|
|
}
|
|
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 new ( config, logger, [ , debug ] )
|
|
|
|
Constructor. Loads all active workflows from each WebGUI site and begins executing them.
|
|
|
|
=head3 config
|
|
|
|
The config object for spectre.
|
|
|
|
=head3 logger
|
|
|
|
A reference to the logger object.
|
|
|
|
=head3 debug
|
|
|
|
A boolean indicating Spectre should spew forth debug as it runs.
|
|
|
|
=cut
|
|
|
|
sub new {
|
|
my $class = shift;
|
|
my $config = shift;
|
|
my $logger = shift;
|
|
my $debug = shift;
|
|
my $self = {_debug=>$debug, _config=>$config, _logger=>$logger};
|
|
bless $self, $class;
|
|
my @publicEvents = qw(addInstance deleteInstance editWorkflowPriority getStatus getJsonStatus);
|
|
POE::Session->create(
|
|
object_states => [ $self => [qw(_start _stop returnInstanceToRunnableState addInstance checkInstances deleteInstance suspendInstance runWorker workerResponse), @publicEvents] ],
|
|
args=>[\@publicEvents]
|
|
);
|
|
my $cookies = HTTP::Cookies->new(file => '/tmp/cookies');
|
|
POE::Component::Client::HTTP->spawn(
|
|
Agent => 'Spectre',
|
|
Alias => 'workflow-ua',
|
|
CookieJar => $cookies
|
|
);
|
|
$self->{_runningQueue} = POE::Queue::Array->new;
|
|
$self->{_waitingQueue} = POE::Queue::Array->new;
|
|
$self->{_suspendedQueue} = POE::Queue::Array->new;
|
|
}
|
|
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 removeInstanceFromRunningQueue ( )
|
|
|
|
Removes a workflow instance from the queue that tracks what's running and returns a reference to it.
|
|
|
|
=cut
|
|
|
|
sub removeInstanceFromRunningQueue {
|
|
my $self = shift;
|
|
my $instanceId = shift;
|
|
my @items = $self->getRunningQueue->remove_items(
|
|
sub {
|
|
my $payload = shift;
|
|
return 1 if ($payload->{instanceId} eq $instanceId);
|
|
return 0;
|
|
}
|
|
);
|
|
my $instance = $items[0][2];
|
|
return $instance;
|
|
}
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 returnInstanceToRunnableState ( )
|
|
|
|
Returns a workflow instance back to runnable queue.
|
|
|
|
=cut
|
|
|
|
sub returnInstanceToRunnableState {
|
|
my ($self, $instance) = @_[OBJECT, ARG0];
|
|
$self->debug("Returning ".$instance->{instanceId}." to runnable state.");
|
|
$self->getSuspendedQueue->remove_items(
|
|
sub {
|
|
my $payload = shift;
|
|
return 1 if ($payload->{instanceId} eq $instance->{instanceId});
|
|
return 0;
|
|
}
|
|
);
|
|
$self->getWaitingQueue->enqueue($instance->{workingPriority}+1, $instance);
|
|
}
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 runWorker ( )
|
|
|
|
Calls a worker to execute a workflow activity.
|
|
|
|
=cut
|
|
|
|
sub runWorker {
|
|
my ($kernel, $self, $instance, $session) = @_[KERNEL, OBJECT, ARG0, SESSION];
|
|
$self->debug("Preparing to run workflow instance ".$instance->{instanceId}.".");
|
|
my $url = "http://".$instance->{sitename}.':'.$self->config->get("webguiPort").$instance->{gateway};
|
|
my $request = POST $url, [op=>"runWorkflow", instanceId=>$instance->{instanceId}];
|
|
my $cookie = $self->{_cookies}{$instance->{sitename}};
|
|
$request->header("Cookie",$instance->{cookieName}."=".$cookie) if (defined $cookie);
|
|
$request->header("X-instanceId",$instance->{instanceId});
|
|
$request->header("User-Agent","Spectre");
|
|
$self->debug("Posting workflow instance ".$instance->{instanceId}." to $url.");
|
|
$kernel->post('workflow-ua','request', 'workerResponse', $request);
|
|
$self->debug("Workflow instance ".$instance->{instanceId}." posted.");
|
|
}
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 suspendInstance ( )
|
|
|
|
Suspends a workflow instance for a number of seconds defined in the config file, and then returns it to the runnable queue.
|
|
|
|
=cut
|
|
|
|
sub suspendInstance {
|
|
my ($self, $instance, $waitTimeout, $kernel) = @_[OBJECT, ARG0, ARG1, KERNEL];
|
|
$waitTimeout ||= $self->config->get("suspensionDelay");
|
|
$self->debug("Suspending workflow instance ".$instance->{instanceId}." for ".$waitTimeout." seconds.");
|
|
my $priority = ($instance->{priority} - 1) * 10;
|
|
$self->getSuspendedQueue->enqueue($priority, $instance);
|
|
$kernel->delay_set("returnInstanceToRunnableState", $waitTimeout, $instance);
|
|
}
|
|
|
|
#-------------------------------------------------------------------
|
|
|
|
=head2 workerResponse ( )
|
|
|
|
This method is called when the response from the runWorker() method is received.
|
|
|
|
=cut
|
|
|
|
sub workerResponse {
|
|
my ($self, $kernel, $requestPacket, $responsePacket) = @_[OBJECT, KERNEL, ARG0, ARG1];
|
|
$self->debug("Retrieving response from workflow instance.");
|
|
my $request = $requestPacket->[0];
|
|
my $response = $responsePacket->[0];
|
|
my $instanceId = $request->header("X-instanceId"); # got to figure out how to get this from the request, cuz the response may die
|
|
$self->debug("Response retrieved is for $instanceId.");
|
|
my $instance = $self->removeInstanceFromRunningQueue($instanceId);
|
|
if ($response->is_success) {
|
|
$self->debug("Response for $instanceId retrieved successfully.");
|
|
if ($response->header("Set-Cookie") ne "") {
|
|
$self->debug("Storing cookie for $instanceId for later use.");
|
|
my $cookie = $response->header("Set-Cookie");
|
|
my $pattern = $instance->{cookieName}."=([a-zA-Z0-9\_\-]{22}).*";
|
|
$cookie =~ s/$pattern/$1/;
|
|
$self->{_cookies}{$instance->{sitename}} = $cookie;
|
|
}
|
|
my $state = $response->content;
|
|
$instance->{lastState} = $state;
|
|
$instance->{lastRunTime} = localtime(time());
|
|
if ($state =~ m/^waiting\s*(\d+)?$/) {
|
|
my $waitTime = $1;
|
|
$self->debug("Was told to wait on $instanceId because we're still waiting on some external event.");
|
|
$kernel->yield("suspendInstance",$instance, $waitTime);
|
|
}
|
|
elsif ($state eq "complete") {
|
|
$self->debug("Workflow instance $instanceId ran one of it's activities successfully.");
|
|
$kernel->yield("returnInstanceToRunnableState",$instance);
|
|
}
|
|
elsif ($state eq "disabled") {
|
|
$self->debug("Workflow instance $instanceId is disabled.");
|
|
$kernel->yield("suspendInstance",$instance);
|
|
}
|
|
elsif ($state eq "done") {
|
|
$self->debug("Workflow instance $instanceId is now complete.");
|
|
$kernel->yield("deleteInstance",$instanceId);
|
|
}
|
|
elsif ($state eq "error") {
|
|
$self->debug("Got an error response for $instanceId.");
|
|
$kernel->yield("suspendInstance",$instance);
|
|
}
|
|
else {
|
|
$self->error("Something bad happened on the return of $instance->{sitename} - $instanceId. ".$response->error_as_HTML);
|
|
$kernel->yield("suspendInstance",$instance);
|
|
}
|
|
}
|
|
elsif ($response->is_redirect) {
|
|
$self->error("Response for $instance->{sitename} - $instanceId was redirected. This should never happen if configured properly!!!");
|
|
$instance->{lastState} = "redirect";
|
|
$instance->{lastRunTime} = localtime(time());
|
|
}
|
|
elsif ($response->is_error) {
|
|
$instance->{lastState} = "comm error";
|
|
$instance->{lastRunTime} = localtime(time());
|
|
$self->error("Response for $instance->{sitename} - $instanceId had a communications error. ".$response->error_as_HTML);
|
|
$kernel->yield("suspendInstance",$instance)
|
|
}
|
|
}
|
|
|
|
|
|
1;
|