lev0/apps/build_lev1_mgr_jim.c

/*-----------------------------------------------------------------------------
 * cvs/JSOC/proj/lev0/apps/build_lev1_mgr.c
 *-----------------------------------------------------------------------------
 *
 * This is a stand alone module that processes lev0
 * filtergrams to lev1 by running the build_lev1 module. 
 * (See "Normal Calls:" below for normal running calls)
 *
 *build_lev1_mgr
 *      mode= recnum or fsn
 *      instru= hmi or aia
 *      dsin= default hmi.lev0e or aia.lev0e
 *      dsout= hmi.lev1e or aia.lev1e
 *      brec= first lev0 record# to process
 *      erec= last lev0 record# to process
 * or
 *      bfsn= first lev0 fsn# to process
 *      efsn= last lev0 fsn# to process
 *
 * Runs build_lev1 processes to create lev1 datasets
 * Has a mode to specify lev0 input by record# or fsn. Also if in
 * mode=recnum and brec=0 and erec=0 then runs in Stream Mode:
 * Stream Mode (one instance):
 *  This is the normal quick look mode that runs continuously and
 *  keeps up with the lev0 records.
 *      brec=0, erec=0 
 *      - start at the previous highest lev0 record processed
 *        This is keep in the DB table lev1_highest_lev0_recnum
 *      - fork from 8 to MAXCPULEV1 build_lev1 for every 
 *        12 (NUMRECLEV1) lev0 records. 
 *      - when an build_lev1 completes, fork another for next 12 rec
 *      - if no more lev0 records available, sleep and try again
 *      - if 8 CPU not enough to keep up with lev0, go to 16, etc.
 *
 * Recnum/fsn range Mode (any number of instances):
 *  This is lev1 processing by a lev0 range. Does consistency check for mode= and
 *  the use of brec/erec of bfsn/efsn, e.g.:
 *      brec=1000, erec=2000  or bfsn=4836500, efsn=4836590
 *      - qsub up to 16 (MAXQSUBLEV1) build_lev1 for 12 records ea
 *      - when a job completes qsub next 12 records until erec or efsn is reached
 *      - when all jobs are done, build_lev1_mgr will exit
 *
 */ 
/* Normal Calls:
 * For hmi call on cl1n002 as user jsocprod:
 *   build_lev1_mgr mode=recnum instru=hmi dsin=hmi.lev0a dsout=hmi.lev1_nrt 
 *      brec=0 erec=0
 * For aia call on cl1n003 as user jsocprod:
 *   build_lev1_mgr mode=recnum instru=aia dsin=aia.lev0 dsout=aia.lev1_nrt2 
 *      brec=0 erec=0
 * For iris call on cl1n001 as user jsocprod:
 *   build_lev1_mgr mode=recnum instru=iris dsin=iris_ground.lev0_dc1
 *      dsout=iris_ground.lev1_dc1 brec=0 erec=0
*/

#include <jsoc.h>
#include <cmdparams.h>
#include <drms.h>
#include <stdio.h>
#include <stdlib.h>
#include <strings.h>
#include <signal.h>
#include <sys/types.h>
#include <sys/time.h>
#include <sys/stat.h>   //for umask(2)
#include <unistd.h>     //for alarm(2) among other things...
#include <printk.h>
#include <errno.h>
#include <sys/wait.h>
#include "lev0lev1.h"   //defines NUMRECLEV1. Used by this and build_lev1.c

//default in and out data series
#define LEV0SERIESNAMEHMI "hmi.lev0e"
#define LEV0SERIESNAMEAIA "aia.lev0e"
#define LEV0SERIESNAMEIRIS "iris_ground.lev0_dc1"
#define LEV1SERIESNAMEHMI "su_production.hmi_lev1e"     //temp test case
#define LEV1SERIESNAMEAIA "su_production.aia_lev1e"     //temp test case
#define LEV1SERIESNAMEIRIS "iris_ground.lev1_dc1"
#define DSAIABAD "lm_jps.aia_bad_blobs"                 //dsaiabad= arg value to build_lev1 for aia. Eliminated by jim on 03Mar2011

#define LEV1LOG_BASEDIR "/usr/local/logs/lev1"
#define H1LOGFILE "/usr/local/logs/lev1/build_lev1_mgr_%s.%s.log" 
#define QSUBDIR "/surge40/jsocprod/qsub/tmp"
#define NUMTIMERS 8             //number of seperate timers avail
#define MAXCPULEV1 32           //max# of forks can do at a time for stream mode
#define DEFAULTCPULEV1 "8"      //default# of forks can do at a time 
#define MAXQSUBLEV1 64  //max# of qsub can do at a time for reprocessing mode
#define DEFAULTQSUBLEV1 "16"
#define MAXJIDSTR MAXQSUBLEV1*16
#define NOTSPECIFIED "***NOTSPECIFIED***"
#define LOGTEST 0
char args8sv[128];      //used when LOGTEST = 1


int qsubjob(long long rec1, long long rec2);

// List of default parameter values. 
ModuleArgs_t module_args[] = { 
  {ARG_STRING, "instru", "hmi" , "instrument. either hmi,aia or iris"},
  {ARG_STRING, "dsin", NOTSPECIFIED, "dataset of lev0 filtergrams"},
  {ARG_STRING, "dsout", NOTSPECIFIED, "dataset of lev1 output"},
  {ARG_STRING, "logfile", NOTSPECIFIED, "optional log file name. Will create one if not given"},
  {ARG_INTS, "brec", "-1", "first lev0 rec# to process"},
  {ARG_INTS, "erec", "-1", "last lev0 rec# to process"},
  {ARG_INTS, "bfsn", "-1", "first lev0 fsn to process"},
  {ARG_INTS, "efsn", "-1", "last lev0 fsn to process"},
  {ARG_INTS, "quicklook", "0", "1=quick look, 0 = definitive mode"},
  {ARG_INTS, "numrec", NUMRECLEV1S, "number of lev0 to lev1 records at a time"},
  {ARG_INTS, "numcpu", DEFAULTCPULEV1, "max# of forks to do at a time for stream mode"},
  {ARG_INTS, "numqsub", DEFAULTQSUBLEV1, "max# of qsub to do at a time for reprocessing mode"},
  {ARG_FLAG, "v", "0", "verbose flag"},
  {ARG_FLAG, "h", "0", "help flag"},
  {ARG_END}
};

