feat(lsm): add entry data reading
parent
1461956d98
commit
fc4187e6ce
|
@ -1,5 +1,46 @@
|
|||
#include <assert.h>
|
||||
#include <stdio.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_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
|
||||
|
|
|
@ -62,6 +62,7 @@ void lsm_entry_wrapper_free(lsm_entry_wrapper *wrapper);
|
|||
struct lsm_entry_handle {
|
||||
lsm_entry_wrapper *wrapper;
|
||||
FILE *f;
|
||||
uint64_t pos;
|
||||
};
|
||||
|
||||
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;
|
||||
}
|
||||
|
||||
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