core/sqf/sql/scripts/regensq.pl

#!/usr/bin/perl
#
# @@@ START COPYRIGHT @@@
#
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.
#
# @@@ END COPYRIGHT @@@
#

use strict;
use warnings;

use sqconfigdb;

my $bGen=1;
my $bVirtualNodes=0;
my $infile;
my $nodesAdded=0;
my $nodesRemoved=0;

sub doInit {
    $infile=$ARGV[0];
}

sub openFiles {
    open(SRC, "<$infile")
        or die("unable to open $infile");

    open(TMP, ">/tmp/zregen")
        or die("unable to open /tmp/zregen");

    sqconfigdb::openDb();
}

sub printScript {
    my (@rest) = @_;

    print TMP @rest;
}

sub printSQShellCommand {
    printScript($_);
}

sub processFloatingIp {
    printScript($_);
    while (<SRC>) {
        printScript($_);
        if (/^end floating_ip/) {
           return;
        }
    }
}

sub processNodes {
    printScript($_);
    while (<SRC>) {
        if (/^#/) { # comment
            printScript($_);
        } elsif (/^_virtualnodes/) {
            $bVirtualNodes=1;
            $bGen=0;
            printScript($_);
            print "Warning - virtual nodes cannot be regenerated - sqconfig NOT changed.\n";
        } elsif (/^end node/) {
            if (!$bVirtualNodes) {
                my $node_section = readpipe("trafconf -node");
                printScript($node_section);
            }
            printScript($_);
            return;
        }
    }
}

sub processPersist {
    printScript($_);
    while (<SRC>) {
        if (/^#/) { # comment
            printScript($_);
        } elsif (/^end persist/) {
            my $persist_section = readpipe("trafconf -persist");
            printScript($persist_section);
            printScript($_);
            return;
        }
    }
}

sub endGame {
    close(SRC);
    close(TMP);

    if ($bGen) {
        open(TMP, "</tmp/zregen")
            or die("unable to open /tmp/zregen");
        open(SRC, ">$infile")
            or die("unable to open $infile");
        while (<TMP>) {
            print SRC $_;
        }
        close(SRC);
        close(TMP);

        print "Re-Generated configuration file: $infile\n";
    }
}

#
# Main
#

doInit();

openFiles;


while (<SRC>) {
    if (/^begin node/) {
        processNodes();
    }
    elsif (/^begin floating_ip/) {
        processFloatingIp();
    }
    elsif (/^begin persist/) {
        processPersist();
    }
    else {
        printScript($_);
    }
}

endGame();