ModuleArgs_t *gModArgs = module_args;
CmdParams_t cmdparams;
// Module name presented to DRMS. 
char *module_name = "build_lev1_mgr";

FILE *h1logfp;          // fp for h1 ouput log for this run 
FILE *qsubfp;           // fp for qsub script
static char datestr[32];
static char open_dsname[256];
static struct timeval first[NUMTIMERS], second[NUMTIMERS];
static struct stat stbuf;

pid_t mypid;
uint64_t jid;
int verbose;
long long brec, erec, bfsn, efsn; //begin and end lev0 rec#/fsn to do.Must be same data type
long long lastrecnum0_now = 0;  //last RT lev0 recnum seen
long long lastrecnum0_prev = 0; //prvious RT lev0 recnum seen
int numrec, numcpu, numqsub;
int qcnt = 0;
//stream_mode is also quick look mode, i.e. brec=erec=0
int stream_mode = 0;            //0=qsub build_lev1, 1=fork it locally
int quicklook = 0;              //can force this flg to be passed to build_lev1
//int hmiaiaflg = 0;            //0=hmi, 1=aia
int instruflg = 0;              //0=hmi, 1=aia, 2=iris
int modeflg = 0;                //0=fsn, 1=recnum
int loopcnt = 0;                //force last lev0 records to lev1
int abortnow = 0;
char stopfile[80];              // e.g. /usr/local/logs/lev1/build_mgr_stop_hmi
char hmiaianame[16];            // "hmi" or "aia" or "iris"
char logname[128];
char argdsin[128], argdsout[128], arglogfile[128], arginstru[80], argmode[80];
char argbx[80], argex[80], argnumrec[80], argnumcpu[80], argnumqsub[80];
char timetag[32];
char *username;                 // from getenv("USER") 
char *logfile;                  // optional log name passed in 
char *instru;                   // instument. hmi or aia 
char *mode;                     // mode. recnum or fsn
char *dsin;                     // lev0 input dataset
char *dsout;                    // lev1 output dataset


int nice_intro ()
{
  int usage = cmdparams_get_int (&cmdparams, "h", NULL);
  if (usage)
    {
    printf ("Runs build_lev1 processes to create lev1 datasets.\n\n");
    printf ("Usage: build_lev1_mgr [-vh]\n"
        "mode=<recnum|fsn> instru=<hmi|aia|iris> dsin=<lev0> dsout=<lev1>\n"
        "brec=<rec#>|bfsn=<fsn#> erec=<rec#>|efsn=<fsn#>\n"
        "numcpu=<#> numqsub=<#> logfile=<file>\n"
        "  -h: help - show this message then exit\n"
        "  -v: verbose\n"
        "mode= recnum: brec and erec have the record # range to process \n"
        "      fsn: bfsn and efsn have the fsn # range to process\n"
        "      For safety, the mode and arg name used must be consistent\n"
        "instru= instrument. must be 'hmi' or 'aia' or 'iris'\n"
        "dsin= data set name of lev0 input\n"
        "      default hmi=hmi.lev0e   aia=aia.lev0e  iris=iris_ground.lev0_dc1\n"
        "dsout= data set name of lev1 output\n"
        "      default hmi=su_production.hmi_lev1e   aia=su_production.aia_lev1e  iris=iris_ground.lev1_dc1\n"
        "brec= first lev0 rec# to process. 0=Stream Mode if erec also 0\n"
        "erec= last lev0 rec# to process. 0=Stream Mode if brec also 0\n"
        "bfsn= first fsn# to process. -1=error must be given if fsn mode\n"
        "efsn= last fsn# to process. -1=error must be given if fsn mode\n"
        "numcpu= max# of forks to do at a time for stream mode. Default %s\n"
        "numqsub= max# of qsub to do at a time for reprocessing mode. Default %s\n"
        "logfile= optional log file name. If not given uses:\n"
        "         /usr/local/logs/lev1/build_lev1_mgr.<time_stamp>.log\n",
         DEFAULTCPULEV1, DEFAULTQSUBLEV1);
     printf ("\n * Has two modes:\n"
          " * Stream Mode (one instance):\n"
          " *  This is the normal quick look mode that runs continuously and\n"
          " *  keeps up with the lev0 records.\n"
          " *   mode=recnum, brec=0, erec=0\n"
          " *   - start at the previous highest lev0 record processed\n"
          " *     This is keep in the DB table lev1_highest_lev0_recnum\n"
          " *   - fork up to 8 (MAXCPULEV1) build_lev1 for every \n"
          " *     12 (NUMRECLEV1) lev0 records. \n"
          " *   - when an build_lev1 completes, fork another for next 12 rec\n"
          " *   - if no more lev0 records available, sleep and try again\n"
          " *   - if 8 CPU not enough to keep up with lev0, go to 16, etc.\n"
          " *\n"
          " * Range Mode (any number of instances):\n"
          " *   brec=1000, erec=2000 or bfsn=44000, efsn=45000\n"
          " *   - qsub up to 16 (MAXQSUBLEV1) build_lev1 for 12 records ea\n"
          " *   - when a job completes qsub next 12 records until erec is reached\n"
          " *   - when all jobs are done, build_lev1_mgr will exit\n\n");
    return(1);
    }
  verbose = cmdparams_get_int (&cmdparams, "v", NULL);
  return (0);
}

/* Return pointer to "Wed Jun 30 21:49:08 1993\n" */
char *get_datetime()
{
  struct timeval tvalr;
  struct tm *t_ptr;
  static char datestr[32];

  gettimeofday(&tvalr, NULL);
  t_ptr = localtime((const time_t *)&tvalr.tv_sec);
  sprintf(datestr, "%s", asctime(t_ptr));
  return(datestr);
}

// Setup global datestr[] like: 2008.07.14_08:29:31
char *do_datestr() {
  time_t tval;
  struct tm *t_ptr;

  tval = time(NULL);
  t_ptr = localtime(&tval);
  sprintf(datestr, "%d.%02d.%02d_%02d:%02d:%02d", 
          (t_ptr->tm_year+1900), (t_ptr->tm_mon+1),
          t_ptr->tm_mday, t_ptr->tm_hour, t_ptr->tm_min, t_ptr->tm_sec);
  return(datestr);
}

