Парсинг YAML к значениям с libyaml в C
Я пытаюсь проанализировать файл YAML с помощью C на сервере Linux (это модификация существующего приложения, поэтому изменение языка не вариант).
Я прочитал учебник по адресу http://wpsoftware.net/andrew/pages/libyaml.html и в вики libyaml.
Я хочу переместить конфигурацию базы данных этого приложения из файла заголовка в YAML, чтобы я мог скомпилировать и настроить его как отдельные шаги, что позволяет мне использовать Chef для управления конфигурацией.
Вот ямл:
---
db_server: "localhost"
db_password: "wibble"
db_username: "test"
national_rail_username: test
national_rail_password: wibble
И то, что я хочу сделать, это перебрать файл и установить переменные на основе имени ключа.
Псевдо-код выглядит следующим образом:
config = YAML::load("file.yaml")
DBUSER = config['db_username']
DBPASS = config['db_password']
DBSERVER = config['db_server']
NATIONAL_RAIL_USERNAME = config['national_rail_username']
NATIONAL_RAIL_PASSWORD = config['national_rail_password']
(если вышеприведенное выглядит немного как ruby /python, это потому, что я привык использовать эти языки!)
Мне удалось заставить тестовую установку работать с использованием YAML-cpp, затем я понял, что три часа лаю неправильное дерево, потому что основное приложение написано на C, а не на C++.
РЕДАКТИРОВАТЬ: Вот код, который я до сих пор. Это вырезка и вставка из учебного веб-сайта выше, однако я не верю, что это правильный подход, и он, по-видимому, не дает мне способа выделить "ключи" YAML для переменных в коде C.
#include <stdio.h>
#include <yaml.h>
int main(void)
{
FILE *fh = fopen("config.yaml", "r");
yaml_parser_t parser;
yaml_token_t token; /* new variable */
/* Initialize parser */
if(!yaml_parser_initialize(&parser))
fputs("Failed to initialize parser!\n", stderr);
if(fh == NULL)
fputs("Failed to open file!\n", stderr);
/* Set input file */
yaml_parser_set_input_file(&parser, fh);
/* BEGIN new code */
do {
yaml_parser_scan(&parser, &token);
switch(token.type)
{
/* Stream start/end */
case YAML_STREAM_START_TOKEN: puts("STREAM START"); break;
case YAML_STREAM_END_TOKEN: puts("STREAM END"); break;
/* Token types (read before actual token) */
case YAML_KEY_TOKEN: printf("(Key token) "); break;
case YAML_VALUE_TOKEN: printf("(Value token) "); break;
/* Block delimeters */
case YAML_BLOCK_SEQUENCE_START_TOKEN: puts("<b>Start Block (Sequence)</b>"); break;
case YAML_BLOCK_ENTRY_TOKEN: puts("<b>Start Block (Entry)</b>"); break;
case YAML_BLOCK_END_TOKEN: puts("<b>End block</b>"); break;
/* Data */
case YAML_BLOCK_MAPPING_START_TOKEN: puts("[Block mapping]"); break;
case YAML_SCALAR_TOKEN: printf("scalar %s \n", token.data.scalar.value); break;
/* Others */
default:
printf("Got token of type %d\n", token.type);
}
if(token.type != YAML_STREAM_END_TOKEN)
yaml_token_delete(&token);
} while(token.type != YAML_STREAM_END_TOKEN);
yaml_token_delete(&token);
/* END new code */
/* Cleanup */
yaml_parser_delete(&parser);
fclose(fh);
return 0;
}
2 ответа
Я взял кусок примера кода из этого урока и взломал его. Будьте бдительны, прошло время с тех пор, как я написал C в последний раз!
Я использовал токен API, API событий на самом деле выглядит проще.
#include <stdio.h>
#include <yaml.h>
typedef struct Conf {
char* db_server;
char* db_pass;
char* db_user;
char* rail_user;
char* rail_pass;
}
Conf* readConf(char* filename) {
FILE* fh = fopen(filename, "r");
yaml_parser_t parser;
yaml_token_t token;
Conf* conf = malloc(sizeof(Conf));
if (!yaml_parser_initialize(&parser))
fputs("Failed to initialize parser!\n", stderr);
if (fh == NULL)
fputs("Failed to open file!\n", stderr);
yaml_parser_set_input_file(&parser, fh);
do {
/* As this is an example, I'll just use:
* state = 0 = expect key
* state = 1 = expect value
*/
int state = 0;
char** datap;
char* tk;
yaml_parser_scan(&parser, &token);
switch(token.type)
{
case YAML_KEY_TOKEN: state = 0; break;
case YAML_VALUE_TOKEN: state = 1; break;
case YAML_SCALAR_TOKEN:
tk = token.data.scalar.value;
if (state == 0) {
/* It's safe to not use strncmp as one string is a literal */
if (!strcmp(tk, "db_server")) {
datap = &conf.db_server;
} else if (!strcmp(tk, "db_password")) {
datap = &conf.db_pass;
} else if (!strcmp(tk, "db_username")) {
datap = &conf.db_user;
} else if (!strcmp(tk, "national_rail_username")) {
datap = &conf.rail_user;
} else if (!strcmp(tk, "national_rail_password")) {
datap = &conf.rail_pass;
} else {
printf("Unrecognised key: %s\n", tk);
}
} else {
*datap = strdup(tk);
}
break;
default: break;
}
if (token.type != YAML_STREAM_END_TOKEN)
yaml_token_delete(&token);
} while (token.type != YAML_STREAM_END_TOKEN);
yaml_token_delete(&token);
yaml_parser_delete(&parser);
fclose(fh);
return conf;
}
Я думаю, что в LXS ответ должен быть
int state = 0;
char** datap;
char* tk;
перед циклом do-while и ключом &conf-> вместо &conf.key.
есть код, который работает для меня:
#include <stdio.h>
#include <string.h>
#include <yaml.h>
typedef struct Conf {
char* db_server;
char* db_pass;
char* db_user;
char* rail_user;
char* rail_pass;
}
Conf* readConf(char* filename) {
FILE* fh = fopen(filename, "r");
yaml_parser_t parser;
yaml_token_t token;
Conf* conf = malloc(sizeof(Conf));
if (!yaml_parser_initialize(&parser))
fputs("Failed to initialize parser!\n", stderr);
if (fh == NULL)
fputs("Failed to open file!\n", stderr);
yaml_parser_set_input_file(&parser, fh);
/* As this is an example, I'll just use:
* state = 0 = expect key
* state = 1 = expect value
*/
int state = 0;
char** datap;
char* tk;
do {
yaml_parser_scan(&parser, &token);
switch(token.type) {
case YAML_KEY_TOKEN: state = 0; break;
case YAML_VALUE_TOKEN: state = 1; break;
case YAML_SCALAR_TOKEN:
tk = token.data.scalar.value;
if (state == 0) {
/* It's safe to not use strncmp as
one string is a literal */
if (!strcmp(tk, "db_server")) {
datap = &conf->db_server;
} else if (!strcmp(tk, "db_password")) {
datap = &conf->db_pass;
} else if (!strcmp(tk, "db_username")) {
datap = &conf->db_user;
} else if (!strcmp(tk, "national_rail_username")) {
datap = &conf->rail_user;
} else if (!strcmp(tk, "national_rail_password")) {
datap = &conf->rail_pass;
} else {
printf("Unrecognised key: %s\n", tk);
}
} else {
*datap = strdup(tk);
}
break;
default: break;
}
if (token.type != YAML_STREAM_END_TOKEN)
yaml_token_delete(&token);
} while (token.type != YAML_STREAM_END_TOKEN);
yaml_token_delete(&token);
yaml_parser_delete(&parser);
fclose(fh);
return conf;
}