#include "db_config.h"
#include "db_int.h"
#include "dbinc/db_page.h"
#include "dbinc/db_am.h"
#include "dbinc/lock.h"
#include "dbinc/mp.h"
#include "dbinc/qam.h"
#include "dbinc/txn.h"
static int __qam_rr __P((DB *, DB_TXN *,
const char *, const char *, const char *, qam_name_op));
static int __qam_set_extentsize __P((DB *, u_int32_t));
int
__qam_db_create(dbp)
DB *dbp;
{
QUEUE *t;
int ret;
if ((ret = __os_calloc(dbp->dbenv, 1, sizeof(QUEUE), &t)) != 0)
return (ret);
dbp->q_internal = t;
dbp->get_q_extentsize = __qam_get_extentsize;
dbp->set_q_extentsize = __qam_set_extentsize;
t->re_pad = ' ';
return (0);
}
int
__qam_db_close(dbp, flags)
DB *dbp;
u_int32_t flags;
{
DB_MPOOLFILE *mpf;
MPFARRAY *array;
QUEUE *t;
struct __qmpf *mpfp;
u_int32_t i;
int ret, t_ret;
ret = 0;
if ((t = dbp->q_internal) == NULL)
return (0);
array = &t->array1;
again:
mpfp = array->mpfarray;
if (mpfp != NULL) {
for (i = array->low_extent;
i <= array->hi_extent; i++, mpfp++) {
mpf = mpfp->mpf;
mpfp->mpf = NULL;
if (mpf != NULL && (t_ret = __memp_fclose(mpf,
LF_ISSET(DB_AM_DISCARD) ? DB_MPOOL_DISCARD : 0))
!= 0 && ret == 0)
ret = t_ret;
}
__os_free(dbp->dbenv, array->mpfarray);
}
if (t->array2.n_extent != 0) {
array = &t->array2;
array->n_extent = 0;
goto again;
}
if (LF_ISSET(DB_AM_DISCARD) &&
(t_ret = __qam_nameop(dbp, NULL,
NULL, QAM_NAME_DISCARD)) != 0 && ret == 0)
ret = t_ret;
if (t->path != NULL)
__os_free(dbp->dbenv, t->path);
__os_free(dbp->dbenv, t);
dbp->q_internal = NULL;
return (ret);
}
int
__qam_get_extentsize(dbp, q_extentsizep)
DB *dbp;
u_int32_t *q_extentsizep;
{
*q_extentsizep = ((QUEUE*)dbp->q_internal)->page_ext;
return (0);
}
static int
__qam_set_extentsize(dbp, extentsize)
DB *dbp;
u_int32_t extentsize;
{
DB_ILLEGAL_AFTER_OPEN(dbp, "DB->set_extentsize");
if (extentsize < 1) {
__db_errx(dbp->dbenv, "Extent size must be at least 1");
return (EINVAL);
}
((QUEUE*)dbp->q_internal)->page_ext = extentsize;
return (0);
}
int
__queue_pageinfo(dbp, firstp, lastp, emptyp, prpage, flags)
DB *dbp;
db_pgno_t *firstp, *lastp;
int *emptyp;
int prpage;
u_int32_t flags;
{
DB_MPOOLFILE *mpf;
QMETA *meta;
db_pgno_t first, i, last;
int empty, ret, t_ret;
mpf = dbp->mpf;
i = PGNO_BASE_MD;
if ((ret = __memp_fget(mpf, &i, NULL, 0, &meta)) != 0)
return (ret);
first = QAM_RECNO_PAGE(dbp, meta->first_recno);
last = QAM_RECNO_PAGE(
dbp, meta->cur_recno == 1 ? 1 : meta->cur_recno - 1);
empty = meta->cur_recno == meta->first_recno;
if (firstp != NULL)
*firstp = first;
if (lastp != NULL)
*lastp = last;
if (emptyp != NULL)
*emptyp = empty;
#ifdef HAVE_STATISTICS
if (prpage)
ret = __db_prpage(dbp, (PAGE *)meta, flags);
#else
COMPQUIET(prpage, 0);
COMPQUIET(flags, 0);
#endif
if ((t_ret = __memp_fput(mpf, meta, dbp->priority)) != 0 && ret == 0)
ret = t_ret;
return (ret);
}
#ifdef HAVE_STATISTICS
int
__db_prqueue(dbp, flags)
DB *dbp;
u_int32_t flags;
{
PAGE *h;
db_pgno_t first, i, last, pg_ext, stop;
int empty, ret;
if ((ret = __queue_pageinfo(dbp, &first, &last, &empty, 1, flags)) != 0)
return (ret);
if (empty || ret != 0)
return (ret);
i = first;
if (first > last)
stop = QAM_RECNO_PAGE(dbp, UINT32_MAX);
else
stop = last;
begin:
for (; i <= stop; ++i) {
if ((ret = __qam_fget(dbp, &i, NULL, 0, &h)) != 0) {
pg_ext = ((QUEUE *)dbp->q_internal)->page_ext;
if (pg_ext == 0) {
if (ret == DB_PAGE_NOTFOUND && first == last)
return (0);
return (ret);
}
if (ret == ENOENT || ret == DB_PAGE_NOTFOUND) {
i += (pg_ext - ((i - 1) % pg_ext)) - 1;
continue;
}
return (ret);
}
(void)__db_prpage(dbp, h, flags);
if ((ret = __qam_fput(dbp, i, h, dbp->priority)) != 0)
return (ret);
}
if (first > last) {
i = 1;
stop = last;
first = last;
goto begin;
}
return (0);
}
#endif
int
__qam_remove(dbp, txn, name, subdb)
DB *dbp;
DB_TXN *txn;
const char *name, *subdb;
{
return (__qam_rr(dbp, txn, name, subdb, NULL, QAM_NAME_REMOVE));
}
int
__qam_rename(dbp, txn, name, subdb, newname)
DB *dbp;
DB_TXN *txn;
const char *name, *subdb, *newname;
{
return (__qam_rr(dbp, txn, name, subdb, newname, QAM_NAME_RENAME));
}
static int
__qam_rr(dbp, txn, name, subdb, newname, op)
DB *dbp;
DB_TXN *txn;
const char *name, *subdb, *newname;
qam_name_op op;
{
DB_ENV *dbenv;
DB *tmpdbp;
QUEUE *qp;
int ret, t_ret;
dbenv = dbp->dbenv;
ret = 0;
if (subdb != NULL && name != NULL) {
__db_errx(dbenv,
"Queue does not support multiple databases per file");
return (EINVAL);
}
if (F_ISSET(dbp, DB_AM_OPEN_CALLED))
tmpdbp = dbp;
else {
if ((ret = __db_create_internal(&tmpdbp, dbenv, 0)) != 0)
return (ret);
tmpdbp->locker = dbp->locker;
if ((ret = __db_open(tmpdbp, txn,
name, NULL, DB_QUEUE, DB_RDONLY, 0, PGNO_BASE_MD)) != 0)
goto err;
}
qp = (QUEUE *)tmpdbp->q_internal;
if (qp->page_ext != 0)
ret = __qam_nameop(tmpdbp, txn, newname, op);
if (!F_ISSET(dbp, DB_AM_OPEN_CALLED)) {
err:
tmpdbp->locker = NULL;
if (txn != NULL)
__txn_remlock(dbenv,
txn, &tmpdbp->handle_lock, DB_LOCK_INVALIDID);
if (txn == NULL ) {
if ((t_ret = __db_close(tmpdbp,
txn, DB_NOSYNC)) != 0 && ret == 0)
ret = t_ret;
} else {
if ((t_ret = __txn_closeevent(dbenv,
txn, tmpdbp)) != 0 && ret == 0)
ret = t_ret;
}
}
return (ret);
}
void
__qam_map_flags(dbp, inflagsp, outflagsp)
DB *dbp;
u_int32_t *inflagsp, *outflagsp;
{
COMPQUIET(dbp, NULL);
if (FLD_ISSET(*inflagsp, DB_INORDER)) {
FLD_SET(*outflagsp, DB_AM_INORDER);
FLD_CLR(*inflagsp, DB_INORDER);
}
}
int
__qam_set_flags(dbp, flagsp)
DB *dbp;
u_int32_t *flagsp;
{
__qam_map_flags(dbp, flagsp, &dbp->flags);
return (0);
}