Merge default.

[irc/quakenet/newserv.git] / trusts / db-migration.c

#include "../dbapi2/dbapi2.h"
#include "../core/error.h"
#include "../core/nsmalloc.h"
#include "trusts.h"

extern DBAPIConn *trustsdb;
static trustmigration *migration;

static void tm_group(void *, trustgroup *);
static void tm_host(void *, trusthost *, unsigned int);
static void tm_final(void *, int);

trustmigration *migration_start(TrustMigrationGroup, TrustMigrationHost, TrustMigrationFini, void *);
void migration_stop(trustmigration *);
void createtrusttables(int);

struct callbackdata {
  void *tag;
  unsigned int hostid;
  TrustDBMigrationCallback callback;
};

int trusts_migration_start(TrustDBMigrationCallback callback, void *tag) {
  struct callbackdata *cbd;

  if(migration)
    return 1;

  cbd = nsmalloc(POOL_TRUSTS, sizeof(struct callbackdata));
  if(!cbd)
    return 2;

  cbd->callback = callback;
  cbd->tag = tag;
  cbd->hostid = 1;

  createtrusttables(TABLES_MIGRATION);
  trustsdb->squery(trustsdb, "DELETE FROM ?", "T", "migration_groups");
  trustsdb->squery(trustsdb, "DELETE FROM ?", "T", "migration_hosts");

  migration = migration_start(tm_group, tm_host, tm_final, cbd);
  if(!migration) {
    nsfree(POOL_TRUSTS, cbd);
    return 3;
  }

  return 0;
}

void trusts_migration_stop(void) {
  if(!migration)
    return;

  migration_stop(migration);
}

static void tm_group(void *tag, trustgroup *tg) {
  if(tg->id % 25 == 0)
    Error("trusts_migration", ERR_INFO, "Migration currently at id: %d", tg->id);

  trustsdb_inserttg("migration_groups", tg);
}

static void tm_host(void *tag, trusthost *th, unsigned int groupid) {
  struct callbackdata *cbd = tag;

  th->id = cbd->hostid++;
  trustsdb_insertth("migration_hosts", th, groupid);
}

static void tm_complete(const DBAPIResult *r, void *tag) {
  struct callbackdata *cbd = tag;
  int errcode = 0;

  if(!r) {
    errcode = MIGRATION_STOPPED;
  } else {
    if(!r->success) {
      Error("trusts_migration", ERR_ERROR, "A error occured executing the rename table query.");
      errcode = MIGRATION_LASTERROR;
    } else {
      Error("trusts_migration", ERR_INFO, "Migration table copying complete.");
    }
    r->clear(r);
  }

  if(cbd->callback)
    cbd->callback(errcode, cbd->tag);

  nsfree(POOL_TRUSTS, cbd);
}

static void tm_final(void *tag, int errcode) {
  struct callbackdata *cbd = tag;
  migration = NULL;

  if(errcode) {
    Error("trusts_migration", ERR_ERROR, "Migration error: %d", errcode);
    if(cbd) {
      cbd->callback(errcode, cbd->tag);
      nsfree(POOL_TRUSTS, cbd);
    }
  } else {
    trusts_closedb(0);

    Error("trusts_migration", ERR_INFO, "Migration completed, copying tables...");

    trustsdb->squery(trustsdb, "BEGIN TRANSACTION", "");
    trustsdb->squery(trustsdb, "DROP TABLE ?", "T", "groups");
    trustsdb->squery(trustsdb, "ALTER TABLE ? RENAME TO ?", "Ts", "migration_groups", "groups");
    trustsdb->squery(trustsdb, "DROP TABLE ?", "T", "hosts");
    trustsdb->squery(trustsdb, "ALTER TABLE ? RENAME TO ?", "Ts", "migration_hosts", "hosts");
    trustsdb->query(trustsdb, tm_complete, cbd, "COMMIT", "");
  }
}