// Returns a time tag like  yyyy.mm.dd.hhmmss 
char *gettimetag()
{
  struct timeval tvalr;
  struct tm *t_ptr;

  gettimeofday(&tvalr, NULL);
  t_ptr = localtime((const time_t *)&tvalr);
  sprintf(timetag, "%04d.%02d.%02d.%02d%02d%02d",
        (t_ptr->tm_year+1900), (t_ptr->tm_mon+1), t_ptr->tm_mday, t_ptr->tm_hour, t_ptr->tm_min, t_ptr->tm_sec);
  return(timetag);
}


void BeginTimer(int n)
{
  gettimeofday (&first[n], NULL);
}

float EndTimer(int n)
{
  gettimeofday (&second[n], NULL);
  if (first[n].tv_usec > second[n].tv_usec) {
    second[n].tv_usec += 1000000;
    second[n].tv_sec--;
  }
  return (float) (second[n].tv_sec-first[n].tv_sec) +
    (float) (second[n].tv_usec-first[n].tv_usec)/1000000.0;
}

// Outputs the variable format message (re: printf) to the log file.
int h1log(const char *fmt, ...)
{
  va_list args;
  char string[32768];

  va_start(args, fmt);
  vsprintf(string, fmt, args);
  if(h1logfp) {
    fprintf(h1logfp, string);
    fflush(h1logfp);
  } 
  else {                        // couldn't open log 
    printf(string);             // also print to stdout
    fflush(stdout);
  }
  va_end(args);
  return(0);
}

int send_mail(char *fmt, ...)
{
  va_list args;
  char string[1024], cmd[1024];

  va_start(args, fmt);
  vsprintf(string, fmt, args);
  sprintf(cmd, "echo \"%s\" | Mail -s \"%s mail\" lev0_user", string, module_name);
  system(cmd);
  va_end(args);
  return(0);
}

// Got a fatal error. 
void abortit(int stat)
{
  char pcmd[128];

  printk("***Abort in progress ...\n");
  printk("**Exit build_lev1_mgr w/ status = %d\n", stat);
  sprintf(pcmd, "/bin/rm %s/build_lev1_mgr_%s.stream.touch 2>/dev/null",
              LEV1LOG_BASEDIR, hmiaianame);
  system(pcmd);
  if (h1logfp) fclose(h1logfp);
  exit(stat);
}

