validate.c

Go to the documentation of this file.

/***************************************************************
 Copyright (C) 2010-2014 Hewlett-Packard Development Company, L.P.

 This program is free software; you can redistribute it and/or
 modify it under the terms of the GNU General Public License
 version 2 as published by the Free Software Foundation.

 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License along
 with this program; if not, write to the Free Software Foundation, Inc.,
 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

 ***************************************************************/
#include "buckets.h"
extern int debug;

FUNCTION int arrayAinB(int *arrayA, int *arrayB)
{
  int *arrayBHead;

  if (!arrayA || !arrayB) return 0;

  arrayBHead = arrayB;
  while(*arrayA)
  {
    arrayB = arrayBHead;
    while (*arrayB)
    {
      if (*arrayA == *arrayB) break;
      arrayB++;
    }
    if (!*arrayB) return 0;
    arrayA++;
  }
  return 1;
}

FUNCTION int intAinB(int intA, int *arrayB)
{

  if (!arrayB) return 0;

  while(*arrayB)
  {
    if (intA == *arrayB) return 1;
    arrayB++;
  }
  return 0;
}


FUNCTION int validate_pk(PGconn *pgConn, char *sql)
{
  char *fcnName = "validate_pk";
  int pk = 0;
  PGresult *result;

  /* Skip file if it has already been processed for buckets. */
  result = PQexec(pgConn, sql);
  if (fo_checkPQresult(pgConn, result, sql, fcnName, __LINE__)) return 0;
  if (PQntuples(result) > 0) pk = atoi(PQgetvalue(result, 0, 0));
  PQclear(result);
  return pk;
}

FUNCTION void Usage(char *Name)
{
  printf("Usage: %s [debug options]\n", Name);
  printf("  Debug options are:\n");
  printf("  -i   :: Initialize the database, then exit.\n");
  printf("  -n   :: bucketpool name of bucketpool to use.\n");
  printf("  -p   :: bucketpool_pk of bucketpool to use.\n");
  printf("  -r   :: rerun buckets.\n");
  printf("  -t   :: uploadtree_pk, root of tree to scan.\n");
  printf("  -u   :: upload_pk to scan.\n");
  printf("  -v   :: verbose (turns on copious debugging output)\n");
  printf("  -V   :: print the version info, then exit.\n");
  printf("  -c SYSCONFDIR :: Specify the directory for the system configuration. \n");
  printf("  NOTE: -n and -p are mutually exclusive.  If both are specified\n");
  printf("         -p is used.  One of these is required.\n");
  printf("  NOTE: -t and -u are mutually exclusive.  If both are specified\n");
  printf("         -u is used.  One of these is required.\n");
  printf("  NOTE: If none of -nptu are specified, the bucketpool_pk and upload_pk are read from stdin, one comma delimited pair per line.  For example, 'bppk=123, upk=987' where 123 is the bucketpool_pk and 987 is the upload_pk.  This is the normal execution from the scheduler.\n");
} /* Usage() */


FUNCTION int processed(PGconn *pgConn, int agent_pk, int nomos_agent_pk, int pfile_pk, int uploadtree_pk,
                       int bucketpool_pk, int bucket_pk)
{
  char *fcnName = "processed";
  int numRecs=0;
  char sqlbuf[512];
  PGresult *result;

  /* Skip file if it has already been processed for buckets.
     See if this pfile or uploadtree_pk has any buckets. */
  if (bucket_pk)
  {
    sprintf(sqlbuf,
    "select bf_pk from bucket_file, bucket_def \
      where pfile_fk=%d and agent_fk=%d and nomosagent_fk=%d and bucketpool_fk=%d \
            and bucket_pk=%d and bucket_fk=bucket_pk \
     union \
     select bf_pk from bucket_container, bucket_def \
      where uploadtree_fk=%d and agent_fk=%d and nomosagent_fk=%d and bucketpool_fk=%d \
            and bucket_pk=%d and bucket_fk=bucket_pk limit 1",
    pfile_pk, agent_pk, nomos_agent_pk, bucketpool_pk, bucket_pk,
    uploadtree_pk, agent_pk, nomos_agent_pk, bucketpool_pk, bucket_pk);
  }
  else
  {
    sprintf(sqlbuf,
    "select bf_pk from bucket_file, bucket_def \
      where pfile_fk=%d and agent_fk=%d and nomosagent_fk=%d and bucketpool_fk=%d \
            and bucket_fk=bucket_pk \
     union \
     select bf_pk from bucket_container, bucket_def \
      where uploadtree_fk=%d and agent_fk=%d and nomosagent_fk=%d and bucketpool_fk=%d \
            and bucket_fk=bucket_pk limit 1",
    pfile_pk, agent_pk, nomos_agent_pk, bucketpool_pk,
    uploadtree_pk, agent_pk, nomos_agent_pk, bucketpool_pk);
  }
  result = PQexec(pgConn, sqlbuf);
  if (fo_checkPQresult(pgConn, result, sqlbuf, __FILE__, __LINE__)) return -1;
  numRecs = PQntuples(result);
  PQclear(result);

  if (debug) printf("%s: returning %d, for pfile_pk %d, uploadtree_pk %d\n",fcnName,numRecs,pfile_pk, uploadtree_pk);
  return numRecs;
}


FUNCTION int UploadProcessed(PGconn *pgConn, int bucketagent_pk, int nomosagent_pk,
                             int pfile_pk, int uploadtree_pk,
                             int upload_pk, int bucketpool_pk)
{
  char *fcnName = "UploadProcessed";
  int numRecs=0;
  char sqlbuf[512];
  PGresult *result;

  /* Check bucket_ars to see if there has been a successful run */
  sprintf(sqlbuf,
    "select ars_pk from bucket_ars \
      where agent_fk=%d and nomosagent_fk=%d and upload_fk=%d and bucketpool_fk=%d \
            and ars_success=true limit 1",
     bucketagent_pk, nomosagent_pk,  upload_pk, bucketpool_pk);
  result = PQexec(pgConn, sqlbuf);
  if (fo_checkPQresult(pgConn, result, sqlbuf, fcnName, __LINE__)) return -1;
  numRecs = PQntuples(result);
  PQclear(result);

  return numRecs;
}