amxrt_save_load.c File Reference

#include <stdlib.h>
#include <string.h>
#include <amxrt/amxrt.h>
#include <amxo/amxo_save.h>
#include <amxd/amxd_object.h>
#include <amxp/amxp_dir.h>
#include "amxrt_priv.h"

Go to the source code of this file.

Data Structures
struct	_odl_storage

Typedefs
typedef struct _odl_storage	odl_storage_t

Functions
static int	isdot (int c)
static amxc_var_t *	amxrt_get_save_objects (amxo_parser_t *parser)
static char *	amxrt_get_directory (amxo_parser_t *parser)
static int	amxrt_dm_load (amxd_dm_t *dm, amxo_parser_t *parser)
static int	amxrt_dm_save_object (amxo_parser_t *parser, amxd_object_t *object, const char *dir, const char *name)
static int	amxrt_dm_save (amxd_dm_t *dm, amxo_parser_t *parser)
static bool	amxrt_save_check_tree (amxd_object_t *object)
static bool	amxrt_save_check_persistent_params (amxd_object_t *object, const amxc_var_t *data)
static void	amxrt_save_add_path (const char *path)
static bool	amxrt_save_check_paths (amxo_parser_t *parser, amxd_object_t *object)
static void	amxrt_timed_save (UNUSED amxp_timer_t *const timer, UNUSED void *data)
static void	amxrt_save_changed (const char *const sig_name, const amxc_var_t *const data, UNUSED void *const priv)
static void	amxrt_save_subscribe (amxd_dm_t *dm)
static bool	amxrt_is_odl_storage_enabled (amxo_parser_t *parser)
static void	amxrt_monitor_changes (UNUSED const char *const sig_name, UNUSED const amxc_var_t *const data, UNUSED void *const priv)
static int	amxrt_odl_save_load_init (amxd_dm_t *dm, amxo_parser_t *parser)
static void	amxrt_odl_save_load_cleanup (amxd_dm_t *dm, amxo_parser_t *parser)
int	amxrt_dm_create_dir (amxo_parser_t *parser, uid_t uid, gid_t gid)
int	amxrt_dm_save_load_main (int reason, amxd_dm_t *dm, amxo_parser_t *parser)
	The data model auto load and save module. More...

Variables
static odl_storage_t	storage

Typedef Documentation

odl_storage_t

typedef struct _odl_storage odl_storage_t

Function Documentation

amxrt_dm_create_dir()

int amxrt_dm_create_dir	(	amxo_parser_t *	parser,
		uid_t	uid,
		gid_t	gid
	)

Definition at line 436 of file amxrt_save_load.c.

                                                                     {
    int rv = -1;
    char* directory = NULL;
 
    when_null(parser, exit);
 
    if(!amxrt_is_odl_storage_enabled(parser)) {
        goto exit;
    }
 
    directory = amxrt_get_directory(parser);
    rv = amxp_dir_owned_make(directory, 0777, uid, gid);
    if(rv != 0) {
        amxrt_print_error("Failed to create directory %s", directory);
    }
    free(directory);
 
exit:
    return rv;
}

amxrt_dm_load()

static int amxrt_dm_load ( amxd_dm_t *  dm, amxo_parser_t *  parser  )

static

