putsubs.c

Go to the documentation of this file.

#include "getln.h"
#include "error.h"
#include "logmsg.h"
#include "readwrite.h"
#include "str.h"
#include "fmt.h"
#include "stralloc.h"
#include "open.h"
#include "buffer.h"
#include "case.h"
#include "errtxt.h"
#include "subscribe.h"
#include "qmail.h"
#include "idx.h"
#include <unistd.h>
#include <libpq-fe.h>

 
extern PGconn *pgsql;
 
static buffer bi;
static char inbuf[512];
char strnum[FMT_ULONG];
static stralloc line = {0};
static stralloc fn = {0};
 
static void die_write(void) { logmsg(WHO,111,FATAL,B("ERR_WRITE :","stdout")); }

 
unsigned long putsubs(const char *dbname,unsigned long hash_lo,unsigned long hash_hi,int subwrite(),int flagsql)
{
  PGresult *result;
  int row_nr;
  int length;
  char *row;
  const char *table = (char *) 0;
 
  unsigned int i;
  int fd;
  unsigned long no = 0L;
  int match;
  unsigned int hashpos;
  const char *ret = (char *) 0;
 
  if (!flagsql || (ret = opensql(dbname,&table))) {
    if (flagsql && *ret) logmsg(WHO,111,FATAL,ret);        /* fallback to local db */
 
    if (!stralloc_copys(&fn,dbname)) die_nomem();
    if (!stralloc_catb(&fn,"/subscribers/?",15)) die_nomem(); /* NOTE: Also copies terminal '\0' */
    hashpos = fn.len - 2;
    if (hash_lo > 52) hash_lo = 52;
    if (hash_hi > 52) hash_hi = 52;
    if (hash_hi < hash_lo) hash_hi = hash_lo;
 
    for (i = hash_lo; i <= hash_hi; ++i) {
      fn.s[hashpos] = 64 + i;                              /* hash range 0-52 */
      fd = open_read(fn.s);
      if (fd == -1) {
        if (errno != ENOENT)
          logmsg(WHO,111,FATAL,B("ERR_READ :",fn.s));
      } else {
        buffer_init(&bi,buffer_unixread,fd,inbuf,sizeof(inbuf));
 
        for (;;) {
          if (getln(&bi,&line,&match,'\0') == -1)
            logmsg(WHO,111,FATAL,B("ERR_READ :",fn.s));
          if (!match)
            break;
          if (subwrite(line.s + 1,line.len - 2) == -1) die_write();
          no++;
        }
        close(fd);
      } 
    }
    return no;
 
  } else {                                                 /* SQL Version  main query */
    if (!stralloc_copys(&line,"SELECT address FROM ")) die_nomem();
    if (!stralloc_cats(&line,table)) die_nomem();
    if (!stralloc_cats(&line," WHERE hash BETWEEN ")) die_nomem();
    if (!stralloc_catb(&line,strnum,fmt_ulong(strnum,hash_lo))) die_nomem();
    if (!stralloc_cats(&line," AND ")) die_nomem();
    if (!stralloc_catb(&line,strnum,fmt_ulong(strnum,hash_hi))) die_nomem();
    if (!stralloc_0(&line)) die_nomem();
 
    result = PQexec(pgsql,line.s);
    if (result == NULL)
      logmsg(WHO,111,FATAL,PQerrorMessage(pgsql));
 
    if (PQresultStatus(result) != PGRES_TUPLES_OK)
      logmsg(WHO,111,FATAL,PQresultErrorMessage(result));
 
    /* Next statements are safe even if someone messes with the address field def */
 
    no = 0;
    for (row_nr = 0; row_nr < PQntuples(result); row_nr++) {
      length = PQgetlength(result,row_nr,0);
      row = PQgetvalue(result,row_nr,0);
      if (subwrite(row,length) == -1) die_write();
      no++;                                                /* count for list-list fxn */
    }
    PQclear(result);
    return no;
  }
}