Furi: refactor records (#736)
Co-authored-by: DrZlo13 <who.just.the.doctor@gmail.com>
This commit is contained in:
あく
GitHub
parent
61aaed8abb
commit
05b610f265
@ -6,127 +6,134 @@
|
||||
#include <m-string.h>
|
||||
#include <m-dict.h>
|
||||
|
||||
#define FURI_RECORD_FLAG_UPDATED 0x00000001U
|
||||
|
||||
DICT_SET_DEF(osThreadIdSet, uint32_t)
|
||||
#define FURI_RECORD_FLAG_READY (0x1)
|
||||
|
||||
typedef struct {
|
||||
osEventFlagsId_t flags;
|
||||
void* data;
|
||||
osThreadId_t owner;
|
||||
osThreadIdSet_t holders;
|
||||
} FuriRecord;
|
||||
|
||||
DICT_DEF2(FuriRecordDict, string_t, STRING_OPLIST, FuriRecord, M_POD_OPLIST)
|
||||
|
||||
typedef struct {
|
||||
osMutexId_t records_mutex;
|
||||
FuriRecordDict_t records;
|
||||
size_t holders_count;
|
||||
} FuriRecordData;
|
||||
|
||||
static FuriRecordData* furi_record_data = NULL;
|
||||
DICT_DEF2(FuriRecordDataDict, string_t, STRING_OPLIST, FuriRecordData, M_POD_OPLIST)
|
||||
|
||||
typedef struct {
|
||||
osMutexId_t mutex;
|
||||
FuriRecordDataDict_t records;
|
||||
} FuriRecord;
|
||||
|
||||
static FuriRecord* furi_record = NULL;
|
||||
|
||||
void furi_record_init() {
|
||||
furi_record_data = furi_alloc(sizeof(FuriRecordData));
|
||||
furi_record_data->records_mutex = osMutexNew(NULL);
|
||||
furi_check(furi_record_data->records_mutex);
|
||||
FuriRecordDict_init(furi_record_data->records);
|
||||
furi_record = furi_alloc(sizeof(FuriRecord));
|
||||
furi_record->mutex = osMutexNew(NULL);
|
||||
furi_check(furi_record->mutex);
|
||||
FuriRecordDataDict_init(furi_record->records);
|
||||
}
|
||||
|
||||
FuriRecord* furi_record_get_or_create(string_t name_str) {
|
||||
furi_assert(furi_record_data);
|
||||
FuriRecord* record = FuriRecordDict_get(furi_record_data->records, name_str);
|
||||
if(!record) {
|
||||
FuriRecord new_record;
|
||||
static FuriRecordData* furi_record_data_get_or_create(string_t name_str) {
|
||||
furi_assert(furi_record);
|
||||
FuriRecordData* record_data = FuriRecordDataDict_get(furi_record->records, name_str);
|
||||
if(!record_data) {
|
||||
FuriRecordData new_record;
|
||||
new_record.flags = osEventFlagsNew(NULL);
|
||||
new_record.data = NULL;
|
||||
new_record.owner = NULL;
|
||||
osThreadIdSet_init(new_record.holders);
|
||||
FuriRecordDict_set_at(furi_record_data->records, name_str, new_record);
|
||||
record = FuriRecordDict_get(furi_record_data->records, name_str);
|
||||
new_record.holders_count = 0;
|
||||
FuriRecordDataDict_set_at(furi_record->records, name_str, new_record);
|
||||
record_data = FuriRecordDataDict_get(furi_record->records, name_str);
|
||||
}
|
||||
return record;
|
||||
return record_data;
|
||||
}
|
||||
|
||||
static void furi_record_lock() {
|
||||
furi_check(osMutexAcquire(furi_record->mutex, osWaitForever) == osOK);
|
||||
}
|
||||
|
||||
static void furi_record_unlock() {
|
||||
furi_check(osMutexRelease(furi_record->mutex) == osOK);
|
||||
}
|
||||
|
||||
void furi_record_create(const char* name, void* data) {
|
||||
furi_assert(furi_record_data);
|
||||
osThreadId_t thread_id = osThreadGetId();
|
||||
furi_assert(furi_record);
|
||||
|
||||
string_t name_str;
|
||||
string_init_set_str(name_str, name);
|
||||
|
||||
// Acquire mutex
|
||||
furi_check(osMutexAcquire(furi_record_data->records_mutex, osWaitForever) == osOK);
|
||||
FuriRecord* record = furi_record_get_or_create(name_str);
|
||||
record->data = data;
|
||||
record->owner = thread_id;
|
||||
furi_record_lock();
|
||||
|
||||
// For each holder set event flag
|
||||
osThreadIdSet_it_t it;
|
||||
for(osThreadIdSet_it(it, record->holders); !osThreadIdSet_end_p(it); osThreadIdSet_next(it)) {
|
||||
osThreadFlagsSet((osThreadId_t)*osThreadIdSet_ref(it), FURI_RECORD_FLAG_UPDATED);
|
||||
}
|
||||
// Release mutex
|
||||
furi_check(osMutexRelease(furi_record_data->records_mutex) == osOK);
|
||||
// Get record data and fill it
|
||||
FuriRecordData* record_data = furi_record_data_get_or_create(name_str);
|
||||
furi_assert(record_data->data == NULL);
|
||||
record_data->data = data;
|
||||
osEventFlagsSet(record_data->flags, FURI_RECORD_FLAG_READY);
|
||||
|
||||
furi_record_unlock();
|
||||
|
||||
string_clear(name_str);
|
||||
}
|
||||
|
||||
bool furi_record_destroy(const char* name) {
|
||||
furi_assert(furi_record_data);
|
||||
osThreadId_t thread_id = osThreadGetId();
|
||||
furi_assert(furi_record);
|
||||
|
||||
bool ret = false;
|
||||
|
||||
string_t name_str;
|
||||
string_init_set_str(name_str, name);
|
||||
|
||||
bool destroyed = false;
|
||||
furi_check(osMutexAcquire(furi_record_data->records_mutex, osWaitForever) == osOK);
|
||||
FuriRecord* record = FuriRecordDict_get(furi_record_data->records, name_str);
|
||||
if(record && record->owner == thread_id && osThreadIdSet_size(record->holders) == 0) {
|
||||
osThreadIdSet_clear(record->holders);
|
||||
FuriRecordDict_erase(furi_record_data->records, name_str);
|
||||
destroyed = true;
|
||||
furi_record_lock();
|
||||
|
||||
FuriRecordData* record_data = FuriRecordDataDict_get(furi_record->records, name_str);
|
||||
furi_assert(record_data);
|
||||
if(record_data->holders_count == 0) {
|
||||
FuriRecordDataDict_erase(furi_record->records, name_str);
|
||||
ret = true;
|
||||
}
|
||||
furi_check(osMutexRelease(furi_record_data->records_mutex) == osOK);
|
||||
|
||||
furi_record_unlock();
|
||||
|
||||
string_clear(name_str);
|
||||
return destroyed;
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
void* furi_record_open(const char* name) {
|
||||
furi_assert(furi_record_data);
|
||||
osThreadId_t thread_id = osThreadGetId();
|
||||
furi_assert(furi_record);
|
||||
|
||||
string_t name_str;
|
||||
string_init_set_str(name_str, name);
|
||||
|
||||
FuriRecord* record = NULL;
|
||||
while(1) {
|
||||
furi_check(osMutexAcquire(furi_record_data->records_mutex, osWaitForever) == osOK);
|
||||
record = furi_record_get_or_create(name_str);
|
||||
osThreadIdSet_push(record->holders, (uint32_t)thread_id);
|
||||
furi_check(osMutexRelease(furi_record_data->records_mutex) == osOK);
|
||||
// Check if owner is already arrived
|
||||
if(record->owner) {
|
||||
break;
|
||||
}
|
||||
// Wait for thread flag to appear
|
||||
osThreadFlagsWait(FURI_RECORD_FLAG_UPDATED, osFlagsWaitAny, osWaitForever);
|
||||
}
|
||||
furi_record_lock();
|
||||
|
||||
FuriRecordData* record_data = furi_record_data_get_or_create(name_str);
|
||||
record_data->holders_count++;
|
||||
|
||||
furi_record_unlock();
|
||||
|
||||
// Wait for record to become ready
|
||||
furi_check(
|
||||
osEventFlagsWait(
|
||||
record_data->flags,
|
||||
FURI_RECORD_FLAG_READY,
|
||||
osFlagsWaitAny | osFlagsNoClear,
|
||||
osWaitForever) == FURI_RECORD_FLAG_READY);
|
||||
|
||||
string_clear(name_str);
|
||||
return record->data;
|
||||
|
||||
return record_data->data;
|
||||
}
|
||||
|
||||
void furi_record_close(const char* name) {
|
||||
furi_assert(furi_record_data);
|
||||
osThreadId_t thread_id = osThreadGetId();
|
||||
furi_assert(furi_record);
|
||||
|
||||
string_t name_str;
|
||||
string_init_set_str(name_str, name);
|
||||
|
||||
furi_check(osMutexAcquire(furi_record_data->records_mutex, osWaitForever) == osOK);
|
||||
FuriRecord* record = FuriRecordDict_get(furi_record_data->records, name_str);
|
||||
osThreadIdSet_erase(record->holders, (uint32_t)thread_id);
|
||||
furi_check(osMutexRelease(furi_record_data->records_mutex) == osOK);
|
||||
furi_record_lock();
|
||||
|
||||
FuriRecordData* record_data = FuriRecordDataDict_get(furi_record->records, name_str);
|
||||
furi_assert(record_data);
|
||||
record_data->holders_count--;
|
||||
|
||||
furi_record_unlock();
|
||||
|
||||
string_clear(name_str);
|
||||
}
|
||||
|
||||
@ -6,30 +6,26 @@
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
/**
|
||||
* Initialize record storage
|
||||
/** Initialize record storage
|
||||
* For internal use only.
|
||||
*/
|
||||
void furi_record_init();
|
||||
|
||||
/**
|
||||
* Create record
|
||||
/** Create record
|
||||
* @param name - record name
|
||||
* @param data - data pointer
|
||||
* @note Thread safe. Create and destroy must be executed from the same thread.
|
||||
*/
|
||||
void furi_record_create(const char* name, void* data);
|
||||
|
||||
/**
|
||||
* Destroy record
|
||||
/** Destroy record
|
||||
* @param name - record name
|
||||
* @return true if successful, false if still have holders or thread is not owner.
|
||||
* @note Thread safe. Create and destroy must be executed from the same thread.
|
||||
*/
|
||||
bool furi_record_destroy(const char* name);
|
||||
|
||||
/**
|
||||
* Open record
|
||||
/** Open record
|
||||
* @param name - record name
|
||||
* @return pointer to the record
|
||||
* @note Thread safe. Open and close must be executed from the same thread.
|
||||
@ -37,8 +33,7 @@ bool furi_record_destroy(const char* name);
|
||||
*/
|
||||
void* furi_record_open(const char* name);
|
||||
|
||||
/**
|
||||
* Close record
|
||||
/** Close record
|
||||
* @param name - record name
|
||||
* @note Thread safe. Open and close must be executed from the same thread.
|
||||
*/
|
||||
|
||||
Loading…
Reference in New Issue
Block a user