/* This is stream mode processing that will keep on processing lev0
 * records as they show up in the DB.
 * Process the lev0 to lev1 from recn0 to maxrecn0. 
 * Returns when all children processes are done. 
 * Note: The processing is done in sets of 17 (NUMRECLEV1) lev0 records, 
 * so the maxrecn0 may not be reached, but it will
 * get done with the next set when more lev0 records come in. forkstream()
 * is run again and will automatically process new lev0 records in
 * sets of 17 as they are seen in the DB.
 * Returns non-0 on error.
 * If force is set, then do any non-chunk size of lev0 records left.
*/
int forkstream(long long recn0, long long maxrecn0, int force)
{
  pid_t pid, wpid, fpid[MAXCPULEV1];
  long long numofrecs, frec, lrec;
  int stat_loc, i, j, k, l, numtofork;
  char *args[11], pcmd[128];
  char args1[128], args2[128], args3[128], args4[128], args5[128], args6[128];
  char args7[128], args8[128], args9[128];

  numofrecs = (maxrecn0 - recn0) + 1;
  numtofork = numofrecs/numrec;     //this many to fork 
  j = numtofork;
  if(j == 0) {
   if(force) { j=numtofork=1; }
   else return(0);
  }
  lrec = recn0-1;
  if(j < numcpu) l = j;         //fork less then the max at a time
  else l = numcpu;                      //fork the max at a time
  if(LOGTEST) {
    sprintf(args8sv, "logfile=%s/build_lev1.%s.log", 
                LEV1LOG_BASEDIR, gettimetag());
  }
  for(k=0; k < l; k++) { //fork this many to start 
    if(force) { frec = lrec+1; lrec = (frec + numofrecs)-1; }
    else { frec = lrec+1; lrec = (frec + numrec)-1; }
    if((pid = fork()) < 0) {
      printk("***Can't fork(). errno=%d\n", errno);
      return(1);                        //!!TBD decide what to do
    }
    else if(pid == 0) {         //this is the beloved child
      switch(instruflg) {
      case 0:
        args[0] = "build_lev1_hmi";
        break;
      case 1:
        args[0] = "build_lev1_aia";
        break;
      case 2:
        args[0] = "build_lev1_iris";
        break;
      }
      sprintf(args1, "mode=%s", mode);
      args[1] = args1;
      sprintf(args2, "dsin=%s", dsin);
      args[2] = args2;
      sprintf(args3, "dsout=%s", dsout);
      args[3] = args3;
      if(modeflg) {             //recnum mode
        sprintf(args4, "brec=%lld", frec);
        args[4] = args4;
        sprintf(args5, "erec=%lld", lrec);
        args[5] = args5;
      }
      else {
        sprintf(args4, "bfsn=%lld", frec);
        args[4] = args4;
        sprintf(args5, "efsn=%lld", lrec);
        args[5] = args5;
      }
      sprintf(args6, "instru=%s", instru);
      args[6] = args6;
      sprintf(args7, "quicklook=%d", quicklook);
      args[7] = args7;
      //sprintf(args8, "logfile=%s/l1_%s_%d.log", QSUBDIR, gettimetag(), k);
      if(LOGTEST) {
        sprintf(args8, "%s", args8sv);
      }
      else {
        sprintf(args8, "logfile=%s/l1s_%lld_%lld_%s.log", 
                LEV1LOG_BASEDIR, frec, lrec, hmiaianame);
      }
      args[8] = args8;
/***********The dsaiabad is obsolete***************************************
      if(hmiaiaflg) {           //aia has an extra arg to build_lev1
        sprintf(args9, "dsaiabad=%s", DSAIABAD);
        args[9] = args9;
        args[10] = NULL;
        printk("execvp: %s %s %s %s %s %s %s %s %s %s\n",
              args[0],args[1],args[2],args[3],args[4],args[5],args[6],args[7],args[8],args[9]);
      }
      else {
***************************************************************************/
        args[9] = NULL;
        printk("execvp: %s %s %s %s %s %s %s %s %s\n",
              args[0],args[1],args[2],args[3],args[4],args[5],args[6],args[7],args[8]);
      //} 
      if(execvp(args[0], args) < 0) {
        switch(instruflg) {
        case 0:
          printk("***Can't execvp() build_lev1_hmi. errno=%d\n", errno);
          break;
        case 1:
          printk("***Can't execvp() build_lev1_aia. errno=%d\n", errno);
          break;
        case 2:
          printk("***Can't execvp() build_lev1_iris. errno=%d\n", errno);
          break;
        }
        exit(1);
      }
    }
    --numtofork;
    printf("forked pid = %d\n", pid);
    fpid[k] = pid;
  }
  wpid = -1;                  //wait for any child
  while(1) {
    //don't block and report any status not yet reported
    pid = waitpid(wpid, &stat_loc, WNOHANG+WUNTRACED);
    //pid = waitpid(wpid, &stat_loc, WNOHANG);
    //printf("!!TEMP waitpid returned %d stat_loc=%d\n", pid, stat_loc);
    if(pid == 0) { sleep(5); continue; }  //nothing ready
    if(pid == -1) {
      if(errno == ECHILD) printf("!!No More Children\n");errno=0;
      //!!TBD assumes we catch up at some point and we know where we're at
      //now and can update the DB table. Check that this is ok.
      //now update lev1_highest_lev0_recnum table with lrec
      sprintf(pcmd, "echo \"update lev1_highest_lev0_recnum set lev0recnum=%lld, date='%s' where lev0series='%s'\" | psql -h hmidb -U production jsoc", lrec, get_datetime(), dsin);
      printf("%s\n", pcmd); //!!TEMP echo
      system(pcmd); 
      if(stat(stopfile, &stbuf) == 0) {
        printf("Stop file %s seen.\nWait until all children are done and exit...\n", stopfile);
        abortnow = 1;                   //don't wait for any more lev 0records
      }
      if(numtofork <= 0) break;
    }
    else {
      for(k=0; k < numcpu; k++) {        //make sure a good one replies and
        if(fpid[k] == pid) { break; }   //don't have to worry about wraparound
      }
      if(k == numcpu) continue; //extraneous pid get on first wait
      printf("returned pid = %d stat_loc = %d\n", pid, stat_loc);
      if(numtofork == 0) continue;      //find out no more children
    }
    //fork one more unless the stopfile is seen
    if(stat(stopfile, &stbuf) == 0) {
      printf("Stop file %s seen.\nWait until all children are done and exit...\n", stopfile);
      numtofork = 0;                    //no new forks
      abortnow = 1;                     //don't wait for any more lev 0records
      continue;
    }
    frec = lrec+1; lrec = (frec + numrec)-1;
    if((pid = fork()) < 0) {
      printk("***Can't fork(). errno=%d\n", errno);
      return(1);                        //!!TBD decide what to do
    }
    else if(pid == 0) {                   //this is the beloved child
      switch(instruflg) {
      case 0:
        args[0] = "build_lev1_hmi";
        break;
      case 1:
        args[0] = "build_lev1_aia";
        break;
      case 2:
        args[0] = "build_lev1_iris";
        break;
      }
      sprintf(args1, "mode=%s", mode);
      args[1] = args1;
      sprintf(args2, "dsin=%s", dsin);
      args[2] = args2;
      sprintf(args3, "dsout=%s", dsout);
      args[3] = args3;
      if(modeflg) {             //recnum mode
        sprintf(args4, "brec=%lld", frec);
        args[4] = args4;
        sprintf(args5, "erec=%lld", lrec);
        args[5] = args5;
      }
      else {
        sprintf(args4, "bfsn=%lld", frec);
        args[4] = args4;
        sprintf(args5, "efsn=%lld", lrec);
        args[5] = args5;
      }
      sprintf(args6, "instru=%s", instru);
      args[6] = args6;
      sprintf(args7, "quicklook=%d", quicklook);
      args[7] = args7;
      if(LOGTEST) {
        sprintf(args8, "%s", args8sv);  //use original log name
      }
      else {
        sprintf(args8, "logfile=%s/l1s_%lld_%lld_%s.log", 
                        LEV1LOG_BASEDIR, frec, lrec, hmiaianame);
      }
      args[8] = args8;
/***********The dsaiabad is obsolete***************************************
      if(hmiaiaflg) {                   //aia has an extra arg to build_lev1
        sprintf(args9, "dsaiabad=%s", DSAIABAD);
        args[9] = args9;
        args[10] = NULL;
        printk("execvp: %s %s %s %s %s %s %s %s %s %s\n",
              args[0],args[1],args[2],args[3],args[4],args[5],args[6],args[7],args[8],args[9]);
      }
      else {
***************************************************************************/
        args[9] = NULL;
        printk("execvp: %s %s %s %s %s %s %s %s %s\n",
              args[0],args[1],args[2],args[3],args[4],args[5],args[6],args[7],args[8]);
      //}
      if(execvp(args[0], args) < 0) {
        switch(instruflg) {
        case 0:
          printk("***Can't execvp() build_lev1_hmi. errno=%d\n", errno);
          break;
        case 1:
          printk("***Can't execvp() build_lev1_aia. errno=%d\n", errno);
          break;
        case 2:
          printk("***Can't execvp() build_lev1_iris. errno=%d\n", errno);
          break;
        }
        exit(1);
      }
    }
    --numtofork;
    printf("forked pid = %d\n", pid);
    fpid[k] = pid;
  }
  return(0);
}

