Fix mdb_load append option
Use MDB_CURRENT to avoid seeking to new key when appending DUPs. Reinit cursor on new batch, if appending DUPs.
This commit is contained in:
parent
9705032618
commit
18cd6837b6
|
@ -7231,7 +7231,7 @@ mdb_cursor_put(MDB_cursor *mc, MDB_val *key, MDB_val *data,
|
||||||
|
|
||||||
dkey.mv_size = 0;
|
dkey.mv_size = 0;
|
||||||
|
|
||||||
if (flags == MDB_CURRENT) {
|
if (flags & MDB_CURRENT) {
|
||||||
if (!(mc->mc_flags & C_INITIALIZED))
|
if (!(mc->mc_flags & C_INITIALIZED))
|
||||||
return EINVAL;
|
return EINVAL;
|
||||||
rc = MDB_SUCCESS;
|
rc = MDB_SUCCESS;
|
||||||
|
@ -7624,7 +7624,7 @@ put_sub:
|
||||||
xdata.mv_size = 0;
|
xdata.mv_size = 0;
|
||||||
xdata.mv_data = "";
|
xdata.mv_data = "";
|
||||||
leaf = NODEPTR(mc->mc_pg[mc->mc_top], mc->mc_ki[mc->mc_top]);
|
leaf = NODEPTR(mc->mc_pg[mc->mc_top], mc->mc_ki[mc->mc_top]);
|
||||||
if (flags & MDB_CURRENT) {
|
if (flags == MDB_CURRENT) {
|
||||||
xflags = MDB_CURRENT|MDB_NOSPILL;
|
xflags = MDB_CURRENT|MDB_NOSPILL;
|
||||||
} else {
|
} else {
|
||||||
mdb_xcursor_init1(mc, leaf);
|
mdb_xcursor_init1(mc, leaf);
|
||||||
|
|
|
@ -304,7 +304,7 @@ int main(int argc, char *argv[])
|
||||||
MDB_cursor *mc;
|
MDB_cursor *mc;
|
||||||
MDB_dbi dbi;
|
MDB_dbi dbi;
|
||||||
char *envname;
|
char *envname;
|
||||||
int envflags = 0, putflags = 0;
|
int envflags = MDB_NOSYNC, putflags = 0;
|
||||||
int dohdr = 0, append = 0;
|
int dohdr = 0, append = 0;
|
||||||
MDB_val prevk;
|
MDB_val prevk;
|
||||||
|
|
||||||
|
@ -392,7 +392,6 @@ int main(int argc, char *argv[])
|
||||||
kbuf.mv_data = malloc(kbuf.mv_size * 2);
|
kbuf.mv_data = malloc(kbuf.mv_size * 2);
|
||||||
k0buf.mv_size = kbuf.mv_size;
|
k0buf.mv_size = kbuf.mv_size;
|
||||||
k0buf.mv_data = (char *)kbuf.mv_data + kbuf.mv_size;
|
k0buf.mv_data = (char *)kbuf.mv_data + kbuf.mv_size;
|
||||||
prevk.mv_size = 0;
|
|
||||||
prevk.mv_data = k0buf.mv_data;
|
prevk.mv_data = k0buf.mv_data;
|
||||||
|
|
||||||
while(!Eof) {
|
while(!Eof) {
|
||||||
|
@ -416,6 +415,7 @@ int main(int argc, char *argv[])
|
||||||
fprintf(stderr, "mdb_open failed, error %d %s\n", rc, mdb_strerror(rc));
|
fprintf(stderr, "mdb_open failed, error %d %s\n", rc, mdb_strerror(rc));
|
||||||
goto txn_abort;
|
goto txn_abort;
|
||||||
}
|
}
|
||||||
|
prevk.mv_size = 0;
|
||||||
if (append) {
|
if (append) {
|
||||||
mdb_set_compare(txn, dbi, greater);
|
mdb_set_compare(txn, dbi, greater);
|
||||||
if (flags & MDB_DUPSORT)
|
if (flags & MDB_DUPSORT)
|
||||||
|
@ -443,7 +443,7 @@ int main(int argc, char *argv[])
|
||||||
appflag = MDB_APPEND;
|
appflag = MDB_APPEND;
|
||||||
if (flags & MDB_DUPSORT) {
|
if (flags & MDB_DUPSORT) {
|
||||||
if (prevk.mv_size == key.mv_size && !memcmp(prevk.mv_data, key.mv_data, key.mv_size))
|
if (prevk.mv_size == key.mv_size && !memcmp(prevk.mv_data, key.mv_data, key.mv_size))
|
||||||
appflag = MDB_APPENDDUP;
|
appflag = MDB_CURRENT|MDB_APPENDDUP;
|
||||||
else {
|
else {
|
||||||
memcpy(prevk.mv_data, key.mv_data, key.mv_size);
|
memcpy(prevk.mv_data, key.mv_data, key.mv_size);
|
||||||
prevk.mv_size = key.mv_size;
|
prevk.mv_size = key.mv_size;
|
||||||
|
@ -477,6 +477,10 @@ int main(int argc, char *argv[])
|
||||||
fprintf(stderr, "mdb_cursor_open failed, error %d %s\n", rc, mdb_strerror(rc));
|
fprintf(stderr, "mdb_cursor_open failed, error %d %s\n", rc, mdb_strerror(rc));
|
||||||
goto txn_abort;
|
goto txn_abort;
|
||||||
}
|
}
|
||||||
|
if (appflag & MDB_APPENDDUP) {
|
||||||
|
MDB_val k, d;
|
||||||
|
mdb_cursor_get(mc, &k, &d, MDB_LAST);
|
||||||
|
}
|
||||||
batch = 0;
|
batch = 0;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue