miller/c/input/lrec_reader_gen.c

#include <stdio.h>
#include <stdlib.h>
#include "lib/mlr_globals.h"
#include "lib/mlrutil.h"
#include "input/lrec_readers.h"

typedef struct _lrec_reader_gen_state_t {
	char* field_name;
	unsigned long long start;
	unsigned long long stop;
	unsigned long long step;
	unsigned long long current_value;
} lrec_reader_gen_state_t;

static void    lrec_reader_gen_free(lrec_reader_t* preader);
static void*   lrec_reader_gen_open(void* pvstate, char* prepipe, char* filename);
static void    lrec_reader_gen_close(void* pvstate, void* pvhandle, char* prepipe);
static void    lrec_reader_gen_sof(void* pvstate, void* pvhandle);
static lrec_t* lrec_reader_gen_process(void* pvstate, void* pvhandle, context_t* pctx);

// ----------------------------------------------------------------
lrec_reader_t* lrec_reader_gen_alloc(char* field_name,
	unsigned long long start, unsigned long long stop, unsigned long long step)
{
	lrec_reader_t* plrec_reader = mlr_malloc_or_die(sizeof(lrec_reader_t));

	lrec_reader_gen_state_t* pstate = mlr_malloc_or_die(sizeof(lrec_reader_gen_state_t));
	pstate->field_name    = field_name;
	pstate->start         = start;
	pstate->stop          = stop;
	pstate->step          = step;
	pstate->current_value = start;

	plrec_reader->pvstate       = (void*)pstate;
	plrec_reader->popen_func    = lrec_reader_gen_open;
	plrec_reader->pclose_func   = lrec_reader_gen_close;
	plrec_reader->pprocess_func = lrec_reader_gen_process;
	plrec_reader->psof_func     = lrec_reader_gen_sof;
	plrec_reader->pfree_func    = lrec_reader_gen_free;

	return plrec_reader;
}

static void* lrec_reader_gen_open(void* pvstate, char* prepipe, char* filename) {
	return NULL;
}

static void lrec_reader_gen_close(void* pvstate, void* pvhandle, char* prepipe) {
}

static void lrec_reader_gen_free(lrec_reader_t* preader) {
	free(preader->pvstate);
	free(preader);
}

static void lrec_reader_gen_sof(void* pvstate, void* pvhandle) {
	lrec_reader_gen_state_t* pstate = pvstate;
	pstate->current_value = pstate->start;
}

// ----------------------------------------------------------------
static lrec_t* lrec_reader_gen_process(void* pvstate, void* pvhandle, context_t* pctx) {
	lrec_reader_gen_state_t* pstate = pvstate;
	if (pstate->current_value > pstate->stop) {
		return NULL;
	}

	lrec_t* prec = lrec_unbacked_alloc();
	char* key = pstate->field_name;
	char* value = mlr_alloc_string_from_ll(pstate->current_value);
	pstate->current_value += pstate->step;

	lrec_put(prec, key, value, FREE_ENTRY_VALUE);

	return prec;
}