Shell Programming and Scripting

#/usr/bin/perl
#################################################################################################
#												#
# mqftp 	: This program can start/stop all the qmgr that are running on the give host
#			or can select particular qmgr to start/stop
####################################################

select STDOUT; #$| = 1; 
$MQ_LOG = 'console.log';
sub usage {
    print "Usage, $0 qmgrname|all stop|start\n";
    print "OR    $0 status \n";
    exit 1;
}

sub trim($)
{
	my $string = shift;
	$string =~ s/^\s+//;
	$string =~ s/\s+$//;
	return $string;
}


sub stopqmgr{
	my $qmgr = shift;
	#print("stoping  qmgr: $qmgr called \n");
	@status = `endmqm -i $qmgr`;
#	print("printing status \n");	
	#print @status;
}

sub startqmgr{
	my $qmgr = shift;
	#print("start qmgr $qmgr called\n");
	@status = `strmqm  $qmgr`;
	print @status;	
}


sub validate_user{
	$ami = `whoami`;
	$groups = `groups $ami`;
	if ($groups =~ m/mqm/){
	#	print " Has access....$groups  \n";
	}else{
		print " You are not alowed to perform any of these operation ....\n $groups \n ";
		exit(1);
	}
}

sub do_ctl
{
	if ($#_ < 1) {
        usage();
	}
	
	validate_user(); 

	my $qmgr_list = get_qmgrs();

       # print all the qmgr and their status 

	 #foreach  $qmgr ( keys(%$qmgr_list)){
         #      	print("$qmgr \t : \t $qmgr_list->{$qmgr}\n");
         #}

        if($ARGV[0] eq 'all' && $ARGV[2] eq '')
        {
		if($ARGV[1] eq 'stop')
                {
			foreach  $qmgr ( keys(%$qmgr_list)){
                       		#print("$qmgr \t : \t $qmgr_list->{$qmgr}\n");
				
				if($qmgr_list->{$qmgr} eq 'Running')
				{
					stopqmgr($qmgr);
				}
       			}
                }
                elsif($ARGV[1] eq 'start')
                {
			foreach  $qmgr ( keys(%$qmgr_list)){
                                #print("$qmgr \t : \t $qmgr_list->{$qmgr}\n");
                                if($qmgr_list->{$qmgr} eq 'NotRunning')
                                {
                                        startqmgr($qmgr);
                                }
                        }

                }
		else{
			print " Invalid Argument: $ARGV[1] \n ";
			usage();
		}
		
        }
	else{
		$qmgr_found = 1;
		$search_str = $ARGV[2];
		#print" selected qmgr $ARGV[0] ... $qmgr_list->{$ARGV[0]}  \n";
		
		if ($ARGV[2] eq '')
		{
			$search_str = $ARGV[0];
		}
		foreach  $qmgr ( keys(%$qmgr_list)){
			$qmgr_found = 1;
 			#print("$qmgr \t : \t $qmgr_list->{$qmgr}\n");
			if( $qmgr =~ m/$search_str/)
			{
				
				#print("match $qmgr $qmgr_found  \n");
				if ($qmgr_list->{$qmgr} eq 'Running')
              			  {
                        		if($ARGV[1] eq 'stop')
                        		{
                                		#print("stop qmgr called");
                                 		stopqmgr($qmgr);
                        		}else{
                                		print " $qmgr is already running \n";
                        		}
                		}
                		else
	                	{
        	               		# print "qmgr status is :  $qmgr_list->{$qmgr}   \n";
	        	                if($ARGV[1] eq 'start')
        	        	        {
                	        	        #print("start qmgr called");
                        	        	 startqmgr($qmgr);
		                        }else{
        		                        print " $qmgr is not  running \n";
                		        }
                		}

			}
			else
			{
				$qmgr_found =0;	
				#print("no matching found $qmgr  $qmgr_found \n");
			}
  		}
     }
}


sub get_qmgrs
{
	my %htab;
	my @qmgrs = map { (split)[0..1] } grep { /QMNAME/  } `dspmq`;
	foreach $i (@qmgrs)
	{
		#print " in get_qmgrs : $i \n";
		my @tmp;
		$i =~ m/^QMNAME\((\w+)\)/;
		$tmp[0] = $1;
		$i =~ m/^STATUS\((\w+)\)/;
		$tmp[1] = $1;
		if ($tmp[1] =~  m/Running/)
		{
			$htab{$tmp[0]} = $tmp[1];
		}
		else
		{
			$htab{$tmp[0]} = 'NotRunning';
		}
		
	}
	foreach   $key (keys(%htab))
	 {
		 #print "$key \t : $htab{$key} \n";
	 }
	return \%htab;
}

sub do_status
{
	open(F,"dspmq |");
	while(defined(my $line = <F>)) {
		my @tmp = split /\(+/,$line;
		@tmp1 = split /\)/,$tmp[1];
	       
                @tmp2 = split /\)/,$tmp[2];
		$qmgr = $tmp1[0];
		$status = $tmp2[0];

		if ($status eq 'Running')
	        {
			my @lsr = map { (split)[7..15] } grep { /runmqlsr/ && /$qmgr/  } `/bin/ps -ef`;
			$i=0;	
			foreach $line (@lsr)
			{	
				if ($line eq '-p') 
				{
					$port= $lsr[$i+1];
				}
				$i=$i+1;
			}
	        }
		else
		{
			$port = '';
		}
		print ("$qmgr \t $status \t $port \n");
		push @qmgr_status, "$qmgr \t $status \t $port";
		$port = '';
    	}  
}

    #if ($ARGV[0] eq "status") {
	select STDOUT; $| = 1;

        $curdir = trim(`pwd`);
        my @tmp = split /\//,$curdir;

	$MQ_LOG = "/$tmp[1]/mqtrace/adminlog/console.log"; 
	#print "$MQ_LOG \n";
	open (MQLOG, '>>'.$MQ_LOG);
	$user = trim(`whoami`);
	print(MQLOG  localtime(time)." ($user)  command : $ARGV[0] $ARGV[1] $ARGV[2] $ARGV[3] \n");

	close MQLOG;

if ($ARGV[0] eq "status") {
	do_status(@ARGV);
     }
    else {
        do_ctl(@ARGV);
     }

select STDOUT; #$| = 1; 
$MQ_LOG = 'console.log';