Definition at line 117 of file amxrt_save_load.c.

                                                               {
    int status = 0;
    amxc_string_t include;
    bool is_empty_dir = false;
 
    amxd_object_t* root = amxd_dm_get_root(dm);
    char* odl_dir = amxrt_get_directory(parser);
    amxc_var_t* eventing = GET_ARG(&parser->config, AMXRT_COPT_EVENT);
    bool orig_eventing = GET_BOOL(eventing, NULL);
    bool dm_eventing_enabled = GETP_BOOL(&parser->config, AMXRT_COPT_EVENTS);
 
    amxc_string_init(&include, 0);
 
    if((odl_dir == NULL) || (*odl_dir == 0)) {
        amxrt_print_message("Load - No odl directory specified");
        goto exit;
    }
 
    amxc_var_set(bool, eventing, dm_eventing_enabled);
    amxp_sigmngr_enable(&dm->sigmngr, dm_eventing_enabled);
 
    amxc_string_setf(&include, "include \"%s\";", odl_dir);
    status = amxo_parser_parse_string(parser, amxc_string_get(&include, 0), root);
    if(status == 0) {
        is_empty_dir = amxp_dir_is_empty(odl_dir);
    }
    if((status != 0) || is_empty_dir) {
        const char* def = GETP_CHAR(&parser->config, AMXRT_COPT_DEFAULTS);
        amxrt_print_message("Load - Failed to load %s or directory is empty", odl_dir);
        when_true((def == NULL) || (*def == 0), exit);
        amxrt_print_message("Load - Try to load default from %s", def);
        amxc_string_setf(&include, "include \"%s\";", def);
        status = amxo_parser_parse_string(parser, amxc_string_get(&include, 0), root);
        if(status != 0) {
            amxrt_print_message("Load - Failed to load defaults from %s", def);
        }
    }
 
exit:
    amxp_sigmngr_enable(&dm->sigmngr, true);
    amxc_var_set(bool, eventing, orig_eventing);
    free(odl_dir);
    amxc_string_clean(&include);
    return status;
}

amxrt_dm_save()

static int amxrt_dm_save ( amxd_dm_t *  dm, amxo_parser_t *  parser  )

static

Definition at line 181 of file amxrt_save_load.c.

                                                               {
    int status = 0;
    amxc_var_t* paths = amxrt_get_save_objects(parser);
    char* odl_dir = amxrt_get_directory(parser);
    uint32_t index = 0;
    amxc_string_t file_name;
 
    amxc_string_init(&file_name, 0);
 
    when_true_status((odl_dir == NULL) || (*odl_dir == 0), exit,
                     amxrt_print_error("Export - No odl directory specified"));
 
    if(paths == NULL) {
        amxd_object_t* root = amxd_dm_get_root(dm);
        const char* name = GETP_CHAR(&parser->config, AMXRT_COPT_NAME);
        when_null_status(name, exit, status = -1);
        status = amxrt_dm_save_object(parser, root, odl_dir, name);
        goto exit;
    }
 
    when_true(amxc_var_type_of(paths) != AMXC_VAR_ID_LIST, exit);
 
    amxc_var_for_each(path, paths) {
        const char* op = amxc_var_constcast(cstring_t, path);
        amxd_object_t* obj = NULL;
        amxc_string_setf(&file_name, "%2.2d_%s", index, op);
        amxc_string_trimr(&file_name, isdot);
        obj = amxd_dm_findf(dm, "%s", op);
        if(obj == NULL) {
            amxrt_print_error("Export - Object %s not found", op);
            continue;
        }
        status = amxrt_dm_save_object(parser, obj, odl_dir, amxc_string_get(&file_name, 0));
        index++;
    }
 
exit:
    amxc_string_clean(&file_name);
    free(odl_dir);
    return status;
}

amxrt_dm_save_object()

static int amxrt_dm_save_object ( amxo_parser_t *  parser, amxd_object_t *  object, const char *  dir, const char *  name  )

static

Definition at line 163 of file amxrt_save_load.c.

                                                  {
    int status = 0;
    amxc_string_t file;
    amxc_string_init(&file, 0);
 
    amxc_string_setf(&file, "%s/%s.odl", dir, name);
    status = amxo_parser_save_object(parser, amxc_string_get(&file, 0), object, false);
    when_failed_status(status, exit,
                       amxrt_print_error("Export - Failed to write %s file", amxc_string_get(&file, 0)));
 
exit:
    amxc_string_clean(&file);
    return status;
}

amxrt_get_directory()

static char* amxrt_get_directory ( amxo_parser_t *  parser )

