/* FDB_tp_as_mapping.mp_ass_subscript */
static int
FDB_SetItem(FDB *self, PyObject *pykey, PyObject *pyvalue)
{
long long key;
void *value;
int value_size;
key = PyLong_AsLongLong(pykey);
if (key == -1 && PyErr_Occurred()) {
return -1;
}
if (pyvalue) {
if (bytes_to_void(pyvalue, &value, &value_size)) {
return -1;
}
if (!tcfdbput(self->fdb, key, value, value_size)) {
set_fdb_error(self->fdb, 0);
return -1;
}
}
else {
if (!tcfdbout(self->fdb, key)) {
set_fdb_error(self->fdb, key);
return -1;
}
}
self->changed = true;
return 0;
}
/* perform flwrite command */
int doflwrite(char *name, int rnum){
TCFDB *fdb;
int i, err, len;
char buf[RECBUFSIZ];
if(showprgr) printf("<Writing Test of Fixed-Length>\n name=%s rnum=%d\n\n", name, rnum);
/* open a database */
fdb = tcfdbnew();
tcfdbtune(fdb, 8, 1024 + rnum * 9);
if(!tcfdbopen(fdb, name, FDBOWRITER | FDBOCREAT | FDBOTRUNC)){
fprintf(stderr, "tcfdbopen failed\n");
tcfdbdel(fdb);
return 1;
}
err = FALSE;
/* loop for each record */
for(i = 1; i <= rnum; i++){
/* store a record */
len = sprintf(buf, "%08d", i);
if(!tcfdbput(fdb, i, buf, len)){
fprintf(stderr, "tcfdbput failed\n");
err = TRUE;
break;
}
/* print progression */
if(showprgr && rnum > 250 && i % (rnum / 250) == 0){
putchar('.');
fflush(stdout);
if(i == rnum || i % (rnum / 10) == 0){
printf(" (%08d)\n", i);
fflush(stdout);
}
}
}
/* close the database */
if(!tcfdbclose(fdb)){
fprintf(stderr, "tcfdbclose failed\n");
tcfdbdel(fdb);
return 1;
}
tcfdbdel(fdb);
if(showprgr && !err) printf("ok\n\n");
return err ? 1 : 0;
}
/* perform rcat command */
static int procrcat(const char *path, int rnum, int width, int64_t limsiz,
bool mt, int omode, int pnum, bool dai, bool dad, bool rl, bool ru){
my_my_my_iprintf("<Random Concatenating Test>\n"
" seed=%u path=%s rnum=%d width=%d limsiz=%lld mt=%d omode=%d pnum=%d"
" dai=%d dad=%d rl=%d ru=%d\n\n",
g_randseed, path, rnum, width, (long long)limsiz, mt, omode, pnum, dai, dad, rl, ru);
if(pnum < 1) pnum = rnum;
bool err = false;
double stime = tctime();
TCFDB *fdb = tcfdbnew();
if(g_dbgfd >= 0) tcfdbsetdbgfd(fdb, g_dbgfd);
if(mt && !tcfdbsetmutex(fdb)){
eprint(fdb, __LINE__, "tcfdbsetmutex");
err = true;
}
if(!tcfdbtune(fdb, width, limsiz)){
eprint(fdb, __LINE__, "tcfdbtune");
err = true;
}
if(!tcfdbopen(fdb, path, FDBOWRITER | FDBOCREAT | FDBOTRUNC | omode)){
eprint(fdb, __LINE__, "tcfdbopen");
err = true;
}
for(int i = 1; i <= rnum; i++){
char kbuf[RECBUFSIZ];
int ksiz = sprintf(kbuf, "%d", myrand(pnum) + 1);
if(dai){
if(tcfdbaddint(fdb, tcfdbkeytoid(kbuf, ksiz), myrand(3)) == INT_MIN){
eprint(fdb, __LINE__, "tcfdbaddint");
err = true;
break;
}
} else if(dad){
if(isnan(tcfdbadddouble(fdb, tcfdbkeytoid(kbuf, ksiz), myrand(30) / 10.0))){
eprint(fdb, __LINE__, "tcfdbadddouble");
err = true;
break;
}
} else if(rl){
char vbuf[PATH_MAX];
int vsiz = myrand(PATH_MAX);
for(int j = 0; j < vsiz; j++){
vbuf[j] = myrand(0x100);
}
if(!tcfdbputcat2(fdb, kbuf, ksiz, vbuf, vsiz)){
eprint(fdb, __LINE__, "tcfdbputcat");
err = true;
break;
}
} else if(ru){
int id = myrand(pnum) + 1;
switch(myrand(8)){
case 0:
if(!tcfdbput(fdb, id, kbuf, ksiz)){
eprint(fdb, __LINE__, "tcfdbput");
err = true;
}
break;
case 1:
if(!tcfdbputkeep(fdb, id, kbuf, ksiz) && tcfdbecode(fdb) != TCEKEEP){
eprint(fdb, __LINE__, "tcfdbputkeep");
err = true;
}
break;
case 2:
if(!tcfdbout(fdb, id) && tcfdbecode(fdb) != TCENOREC){
eprint(fdb, __LINE__, "tcfdbout");
err = true;
}
break;
case 3:
if(tcfdbaddint(fdb, id, 1) == INT_MIN && tcfdbecode(fdb) != TCEKEEP){
eprint(fdb, __LINE__, "tcfdbaddint");
err = true;
}
break;
case 4:
if(isnan(tcfdbadddouble(fdb, id, 1.0)) && tcfdbecode(fdb) != TCEKEEP){
eprint(fdb, __LINE__, "tcfdbadddouble");
err = true;
}
break;
case 5:
if(myrand(2) == 0){
if(!tcfdbputproc(fdb, id, kbuf, ksiz, pdprocfunc, NULL) &&
tcfdbecode(fdb) != TCEKEEP){
eprint(fdb, __LINE__, "tcfdbputproc");
err = true;
}
} else {
if(!tcfdbputproc(fdb, id, NULL, 0, pdprocfunc, NULL) &&
tcfdbecode(fdb) != TCEKEEP && tcfdbecode(fdb) != TCENOREC){
eprint(fdb, __LINE__, "tcfdbputproc");
err = true;
}
}
break;
default:
if(!tcfdbputcat(fdb, id, kbuf, ksiz)){
eprint(fdb, __LINE__, "tcfdbputcat");
err = true;
}
break;
}
if(err) break;
} else {
if(!tcfdbputcat2(fdb, kbuf, ksiz, kbuf, ksiz)){
eprint(fdb, __LINE__, "tcfdbputcat");
err = true;
break;
}
}
if(rnum > 250 && i % (rnum / 250) == 0){
iputchar('.');
if(i == rnum || i % (rnum / 10) == 0) my_my_my_iprintf(" (%08d)\n", i);
}
}
my_my_my_iprintf("record number: %llu\n", (unsigned long long)tcfdbrnum(fdb));
my_my_my_iprintf("size: %llu\n", (unsigned long long)tcfdbfsiz(fdb));
mprint(fdb);
sysprint();
if(!tcfdbclose(fdb)){
eprint(fdb, __LINE__, "tcfdbclose");
err = true;
}
tcfdbdel(fdb);
my_my_my_iprintf("time: %.3f\n", tctime() - stime);
my_my_my_iprintf("%s\n\n", err ? "error" : "ok");
return err ? 1 : 0;
}
