feat(lsm): add entry data reading

lsm
Jef Roosens 2023-10-29 14:41:40 +01:00
parent 1461956d98
commit fc4187e6ce
Signed by: Jef Roosens
GPG Key ID: B75D4F293C7052DB
4 changed files with 86 additions and 1 deletions

View File

@ -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);
}

View File

@ -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

View File

@ -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);

View File

@ -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;
}