static

Definition at line 99 of file amxrt_save_load.c.

                                                        {
    amxc_string_t dir;
    const char* odl_dir = GETP_CHAR(&parser->config, AMXRT_COPT_DIRECTORY);
    char* resolved_dir = NULL;
 
    amxc_string_init(&dir, 0);
    if(odl_dir == NULL) {
        odl_dir = GETP_CHAR(&parser->config, AMXRT_COPT_STORAGE_DIR);
    }
    if(odl_dir != NULL) {
        amxc_string_setf(&dir, "%s", odl_dir);
        amxc_string_resolve(&dir, &parser->config);
        resolved_dir = amxc_string_take_buffer(&dir);
    }
 
    return resolved_dir;
}

amxrt_get_save_objects()

static amxc_var_t* amxrt_get_save_objects ( amxo_parser_t *  parser )

static

Definition at line 81 of file amxrt_save_load.c.

                                                                 {
    amxc_var_t* paths = GETP_ARG(&parser->config, AMXRT_COPT_OBJECTS);
 
    when_null(paths, exit);
 
    if(amxc_var_type_of(paths) == AMXC_VAR_ID_CSTRING) {
        amxc_var_cast(paths, AMXC_VAR_ID_CSV_STRING);
        amxc_var_cast(paths, AMXC_VAR_ID_LIST);
    } else if(amxc_var_type_of(paths) == AMXC_VAR_ID_CSV_STRING) {
        amxc_var_cast(paths, AMXC_VAR_ID_LIST);
    } else if(amxc_var_type_of(paths) != AMXC_VAR_ID_LIST) {
        amxrt_print_error("Export - Export objects specified in wrong format");
    }
 
exit:
    return paths;
}

amxrt_is_odl_storage_enabled()

static bool amxrt_is_odl_storage_enabled ( amxo_parser_t *  parser )

static

Definition at line 374 of file amxrt_save_load.c.

                                                                {
    bool retval = true;
    const char* storage_type = GETP_CHAR(&parser->config, AMXRT_COPT_STORAGE);
 
    when_true((storage_type == NULL) || (*storage_type == 0), exit);
    when_true(strcmp(storage_type, "odl") == 0, exit);
    retval = false;
 
exit:
    return retval;
}

amxrt_monitor_changes()

static void amxrt_monitor_changes ( UNUSED const char *const  sig_name, UNUSED const amxc_var_t *const  data, UNUSED void *const  priv  )

static

Definition at line 386 of file amxrt_save_load.c.

                                                           {
    amxp_slot_disconnect_all(amxrt_monitor_changes);
    amxrt_save_subscribe(storage.dm);
}

amxrt_odl_save_load_cleanup()

static void amxrt_odl_save_load_cleanup ( amxd_dm_t *  dm, amxo_parser_t *  parser  )

static

Definition at line 424 of file amxrt_save_load.c.

                                                                              {
    amxp_timer_delete(&storage.save_timer);
    amxp_slot_disconnect_all(amxrt_save_changed);
    amxp_slot_disconnect_all(amxrt_monitor_changes);
    if(GETP_BOOL(&parser->config, AMXRT_COPT_SAVE)) {
        amxrt_dm_save(dm, parser);
    }
    amxc_llist_clean(&storage.save_paths, amxc_string_list_it_free);
    storage.dm = NULL;
    storage.parser = NULL;
}

amxrt_odl_save_load_init()

static int amxrt_odl_save_load_init ( amxd_dm_t *  dm, amxo_parser_t *  parser  )

static

