/*
 * dirtyPageTable.c
 *
 *  Created on: May 18, 2009
 *      Author: sears
 */

#include <stasis/redblack.h>
#include <stasis/common.h>
#include <stasis/flags.h>
#include <stasis/dirtyPageTable.h>
#include <stasis/page.h>

#include <stdio.h>

typedef struct {
  pageid_t p;
  lsn_t lsn;
} dpt_entry;

static int dpt_cmp(const void *ap, const void * bp, const void * ignored) {
  const dpt_entry * a = ap;
  const dpt_entry * b = bp;

  return (a->p < b->p) ? -1 : ((a->p == b->p) ? 0 : 1);
}

struct stasis_dirty_page_table_t {
  struct rbtree * table;
  stasis_buffer_manager_t * bufferManager;
  pageid_t count;
  pthread_mutex_t mutex;
  pthread_cond_t flushDone;
  int flushing;
};

void stasis_dirty_page_table_set_dirty(stasis_dirty_page_table_t * dirtyPages, Page * p) {
  assertlocked(p->rwlatch);
  if(!p->dirty) {
    p->dirty = 1;
    dpt_entry * e = malloc(sizeof(*e));
    e->p = p->id;
    e->lsn = p->LSN;
    pthread_mutex_lock(&dirtyPages->mutex);
    const void * ret = rbsearch(e, dirtyPages->table);
    assert(ret == e); // otherwise, the entry was already in the table.
    dirtyPages->count++;
    pthread_mutex_unlock(&dirtyPages->mutex);
#ifdef SANITY_CHECKS
  } else {
    pthread_mutex_lock(&dirtyPages->mutex);
    dpt_entry e = { p->id, 0};
    assert(rbfind(&e, dirtyPages->table));
    pthread_mutex_unlock(&dirtyPages->mutex);
#endif //SANITY_CHECKS
  }
}

void stasis_dirty_page_table_set_clean(stasis_dirty_page_table_t * dirtyPages, Page * p) {
  assertlocked(p->rwlatch);
  if(p->dirty) {
    pthread_mutex_lock(&dirtyPages->mutex);
    dpt_entry dummy = {p->id, 0};
    const dpt_entry * e = rbdelete(&dummy, dirtyPages->table);
    assert(e);
    assert(e->p == p->id);
    assert(p->dirty);
    p->dirty = 0;
    free((void*)e);
    dirtyPages->count--;
    pthread_mutex_unlock(&dirtyPages->mutex);
  }
}

int stasis_dirty_page_table_is_dirty(stasis_dirty_page_table_t * dirtyPages, Page * p) {
  int ret;
  assertlocked(p->rwlatch);

  ret = p->dirty;
#ifdef SANITY_CHECKS
  pthread_mutex_lock(&dirtyPages->mutex);
  dpt_entry e = { p->id, 0};
  const void* found = rbfind(&e, dirtyPages->table);
  assert((found && ret) || !(found||ret));
  pthread_mutex_unlock(&dirtyPages->mutex);
#endif
  return ret;
}

lsn_t stasis_dirty_page_table_minRecLSN(stasis_dirty_page_table_t * dirtyPages) {
  lsn_t lsn = LSN_T_MAX;
  pthread_mutex_lock(&dirtyPages->mutex);
  for(const dpt_entry * e = rbmin(dirtyPages->table);
          e;
          e = rblookup(RB_LUGREAT, e, dirtyPages->table)) {
    if(e->lsn < lsn) {
      lsn = e->lsn;
    }
  }
  pthread_mutex_unlock(&dirtyPages->mutex);
  return lsn;
}

pageid_t stasis_dirty_page_table_dirty_count(stasis_dirty_page_table_t * dirtyPages) {
  pthread_mutex_lock(&dirtyPages->mutex);
  pageid_t ret = dirtyPages->count;
  assert(dirtyPages->count >= 0);
  pthread_mutex_unlock(&dirtyPages->mutex);
  return ret;
}

int stasis_dirty_page_table_flush(stasis_dirty_page_table_t * dirtyPages) {
  dpt_entry dummy = { 0, 0 };
  const int stride = 200;
  pageid_t vals[stride];
  int off = 0;
  int strides = 0;
  pthread_mutex_lock(&dirtyPages->mutex);
  if(dirtyPages->flushing) {
    pthread_cond_wait(&dirtyPages->flushDone, &dirtyPages->mutex);
    pthread_mutex_unlock(&dirtyPages->mutex);
    return EAGAIN;
  }
  dirtyPages->flushing = 1;
  for(const dpt_entry * e = rblookup(RB_LUGTEQ, &dummy, dirtyPages->table) ;
        e;
        e = rblookup(RB_LUGREAT, &dummy, dirtyPages->table)) {
    dummy = *e;
    vals[off] = dummy.p;
    off++;
    if(off == stride) {
      pthread_mutex_unlock(&dirtyPages->mutex);
      for(pageid_t i = 0; i < off; i++) {
        dirtyPages->bufferManager->writeBackPage(dirtyPages->bufferManager, vals[i]);
      }
      off = 0;
      strides++;
      pthread_mutex_lock(&dirtyPages->mutex);
    }
  }
  pthread_mutex_unlock(&dirtyPages->mutex);
  for(int i = 0; i < off; i++) {
    dirtyPages->bufferManager->writeBackPage(dirtyPages->bufferManager, vals[i]);
  }
  pthread_mutex_lock(&dirtyPages->mutex);
  dirtyPages->flushing = 0;
  pthread_cond_broadcast(&dirtyPages->flushDone);
  pthread_mutex_unlock(&dirtyPages->mutex);

//  if(strides < 5) { DEBUG("strides: %d dirtyCount = %lld\n", strides, stasis_dirty_page_table_dirty_count(dirtyPages)); }

  return 0;
}