//Start a qsub job. Set the global jid.
//The args are either recnum of fsn according to modeflg.
int qsubjob(long long rec1, long long rec2)
{
  FILE *fin;
  char astr[32], bstr[32], string[128], qlogname[128], qsubcmd[512];;
  char recrange[128];

  if(modeflg) sprintf(recrange, ":#%lld-#%lld", rec1, rec2);
  else sprintf(recrange, "%lld-%lld", rec1, rec2);
  sprintf(open_dsname, "%s[%s]", dsin, recrange);
  printk("open_dsname = %s\n", open_dsname); //!!TEMP
  //sprintf(qlogname, "%s/qsub_prod_%d_%d.csh", QSUBDIR, mypid, qcnt++);
  sprintf(qlogname, "%s/p_%d_%d.csh", QSUBDIR, mypid, qcnt++);
  if((qsubfp=fopen(qlogname, "w")) == NULL) {
    fprintf(stderr, "**Can't open the qsub log file %s\n", qlogname);
    return(1);          //!!TBD
  }
  fprintf(qsubfp, "#!/bin/csh\n");
  fprintf(qsubfp, "limit vm 1000M\n");
  fprintf(qsubfp, "limit coredumpsize 0\n");
  fprintf(qsubfp, "echo \"TMPDIR = $TMPDIR\"\n");

  if(modeflg) {         //recnum mode
    switch(instruflg) {
    case 0:
      fprintf(qsubfp, "build_lev1_hmi mode=%s dsin=%s dsout=%s brec=%lld erec=%lld instru=%s quicklook=%d logfile=%s/l1q_b%lld_e%lld_$JOB_ID.log\n", 
        mode, dsin, dsout, rec1, rec2, instru, quicklook, QSUBDIR, rec1, rec2); 
      break;
    case 1:
      fprintf(qsubfp,"build_lev1_aia mode=%s dsin=%s dsout=%s brec=%lld erec=%lld instru=%s quicklook=%d logfile=%s/l1q_b%lld_e%lld_$JOB_ID.log\n", 
        mode, dsin, dsout, rec1, rec2, instru, quicklook, QSUBDIR, rec1, rec2); 
      break;
    case 2:
      fprintf(qsubfp,"build_lev1_iris mode=%s dsin=%s dsout=%s brec=%lld erec=%lld instru=%s quicklook=%d logfile=%s/l1q_b%lld_e%lld_$JOB_ID.log\n", 
        mode, dsin, dsout, rec1, rec2, instru, quicklook, QSUBDIR, rec1, rec2); 
      break;
    }
  }
  else {                //fsn mode
    switch(instruflg) {
    case 0:
      fprintf(qsubfp, "build_lev1_hmi mode=%s dsin=%s dsout=%s bfsn=%lld efsn=%lld instru=%s quicklook=%d logfile=%s/l1q_b%lld_e%lld_$JOB_ID.log\n", 
        mode, dsin, dsout, rec1, rec2, instru, quicklook, QSUBDIR, rec1, rec2); 
      break;
    case 1:
      fprintf(qsubfp,"build_lev1_aia mode=%s dsin=%s dsout=%s bfsn=%lld efsn=%lld instru=%s quicklook=%d logfile=%s/l1q_b%lld_e%lld_$JOB_ID.log\n", 
        mode, dsin, dsout, rec1, rec2, instru, quicklook, QSUBDIR, rec1, rec2); 
      break;
    case 2:
      fprintf(qsubfp,"build_lev1_iris mode=%s dsin=%s dsout=%s bfsn=%lld efsn=%lld instru=%s quicklook=%d logfile=%s/l1q_b%lld_e%lld_$JOB_ID.log\n", 
        mode, dsin, dsout, rec1, rec2, instru, quicklook, QSUBDIR, rec1, rec2); 
      break;
    }
  }
/********************Elim force of stream mode*******************************
  if(modeflg) {         //recnum mode
    if(hmiaiaflg) {     //aia has an extra arg to build_lev1
      fprintf(qsubfp, "build_lev1_aia --loopconn mode=%s dsin=%s dsout=%s brec=%lld erec=%lld instru=%s quicklook=1 logfile=%s/l1q_b%lld_e%lld_$JOB_ID.log\n", 
        mode, dsin, dsout, rec1, rec2, instru, QSUBDIR, rec1, rec2); 
    }
    else {
      fprintf(qsubfp, "build_lev1_hmi --loopconn mode=%s dsin=%s dsout=%s brec=%lld erec=%lld instru=%s quicklook=1 logfile=%s/l1q_b%lld_e%lld_$JOB_ID.log\n", 
        mode, dsin, dsout, rec1, rec2, instru, QSUBDIR, rec1, rec2); 
    }
  }
  else {                //fsn mode
    if(hmiaiaflg) {     //aia has an extra arg to build_lev1
      fprintf(qsubfp, "build_lev1_aia --loopconn mode=%s dsin=%s dsout=%s bfsn=%lld efsn=%lld instru=%s quicklook=1 logfile=%s/l1q_b%lld_e%lld_$JOB_ID.log\n", 
        mode, dsin, dsout, rec1, rec2, instru, QSUBDIR, rec1, rec2); 
    }
    else {
      fprintf(qsubfp, "build_lev1_hmi --loopconn mode=%s dsin=%s dsout=%s bfsn=%lld efsn=%lld instru=%s quicklook=1 logfile=%s/l1q_b%lld_e%lld_$JOB_ID.log\n", 
        mode, dsin, dsout, rec1, rec2, instru, QSUBDIR, rec1, rec2); 
    }
  }
****************************************************************************/
  fclose(qsubfp);
  sprintf(qsubcmd, "qsub -o %s -e %s -q p.q %s", 
                QSUBDIR, QSUBDIR, qlogname);            //note: uses p.q
  //sprintf(qsubcmd, "qsub -o %s -e %s -q j.q %s", 
  //            QSUBDIR, QSUBDIR, qlogname);
  //sprintf(qsubcmd, "qsub -q j.q %s", qlogname);
  printf("%s\n", qsubcmd);
  printk("%s\n", qsubcmd);
/********!!TEMP noop out*********************************************/
  sprintf(qsubcmd, "%s | grep \"Your job\"", qsubcmd);
  fin = popen(qsubcmd, "r");
  while(fgets(string, sizeof string, fin)) {  //get qsub return line
    sscanf(string, "%s %s %d", astr, bstr, &jid); // get job_id
  }
  pclose(fin);
  printf("\$JOB_ID = %u\n", jid);
/********!!TEMP end noop out*********************************************/
  return(0);
}
 
