command_generator_remote.cpp File Reference

#include "support/configCosmos.h"
#include "support/datalib.h"
#include "support/jsonlib.h"
#include "support/timelib.h"
#include "agent/event.h"
#include "agent/scheduler.h"

Include dependency graph for command_generator_remote.cpp:

Functions
int	main (int argc, char *argv[])

Function Documentation

int main	(	int	argc,
		char *	argv[]
	)

{

    uint32_t flag = 0;
    string data = "";
    string condition = "";
    string name= "";
    double utc = 0.;

    string node = "";

    switch (argc)
    {
    case 7: // set repeat flag
    {
        flag |= EVENT_FLAG_REPEAT;
    }
    case 6: // set conditions
    {
        condition = argv[5];
        flag |= EVENT_FLAG_CONDITIONAL;
    }
    case 5: // add command to the scheduler
    {
        node = string(argv[4]);
    }
    case 4: // set time utc in mjd
    {
        switch (argv[3][0])
        {
        // add a few seconds to current time
        case '+':
        {
            double seconds = atof(&argv[3][1]);
            utc = currentmjd() + seconds / 86400.;
            break;
        }
        default:
            // use set time
        {
            utc = atof(argv[3]);
            break;
        }
        }
    }
    case 3: // set command string
    {
        data = argv[2];
    }
    case 2: // set command name
    {
        name = argv[1];
        break;
    }
    default:
    {
        cout << "The command generator produces the command string to be fed into the command\n"
                "queue which is managed by agent_exec. Commands are a subset of events\n"
                "in COSMOS." << endl << endl;

        cout << "Usage" << endl << endl;
        cout << "  command_generator [options]" << endl;
        cout << "  command_generator name command [time | +sec] [node] [condition] [repeat_flag]" << endl << endl;


        cout << "Example" << endl << endl;
        cout << "  $ command_generator myCmd1 \"agent kauaicc_sim execsoh get_queue_size\" +10" << endl << endl;
        cout << "  This will run the command \"agent kauaicc_sim execsoh get_queue_size\" \n"
                "  with name 'myCmd' within 10 seconds from now." << endl << endl;

        cout << "Options" << endl << endl;

        cout << "  -name   \t = name of the command, can be a string or combination of \n "
                "          \t   alphanumeric characters (ex: myCmd1)" << endl;
        cout << "  -command\t = the actual command to be executed, if more than one word \n"
                "          \t   enclose the command string in quotes \n"
                "          \t   (ex: \"agent kauaicc_sim execsoh get_queue_size\"" << endl;
        cout << "  -time   \t = optional argument to enter the desired modified julian date\n"
                "          \t   (mjd). If not entered it will use the current time. \n"
                "          \t   If '+' is used instead then the number of seconds can be\n"
                "          \t   inserted. For 10 seconds in the future use +10" << endl;
        cout << "  -node   \t = optional argument to add the generated command to the command\n"
                "          \t   queue on the specified node (ex: kauaicc_sim) " << endl;
        return 0;
    }
    }

    Event event;
    cout << "Command string:" << endl;

    cout << event.generator(name, data, utc, condition, flag) << endl << endl;
    // Find agent_mongo
    Agent *agent = new Agent("", "temp");
    beatstruc agent_mongo_soh = agent->find_agent("any", "mongo");
    if (!agent_mongo_soh.exists) {
        cout << "could not find agent mongo" << endl;
    }

    if(node.empty()) {
        cout << "Could not add command to node [No Node]" << endl;
        return 0;
    }
    else {

        // create outgoing directory if it doesnt exist
        std::string outgoing_dir = data_base_path(node, "outgoing", "exec") + "/";

        cout << "Adding command/event to node directory:" << outgoing_dir << endl;

        // write command string to file in outgoing_dir
        std::string outfilename = data_name(node, currentmjd(),"","command");
        ofstream cmdfile;
        cmdfile.open(outgoing_dir + outfilename);
        cmdfile << event;
        cmdfile.close();

        // request agent_mongo to insert command
        if(agent_mongo_soh.exists) {
            string out;
            string mongo_request = "insert db " + node + ":command ";
            mongo_request += event.get_event_string();
            agent->send_request(agent_mongo_soh, mongo_request, out, 0);
            cout <<"Request agent mongo: " << mongo_request << endl;
            cout << out << endl;
        }
    }

}