int stasis_dirty_page_table_get_flush_candidates(stasis_dirty_page_table_t * dirtyPages, pageid_t start, pageid_t stop, int count, pageid_t* range_starts, pageid_t* range_ends) {
  pthread_mutex_lock(&dirtyPages->mutex);
  int n = 0;
  int b = -1;
  dpt_entry dummy;
  dummy.lsn = -1;
  dummy.p = start;

  for(const dpt_entry *e = rblookup(RB_LUGTEQ, &dummy, dirtyPages->table);
      e && (stop == 0 || e->p < stop) && n < count;
      e = rblookup(RB_LUGREAT, e, dirtyPages->table)) {
    if(n == 0 || range_ends[b] != e->p) {
      b++;
      range_starts[b] = e->p;
      range_ends[b] = e->p+1;
    } else {
      range_ends[b]++;
    }
    n++;
  }
  pthread_mutex_unlock(&dirtyPages->mutex);
  return b+1;
}
void stasis_dirty_page_table_flush_range(stasis_dirty_page_table_t * dirtyPages, pageid_t start, pageid_t stop) {

  pthread_mutex_lock(&dirtyPages->mutex);
  int waitCount = 0;
  while(dirtyPages->flushing) {
    pthread_cond_wait(&dirtyPages->flushDone, &dirtyPages->mutex);
    waitCount++;
    if(waitCount == 2) {
      // a call to stasis_dirty_page_table_flush was initiated and completed since we were called.
      pthread_mutex_unlock(&dirtyPages->mutex);
      return;
    } // else, a call to flush returned, but that call could have been initiated before we were called...
  }

  pageid_t * staleDirtyPages = 0;
  pageid_t n = 0;
  dpt_entry dummy = { start, 0 };
  for(const dpt_entry * e = rblookup(RB_LUGTEQ, &dummy, dirtyPages->table);
         e && (stop == 0 || e->p < stop);
         e = rblookup(RB_LUGREAT, e, dirtyPages->table)) {
    n++;
    staleDirtyPages = realloc(staleDirtyPages, sizeof(pageid_t) * n);
    staleDirtyPages[n-1] = e->p;
  }
  pthread_mutex_unlock(&dirtyPages->mutex);

  for(pageid_t i = 0; i < n; i++) {
      int err = dirtyPages->bufferManager->writeBackPage(dirtyPages->bufferManager, staleDirtyPages[i]);
      if(stop && (err == EBUSY)) { abort(); /*api violation!*/ }
  }
  free(staleDirtyPages);
}

void stasis_dirty_page_table_set_buffer_manager(stasis_dirty_page_table_t * dpt, stasis_buffer_manager_t *bufferManager) {
  dpt->bufferManager = bufferManager;
}

stasis_dirty_page_table_t * stasis_dirty_page_table_init() {
  stasis_dirty_page_table_t * ret = malloc(sizeof(*ret));
  ret->table = rbinit(dpt_cmp, 0);
  ret->count = 0;
  pthread_mutex_init(&ret->mutex, 0);
  pthread_cond_init(&ret->flushDone, 0);
  ret->flushing = 0;
  return ret;
}

void stasis_dirty_page_table_deinit(stasis_dirty_page_table_t * dirtyPages) {
  int areDirty = 0;
  dpt_entry dummy = {0, 0};
  for(const dpt_entry * e = rblookup(RB_LUGTEQ, &dummy, dirtyPages->table);
         e;
         e = rblookup(RB_LUGREAT, &dummy, dirtyPages->table)) {

    if((!areDirty) &&
       (!stasis_suppress_unclean_shutdown_warnings)) {
      printf("Warning:  dirtyPagesDeinit detected dirty, unwritten pages.  "
         "Updates lost?\n");
      areDirty = 1;
    }
    dummy = *e;
    rbdelete(e, dirtyPages->table);
    free((void*)e);
  }

  rbdestroy(dirtyPages->table);
  pthread_mutex_destroy(&dirtyPages->mutex);
  free(dirtyPages);
}