int qsubmode(long long frec, long long lrec)
{
  FILE *fin;
  char qsubcmd[512], string[128];
  char astr[32], bstr[32], jidstr[MAXJIDSTR];
  uint64_t qjid[MAXQSUBLEV1], qstatjid[MAXQSUBLEV1];
  long long numofrecs, rfirst, rlast;
  int numtoqsub, i, j, l, k, found, status;
  int jobdone=0;

  numofrecs = (lrec - frec) + 1;
  numtoqsub = numofrecs/numrec;     //this many to qsub
  if(numofrecs % numrec) numtoqsub++;
  j = numtoqsub;                        //0 implies one to qsub
  rlast = frec-1;
  if(j < numqsub) l = j;            //qsub less then the max at a time
  else l = numqsub;                  //qsub the max at a time
  for(k=0; k < l; k++) { //qsub this many to start 
    rfirst = rlast+1; rlast = (rfirst + numrec)-1;
    if(rlast > lrec) rlast = lrec;      //don't go past end
    status = qsubjob(rfirst, rlast);    //schedule the qsub job. set jid
    --numtoqsub;
    qjid[k] = jid;
    if(k == 0) sprintf(jidstr, "%u", jid);
    else sprintf(jidstr, "%s,%u", jidstr, jid);
  }
  //printf("jidstr = %s\n", jidstr);    //!!TEMP
  printf("numtoqsub left = %d\n", numtoqsub); //!!TEMP
  //sprintf(qsubcmd, "qstat -j %s 2>/dev/null | grep \"job_number:\"", jidstr);
  //sprintf(qsubcmd, "qstat -u production | grep \"qsub_prod_%d\"", mypid);
  //the grep string is from qlogname set in qsubjob()
  //sprintf(qsubcmd, "qstat -u production | grep \"p_%d_\"", mypid);
  sprintf(qsubcmd, "qstat -u %s | grep \"p_%d_\"", username, mypid);
  while(1) {
    //printf("\ncmd: %s\n", qsubcmd);   //!!TEMP
    if(!(fin = popen(qsubcmd, "r"))) {
      printf("!!!Fatal Error: can't do %s\n", qsubcmd);
      return(1);        //!!TBD check
    }
    //sleep(12);
    found = 0; k = 0;
    while(fgets(string, sizeof string, fin)) {  //get qstat return line
      //sscanf(string, "%s %u", astr, &jid);    // get job_id 
      sscanf(string, "%u", &jid);               // get job_id 
      printf("job id from qstat = %u\n", jid);
      qstatjid[k++] = jid;
      found = 1;
    }
    pclose(fin);

    //now see if any of the started jobs are done
    for(i=0; i < l; i++) {
      for(j=0; j < k; j++) {
        if(qjid[i] == qstatjid[j]) { //job still active
          break;
        }
      }
      if(j == k) {              //job done. start a new one
        if(qjid[i] != 0) {
          printf("Job done jid=%u\n", qjid[i]);
          jobdone++;
          qjid[i] = 0;
        }
        if(numtoqsub) {
          //start_new_qusb
          rfirst = rlast+1; rlast = (rfirst + numrec)-1;
          if(rlast > lrec) rlast = lrec;        //don't go past end
          status = qsubjob(rfirst, rlast);      //schedule the qsub job
          --numtoqsub;
          found = 1;                    //job to be found again
          qjid[i] = jid;
          //if(k == 0) sprintf(jidstr, "%u", jid);
          //else sprintf(jidstr, "%s,%u", jidstr, jid);
        }
        //else break;           //all done
      }
    }
    for(i=0; i < l; i++) {
      if(i == 0) sprintf(jidstr, "%u", qjid[i]);
      else sprintf(jidstr, "%s,%u", jidstr, qjid[i]);
    }
    printf("\n");
    if(!found) break;
    sleep(3);
  }
  printf("All jobs done = %d. numtoqsub = %d\n", jobdone, numtoqsub);
  return(0);
}

/* Create lev1 from lev0 records in either stream mode or 
 * range mode. Return non-0 on error.
 * In stream mode force any non-chunk size of lev0 records at the
 * end to lev1.
*/
int do_ingest(int force)
{
  FILE *fin;
  int rstatus;
  long long recnum0, maxrecnum0;
  char string[128], pcmd[128];

  if(stream_mode) {             //start past last lev0 rec# processed 
    sprintf(pcmd, "echo \"select lev0recnum from lev1_highest_lev0_recnum where lev0series='%s'\" | psql -h hmidb -U production jsoc", dsin);
    fin = popen(pcmd, "r");
    while(fgets(string, sizeof string, fin)) {  //get psql return line
      if(strstr(string, "lev0recnum")) continue;
      if(strstr(string, "-----")) continue;
      //get lev0 rec#
      if((rstatus = sscanf(string, "%lld", &recnum0)) == 0) {
        printf("Abort no lev0 entry in lev1_highest_lev0_recnum\n");
        printk("Abort no lev0 entry in lev1_highest_lev0_recnum\n");
        abortit(1); //no rec#
      }
      recnum0++;                                //start at next rec#
      break;
    }
    pclose(fin);
    sprintf(pcmd, "echo \"select max(recnum) from %s\" | psql -h hmidb -U production jsoc", dsin);
    fin = popen(pcmd, "r");
    while(fgets(string, sizeof string, fin)) {  //get psql return line
      if(strstr(string, "max")) continue;
      if(strstr(string, "-----")) continue;
      if(!strcmp(string, "    \n")) { //new series w/no recnum
        printf("Abort no max lev0 recnum (new series?)\n");
        printk("Abort no max lev0 recnum (new series?)\n");
        abortit(1);
      }
      sscanf(string, "%lld", &maxrecnum0);
      //maxrecnum0 = maxrecnum0 - 25;   //allow time for commit of lev0 
      //maxrecnum0 = maxrecnum0 - 40;   //allow (more) time for commit of lev0 
      if(instruflg == 2) maxrecnum0 = maxrecnum0 - 200; //allow (more) time for iris
      else maxrecnum0 = maxrecnum0 - 80;        //allow (more) time for commit of lev0 
      lastrecnum0_prev = lastrecnum0_now;
      lastrecnum0_now = maxrecnum0;             //save to see if more come in
      break;
    }
    pclose(fin);
    printf("Stream Mode starting at lev0 recnum = %lld maxrecnum = %lld\n", 
                recnum0, maxrecnum0);
    if(recnum0 > maxrecnum0) return(0); //nothing to do. go wait
    rstatus = forkstream(recnum0, maxrecnum0, force);
  }
  else {
    if(modeflg) {               //recnum mode
      //range mode. use brec/erec and qsub build_lev1 programs
      rstatus = qsubmode(brec, erec);
    }
    else {                      //fsn mode
      rstatus = qsubmode(bfsn, efsn);
    }
  }
  return(rstatus);
}