Definition at line 393 of file amxrt_save_load.c.

                                                                          {
    char* directory = amxrt_get_directory(parser);
    int retval = 0;
 
    storage.dm = dm;
    storage.parser = parser;
    amxc_llist_init(&storage.save_paths);
 
    if(amxp_dir_make(directory, 0777) != 0) {
        amxrt_print_error("Failed to create directory %s", directory);
    }
 
    if(GETP_BOOL(&parser->config, AMXRT_COPT_LOAD)) {
        retval = amxrt_dm_load(dm, parser);
        if(retval != 0) {
            // turn off saving and exit
            amxc_var_t* save_opt = GETP_ARG(&parser->config, AMXRT_COPT_SAVE);
            amxc_var_set(bool, save_opt, false);
            goto exit;
        }
    }
 
    if(GETP_BOOL(&parser->config, AMXRT_COPT_ON_CHANGED)) {
        amxp_slot_connect(&dm->sigmngr, "app:start", NULL, amxrt_monitor_changes, NULL);
    }
 
exit:
    free(directory);
    return retval;
}

amxrt_save_add_path()

static void amxrt_save_add_path ( const char *  path )

static

Definition at line 253 of file amxrt_save_load.c.

                                                  {
    amxc_string_t* save_path = NULL;
    bool is_set = false;
 
    amxc_llist_iterate(it, (&storage.save_paths)) {
        amxc_string_t* set_path = amxc_string_from_llist_it(it);
        if(strcmp(path, amxc_string_get(set_path, 0)) == 0) {
            is_set = true;
            break;
        }
    }
    if(!is_set) {
        amxc_string_new(&save_path, 0);
        amxc_string_setf(save_path, "%s", path);
        amxc_llist_append(&storage.save_paths, &save_path->it);
    }
}

amxrt_save_changed()

static void amxrt_save_changed ( const char *const  sig_name, const amxc_var_t *const  data, UNUSED void *const  priv  )

static

Definition at line 321 of file amxrt_save_load.c.

                                                        {
    amxd_object_t* object = amxd_dm_signal_get_object(storage.dm, data);
    amxo_parser_t* parser = storage.parser;
    uint32_t save_delay = amxc_var_dyncast(uint32_t, GETP_ARG(&parser->config, AMXRT_COPT_DELAY));
    bool save_needed = false;
 
    if((strcmp(sig_name, "dm:instance-added") == 0) ||
       (strcmp(sig_name, "dm:instance-removed") == 0)) {
        uint32_t index = GET_UINT32(data, "index");
        amxd_object_t* instance = amxd_object_get_instance(object, NULL, index);
        object = instance != NULL? instance: object;
        if(amxd_object_is_attr_set(object, amxd_oattr_persistent)) {
            save_needed = amxrt_save_check_tree(object);
        }
    } else if(strcmp(sig_name, "dm:object-changed") == 0) {
        if(amxd_object_is_attr_set(object, amxd_oattr_persistent)) {
            save_needed = amxrt_save_check_tree(object);
            if(save_needed) {
                save_needed = amxrt_save_check_persistent_params(object, data);
            }
        }
    } else {
        if(amxd_object_is_attr_set(object, amxd_oattr_persistent)) {
            save_needed = amxrt_save_check_tree(object);
        }
    }
 
    if(save_needed) {
        save_needed = amxrt_save_check_paths(parser, object);
 
    }
    if(save_needed) {
        if(storage.save_timer == NULL) {
            amxp_timer_new(&storage.save_timer, amxrt_timed_save, NULL);
            save_delay = amxc_var_dyncast(uint32_t, GETP_ARG(&parser->config, AMXRT_COPT_INIT_DELAY));
            save_delay = save_delay == 0 ? 30000 : save_delay;
        }
 
        if((amxp_timer_remaining_time(storage.save_timer) < save_delay) ||
           (amxp_timer_get_state(storage.save_timer) != amxp_timer_running)) {
            amxp_timer_start(storage.save_timer, save_delay == 0 ? 500 : save_delay);
        }
    }
}

amxrt_save_check_paths()

static bool amxrt_save_check_paths ( amxo_parser_t *  parser, amxd_object_t *  object  )

static

