feat(lsm): add entry data reading
parent
1461956d98
commit
fc4187e6ce
|
@ -1,5 +1,46 @@
|
||||||
|
#include <assert.h>
|
||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
|
|
||||||
#include "lsm.h"
|
#include "lsm.h"
|
||||||
|
#include "lsm/store.h"
|
||||||
|
#include "lsm/str.h"
|
||||||
|
|
||||||
int main() { printf("yuh"); }
|
int main() {
|
||||||
|
lsm_str *db_path, *data_dir;
|
||||||
|
lsm_str_init_copy(&db_path, "data/data.db");
|
||||||
|
lsm_str_init_copy(&data_dir, "data");
|
||||||
|
|
||||||
|
lsm_store *store;
|
||||||
|
lsm_store_load(&store, db_path, data_dir);
|
||||||
|
|
||||||
|
lsm_str *key;
|
||||||
|
lsm_str_init_copy(&key, "key");
|
||||||
|
|
||||||
|
lsm_entry_handle *handle;
|
||||||
|
assert(lsm_store_insert(&handle, store, key) == lsm_error_ok);
|
||||||
|
|
||||||
|
lsm_str *data;
|
||||||
|
lsm_str_init_copy(&data, "hello");
|
||||||
|
|
||||||
|
for (int i = 0; i < 50; i++) {
|
||||||
|
lsm_entry_data_append(store, handle, data);
|
||||||
|
}
|
||||||
|
|
||||||
|
lsm_entry_close(handle);
|
||||||
|
|
||||||
|
assert(lsm_store_open_read(&handle, store, key) == lsm_error_ok);
|
||||||
|
|
||||||
|
char buf[24];
|
||||||
|
uint64_t read;
|
||||||
|
uint64_t total = 0;
|
||||||
|
|
||||||
|
lsm_entry_data_read(&read, buf, handle, 24);
|
||||||
|
total += read;
|
||||||
|
|
||||||
|
while (read > 0) {
|
||||||
|
printf("%.*s", read, buf);
|
||||||
|
lsm_entry_data_read(&read, buf, handle, 24);
|
||||||
|
total += read;
|
||||||
|
}
|
||||||
|
printf("\n%lu", total);
|
||||||
|
}
|
||||||
|
|
|
@ -151,4 +151,16 @@ lsm_error lsm_store_insert(lsm_entry_handle **out, lsm_store *store,
|
||||||
lsm_error lsm_entry_data_append(lsm_store *store, lsm_entry_handle *handle,
|
lsm_error lsm_entry_data_append(lsm_store *store, lsm_entry_handle *handle,
|
||||||
lsm_str *data);
|
lsm_str *data);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Read a number of bytes from the entry's data field. The position from which
|
||||||
|
* data is read is dependent on previous read calls.
|
||||||
|
*
|
||||||
|
* @param out where to write how many bytes were read
|
||||||
|
* @param buf buffer to store read data in
|
||||||
|
* @param handle entry handle to read from
|
||||||
|
* @param len how many bytes to read at most
|
||||||
|
*/
|
||||||
|
lsm_error lsm_entry_data_read(uint64_t *out, char *buf,
|
||||||
|
lsm_entry_handle *handle, uint64_t len);
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
|
@ -62,6 +62,7 @@ void lsm_entry_wrapper_free(lsm_entry_wrapper *wrapper);
|
||||||
struct lsm_entry_handle {
|
struct lsm_entry_handle {
|
||||||
lsm_entry_wrapper *wrapper;
|
lsm_entry_wrapper *wrapper;
|
||||||
FILE *f;
|
FILE *f;
|
||||||
|
uint64_t pos;
|
||||||
};
|
};
|
||||||
|
|
||||||
lsm_error lsm_entry_handle_init(lsm_entry_handle **out);
|
lsm_error lsm_entry_handle_init(lsm_entry_handle **out);
|
||||||
|
|
|
@ -259,3 +259,34 @@ lsm_error lsm_entry_data_append(lsm_store *store, lsm_entry_handle *handle,
|
||||||
|
|
||||||
return lsm_error_ok;
|
return lsm_error_ok;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
lsm_error lsm_entry_data_read(uint64_t *out, char *buf,
|
||||||
|
lsm_entry_handle *handle, uint64_t len) {
|
||||||
|
lsm_entry *entry = handle->wrapper->entry;
|
||||||
|
|
||||||
|
if (entry->data.len == 0) {
|
||||||
|
*out = 0;
|
||||||
|
|
||||||
|
return lsm_error_ok;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint64_t read;
|
||||||
|
|
||||||
|
if (entry->data.on_disk) {
|
||||||
|
read = fread(buf, sizeof(char), len, handle->f);
|
||||||
|
|
||||||
|
if ((read == 0) && (ferror(handle->f) != 0)) {
|
||||||
|
return lsm_error_failed_io;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
read = (entry->data.len - handle->pos) < len
|
||||||
|
? (entry->data.len - handle->pos)
|
||||||
|
: len;
|
||||||
|
memcpy(buf, &entry->data.value.ptr[handle->pos], read * sizeof(char));
|
||||||
|
}
|
||||||
|
|
||||||
|
handle->pos += read;
|
||||||
|
*out = read;
|
||||||
|
|
||||||
|
return lsm_error_ok;
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in New Issue