void sighandler(sig)
  int sig;
{
  char pcmd[128];
  if(sig == SIGTERM) {
    printf("*** %s build_lev1_mgr got SIGTERM. Exiting.\n", datestr);
    printk("*** %s build_lev1_mgr got SIGTERM. Exiting.\n", datestr);
    sprintf(pcmd, "/bin/rm %s/build_lev1_mgr_%s.stream.touch 2>/dev/null",
                LEV1LOG_BASEDIR, hmiaianame);
    system(pcmd);
    exit(1);
  }
  if(sig == SIGINT) {
    printf("*** %s build_lev1_mgr got SIGINT. Exiting.\n", datestr);
    printk("*** %s build_lev1_mgr got SIGINT. Exiting.\n", datestr);
    sprintf(pcmd, "/bin/rm %s/build_lev1_mgr_%s.stream.touch 2>/dev/null",
                LEV1LOG_BASEDIR, hmiaianame);
    system(pcmd);
    exit(1);
  }
  printk("*** %s build_lev1_mgr got an illegal signal %d, ignoring...\n",
                        datestr, sig);
  if (signal(SIGINT, SIG_IGN) != SIG_IGN)
      signal(SIGINT, sighandler);
  if (signal(SIGALRM, SIG_IGN) != SIG_IGN)
      signal(SIGALRM, sighandler);
}

// Initial setup stuff called when main is first entered.
void setup()
{
  FILE *fin;
  char string[128], cwdbuf[128], idstr[256], lfile[128];
  int tpid;

  do_datestr();
  printk_set(h1log, h1log);     // set for printk calls 
  printk("%s\n", datestr);
  getcwd(cwdbuf, 126);
  mypid = getpid();
  sprintf(idstr, "Cwd: %s\nCall: ", cwdbuf);
  sprintf(string, "%s started as pid=%d user=%s\n", module_name, mypid, username);
  strcat(idstr, string);
  printk("%s", idstr);
  printf("%s", idstr);
  switch(instruflg) {
  case 0:
    sprintf(stopfile, "/usr/local/logs/lev1/build_mgr_stop_hmi");
    break;
  case 1:
    sprintf(stopfile, "/usr/local/logs/lev1/build_mgr_stop_aia");
    break;
  case 2:
    sprintf(stopfile, "/usr/local/logs/lev1/build_mgr_stop_iris");
    break;
  }
  if(stream_mode) {
    sprintf(string, "/bin/rm -f %s", stopfile);   //remove any stop file
    system(string);
  }
  sprintf(argmode, "mode=%s", mode);
  sprintf(arginstru, "instru=%s", instru);
  sprintf(argdsin, "dsin=%s", dsin);
  sprintf(argdsout, "dsout=%s", dsout);
  if(modeflg) {
    sprintf(argbx, "brec=%lld", brec);
    sprintf(argex, "erec=%lld", erec);
  }
  else {
    sprintf(argbx, "bfsn=%lld", bfsn);
    sprintf(argex, "efsn=%lld", efsn);
  }
  sprintf(argnumrec, "numrec=%d", numrec);
  sprintf(argnumcpu, "numcpu=%d", numcpu);
  sprintf(argnumqsub, "numqsub=%d", numqsub);
  sprintf(arglogfile, "logfile=%s", logname);
  printk("%s %s %s %s %s %s %s %s %s\n", argmode, arginstru, argdsin, argdsout, argbx, argex, argnumrec, argnumcpu, argnumqsub);
  printf("%s %s %s %s %s %s %s %s %s\n", argmode, arginstru, argdsin, argdsout, argbx, argex, argnumrec, argnumcpu, argnumqsub);
  if (signal(SIGINT, SIG_IGN) != SIG_IGN)
      signal(SIGINT, sighandler);
  if (signal(SIGTERM, SIG_IGN) != SIG_IGN)
      signal(SIGTERM, sighandler);
  if (signal(SIGALRM, SIG_IGN) != SIG_IGN)
      signal(SIGALRM, sighandler);
  sprintf(idstr,  "ps -ef | grep %s", LEV1VIEWERNAME);
  fin = popen(idstr, "r");
  while(fgets(string, sizeof string, fin)) {  //get ps line
    if(!(strstr(string, "perl"))) continue;
    sscanf(string, "%s %d", idstr, &tpid); // get user name & process id
    sprintf(lfile, "%s/build_lev1_mgr_restart_%d.touch", LEV1LOG_BASEDIR, tpid);
    sprintf(idstr, "/bin/touch %s", lfile);
    printk("%s\n", idstr);
    system(idstr);
  }
  umask(002);                   // allow group write 
}