Definition at line 271 of file amxrt_save_load.c.

                                                                                 {
    bool save_needed = false;
    char* obj_path = amxd_object_get_path(object, AMXD_OBJECT_INDEXED);
 
    amxc_var_t* paths = amxrt_get_save_objects(parser);
    when_null_status(paths, exit, save_needed = true);
 
    when_true(amxc_var_type_of(paths) != AMXC_VAR_ID_LIST, exit);
 
    amxc_var_for_each(path, paths) {
        const char* str_path = amxc_var_constcast(cstring_t, path);
        int len = strlen(str_path);
        if(strncmp(str_path, obj_path, len) == 0) {
            amxrt_save_add_path(str_path);
            save_needed = true;
            break;
        }
    }
 
exit:
    free(obj_path);
    return save_needed;
}

amxrt_save_check_persistent_params()

static bool amxrt_save_check_persistent_params ( amxd_object_t *  object, const amxc_var_t *  data  )

static

Definition at line 236 of file amxrt_save_load.c.

                                                                       {
    bool persistent = false;
    amxc_var_t* params = GET_ARG(data, "parameters");
 
    amxc_var_for_each(param, params) {
        const char* name = amxc_var_key(param);
        amxd_param_t* param_def = amxd_object_get_param_def(object, name);
        persistent |= amxd_param_is_attr_set(param_def, amxd_pattr_persistent);
        if(persistent) {
            break;
        }
    }
 
    return persistent;
}

amxrt_save_check_tree()

static bool amxrt_save_check_tree ( amxd_object_t *  object )

static

Definition at line 223 of file amxrt_save_load.c.

                                                         {
    bool persistent = true;
    amxd_object_t* parent = amxd_object_get_parent(object);
 
    while(amxd_object_get_type(parent) != amxd_object_root &&
          persistent) {
        persistent &= amxd_object_is_attr_set(object, amxd_oattr_persistent);
        parent = amxd_object_get_parent(parent);
    }
 
    return persistent;
}

amxrt_save_subscribe()

static void amxrt_save_subscribe ( amxd_dm_t *  dm )

static

Definition at line 368 of file amxrt_save_load.c.

                                                {
    amxp_slot_connect(&dm->sigmngr, "dm:object-changed", NULL, amxrt_save_changed, NULL);
    amxp_slot_connect(&dm->sigmngr, "dm:instance-added", NULL, amxrt_save_changed, NULL);
    amxp_slot_connect(&dm->sigmngr, "dm:instance-removed", NULL, amxrt_save_changed, NULL);
}

amxrt_timed_save()

static void amxrt_timed_save ( UNUSED amxp_timer_t *const  timer, UNUSED void *  data  )

static

Definition at line 295 of file amxrt_save_load.c.

                                                {
    if(amxc_llist_is_empty(&storage.save_paths)) {
        amxrt_dm_save(storage.dm, storage.parser);
    } else {
        amxd_object_t* object = NULL;
        char* dir = amxrt_get_directory(storage.parser);
        uint32_t index = 0;
        amxc_string_t file_name;
        amxc_string_init(&file_name, 0);
 
        amxc_llist_for_each(it, (&storage.save_paths)) {
            amxc_string_t* path = amxc_string_from_llist_it(it);
            amxc_string_setf(&file_name, "%2.2d_%s", index, amxc_string_get(path, 0));
            amxc_string_trimr(&file_name, isdot);
 
            object = amxd_dm_findf(storage.dm, "%s", amxc_string_get(path, 0));
            amxrt_dm_save_object(storage.parser, object, dir, amxc_string_get(&file_name, 0));
            amxc_string_delete(&path);
        }
 
        amxc_string_clean(&file_name);
        free(dir);
    }
}

isdot()

static int isdot ( int  c )

static

Definition at line 73 of file amxrt_save_load.c.

                        {
    if(c == '.') {
        return 1;
    }
 
    return 0;
}

Variable Documentation

storage

odl_storage_t storage

static

Definition at line 71 of file amxrt_save_load.c.