// Module main function. 
int main(int argc, char **argv)
{
  FILE *fin;
  int wflg = 1;
  char line[128], pcmd[128];

  if (cmdparams_parse(&cmdparams, argc, argv) == -1)
  {
     fprintf(stderr,"Error: Command line parsing failed. Aborting.\n");
     return 1;
  }
  if (nice_intro())
    return (0);
  if(!(username = (char *)getenv("USER"))) username = "nouser"; 
  instru = cmdparams_get_str(&cmdparams, "instru", NULL);
  if(strcmp(instru, "hmi") && strcmp(instru, "aia") && strcmp(instru, "iris")) {
    printf("instru= %s\n", instru);      //!!TEMP
    printf("Error: instru= must be given as 'hmi' or 'aia' or 'iris'\n");
    return(0);
  }
  if(!strcmp(instru, "aia")) {
    instruflg = 1;
    sprintf(hmiaianame, "aia");
  }
  else if(!strcmp(instru, "hmi")) {
    instruflg = 0;
    sprintf(hmiaianame, "hmi");
  }
  else {
    instruflg = 2;
    sprintf(hmiaianame, "iris");
  }
  mode = cmdparams_get_str(&cmdparams, "mode", NULL);
  if(strcmp(mode, "recnum") && strcmp(mode, "fsn")) {
    printf("Error: mode= must be given as 'recnum' or 'fsn'\n");
    return(0);
  }
  if(!strcmp(mode, "recnum")) modeflg = 1;
  dsin = cmdparams_get_str(&cmdparams, "dsin", NULL);
  dsout = cmdparams_get_str(&cmdparams, "dsout", NULL);
  brec = cmdparams_get_int(&cmdparams, "brec", NULL);
  erec = cmdparams_get_int(&cmdparams, "erec", NULL);
  bfsn = cmdparams_get_int(&cmdparams, "bfsn", NULL);
  efsn = cmdparams_get_int(&cmdparams, "efsn", NULL);
  quicklook = cmdparams_get_int(&cmdparams, "quicklook", NULL);
  numrec = cmdparams_get_int(&cmdparams, "numrec", NULL);
  numcpu = cmdparams_get_int(&cmdparams, "numcpu", NULL);
  numqsub = cmdparams_get_int(&cmdparams, "numqsub", NULL);
  if(numcpu > MAXCPULEV1) {
    printf("numcpu exceeds max of %d\n", MAXCPULEV1);
    return(0);
  }
  if(numqsub > MAXQSUBLEV1) {
    printf("numqsub exceeds max of %d\n", MAXQSUBLEV1);
    return(0);
  }
  if(modeflg) {                 //recnum mode
    if(brec == -1 || erec == -1) {
      fprintf(stderr, "brec and erec must be given. -1 not allowed\n");
      fprintf(stderr, "use brec=0 erec=0 to process in stream mode\n");
      return(0);
    }
    if(brec > erec) {
      fprintf(stderr, "brec must be <= erec\n");
      return(0);
    }
    if(brec == 0 && erec == 0) {
      //make sure there isn't a stream mode already running
      sprintf(pcmd, "ls %s/build_lev1_mgr_%s.stream.touch 2>/dev/null",
                LEV1LOG_BASEDIR, hmiaianame);
      fin = popen(pcmd, "r");
      while(fgets(line, sizeof line, fin)) {  //get ps return line
        printf("Error: build_lev1_mgr already running.");
        printf(" If not so, do:\n");
        printf("/bin/rm %s/build_lev1_mgr_%s.stream.touch\n",
                LEV1LOG_BASEDIR, hmiaianame);
        pclose(fin);
        return(0);
      }
      pclose(fin);
      sprintf(pcmd, "touch %s/build_lev1_mgr_%s.stream.touch 2>/dev/null",
                  LEV1LOG_BASEDIR, hmiaianame);
      system(pcmd);
      stream_mode = 1;          //aka quick look mode
      quicklook = 1;            //and force quicklook to 1
    }
  }
  else {                        //fsn mode
    if(bfsn == -1 || efsn == -1) {
      fprintf(stderr, "bfsn and efsn must be given. -1 not allowed\n");
      return(0);
    }
    if(bfsn == 0 || efsn == 0) {
      fprintf(stderr, "bfsn and efsn must be given. 0 not allowed\n");
      return(0);
    }
    if(bfsn > efsn) {
      fprintf(stderr, "bfsn must be <= efsn\n");
      return(0);
    }
  }
  logfile = cmdparams_get_str(&cmdparams, "logfile", NULL);
  if (strcmp(dsin, NOTSPECIFIED) == 0) {
    switch(instruflg) {
    case 0:
      dsin = LEV0SERIESNAMEHMI;
      break;
    case 1:
      dsin = LEV0SERIESNAMEAIA;
      break;
    case 2:
      dsin = LEV0SERIESNAMEIRIS;
      break;
    }
  }
  if (strcmp(dsout, NOTSPECIFIED) == 0) {
    switch(instruflg) {
    case 0:
      dsout = LEV1SERIESNAMEHMI;
      break;
    case 1:
      dsout = LEV1SERIESNAMEAIA;
      break;
    case 2:
      dsout = LEV1SERIESNAMEIRIS;
      break;
    }
  }
  if (strcmp(logfile, NOTSPECIFIED) == 0) {
    sprintf(logname, H1LOGFILE, hmiaianame, gettimetag());
  }
  else {
    sprintf(logname, "%s", logfile);
  }
  if((h1logfp=fopen(logname, "w")) == NULL)
    fprintf(stderr, "**Can't open the log file %s\n", logname);
  setup();
  while(wflg) {
    if(do_ingest(0)) {        // loop to get files from the lev0
      printk("**ERROR: in do_ingest() for %s\n", open_dsname);
    }
    if(!stream_mode) return(0); //all done for reprocessing
    if(abortnow) break;
    sleep(30);          //wait for more lev0 to appear
    if(lastrecnum0_now == lastrecnum0_prev) {  //no new lev0 in
      if(loopcnt++ == 15) {  //force any left over lev0 records to lev1
        printk("Timeout: force any left over lev0 to lev1\n");
        if(do_ingest(1)) {        // process the last lev0 records
          printk("**ERROR: in do_ingest() for %s\n", open_dsname);
        }
        loopcnt = 0;
      }
    }
    else loopcnt = 0;
    //wflg = 0;
  }
  sprintf(pcmd, "/bin/rm %s/build_lev1_mgr_%s.stream.touch 2>/dev/null",
              LEV1LOG_BASEDIR, hmiaianame);
  system(pcmd);
  return(0);
}

Revision:	1.1
Committed:	Tue Apr 22 17:59:12 2014 UTC (9 years, 5 months ago) by prodtest
Content type:	text/plain
Branch:	MAIN
CVS Tags:	Ver_8-8, Ver_8-11, Ver_8-10, Ver_8-6, Ver_LATEST, Ver_9-41, Ver_8-5, Ver_8-12, Ver_8-7, Ver_9-5, Ver_9-4, Ver_9-3, Ver_9-2, Ver_9-1, Ver_9-0, HEAD
Log Message:	* empty log message *