summaryrefslogtreecommitdiffstats
path: root/tests/datetime.rs
blob: 9c182643402d5ebcd71fbe6d258a780d07dcc97c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
#![cfg(all(
    feature = "toml",
    feature = "json",
    feature = "hjson",
    feature = "yaml",
    feature = "ini",
))]

extern crate chrono;
extern crate config_maint;

use chrono::{DateTime, TimeZone, Utc};
use config_maint::*;

fn make() -> Config {
    Config::default()
        .merge(File::from_str(
            r#"
            {
                "json_datetime": "2017-05-10T02:14:53Z"
            }
            "#,
            FileFormat::Json,
        ))
        .unwrap()
        .merge(File::from_str(
            r#"
            yaml_datetime: 2017-06-12T10:58:30Z
            "#,
            FileFormat::Yaml,
        ))
        .unwrap()
        .merge(File::from_str(
            r#"
            toml_datetime = 2017-05-11T14:55:15Z
            "#,
            FileFormat::Toml,
        ))
        .unwrap()
        .merge(File::from_str(
            r#"
            {
                "hjson_datetime": "2017-05-10T02:14:53Z"
            }
            "#,
            FileFormat::Hjson,
        ))
        .unwrap()
        .merge(File::from_str(
            r#"
                ini_datetime = 2017-05-10T02:14:53Z
            "#,
            FileFormat::Ini,
        ))
        .unwrap()
        .clone()
}

#[test]
fn test_datetime_string() {
    let s = make();

    // JSON
    let date: String = s.get("json_datetime").unwrap();

    assert_eq!(&date, "2017-05-10T02:14:53Z");

    // TOML
    let date: String = s.get("toml_datetime").unwrap();

    assert_eq!(&date, "2017-05-11T14:55:15Z");

    // YAML
    let date: String = s.get("yaml_datetime").unwrap();

    assert_eq!(&date, "2017-06-12T10:58:30Z");

    // HJSON
    let date: String = s.get("hjson_datetime").unwrap();

    assert_eq!(&date, "2017-05-10T02:14:53Z");

    // INI
    let date: String = s.get("ini_datetime").unwrap();

    assert_eq!(&date, "2017-05-10T02:14:53Z");
}

#[test]
fn test_datetime() {
    let s = make();

    // JSON
    let date: DateTime<Utc> = s.get("json_datetime").unwrap();

    assert_eq!(date, Utc.ymd(2017, 5, 10).and_hms(2, 14, 53));

    // TOML
    let date: DateTime<Utc> = s.get("toml_datetime").unwrap();

    assert_eq!(date, Utc.ymd(2017, 5, 11).and_hms(14, 55, 15));

    // YAML
    let date: DateTime<Utc> = s.get("yaml_datetime").unwrap();

    assert_eq!(date, Utc.ymd(2017, 6, 12).and_hms(10, 58, 30));

    // HJSON
    let date: DateTime<Utc> = s.get("hjson_datetime").unwrap();

    assert_eq!(date, Utc.ymd(2017, 5, 10).and_hms(2, 14, 53));

    // INI
    let date: DateTime<Utc> = s.get("ini_datetime").unwrap();

    assert_eq!(date, Utc.ymd(2017, 5, 10).and_hms(2, 14, 53));
}
ass="n">end_time; time_t wall_clock_time; bool ml_locked; } v2; } PARSER_USER_OBJECT; typedef struct parser { uint8_t version; // Parser version PARSER_REPERTOIRE repertoire; uint32_t flags; int fd; // Socket size_t line; FILE *fp_input; // Input source e.g. stream FILE *fp_output; // Stream to send commands to plugin #ifdef ENABLE_HTTPS NETDATA_SSL *ssl_output; #endif PARSER_USER_OBJECT user; // User defined structure to hold extra state between calls struct buffered_reader reader; struct { const char *end_keyword; BUFFER *response; void (*action)(struct parser *parser, void *action_data); void *action_data; } defer; struct { DICTIONARY *functions; usec_t smaller_timeout; } inflight; struct { SPINLOCK spinlock; } writer; } PARSER; PARSER *parser_init(struct parser_user_object *user, FILE *fp_input, FILE *fp_output, int fd, PARSER_INPUT_TYPE flags, void *ssl); void parser_init_repertoire(PARSER *parser, PARSER_REPERTOIRE repertoire); void parser_destroy(PARSER *working_parser); void pluginsd_cleanup_v2(PARSER *parser); void inflight_functions_init(PARSER *parser); void pluginsd_keywords_init(PARSER *parser, PARSER_REPERTOIRE repertoire); PARSER_RC parser_execute(PARSER *parser, PARSER_KEYWORD *keyword, char **words, size_t num_words); static inline int find_first_keyword(const char *src, char *dst, int dst_size, bool *isspace_map) { const char *s = src, *keyword_start; while (unlikely(isspace_map[(uint8_t)*s])) s++; keyword_start = s; while (likely(*s && !isspace_map[(uint8_t)*s]) && dst_size > 1) { *dst++ = *s++; dst_size--; } *dst = '\0'; return dst_size == 0 ? 0 : (int) (s - keyword_start); } PARSER_KEYWORD *gperf_lookup_keyword(register const char *str, register size_t len); static inline PARSER_KEYWORD *parser_find_keyword(PARSER *parser, const char *command) { PARSER_KEYWORD *t = gperf_lookup_keyword(command, strlen(command)); if(t && (t->repertoire & parser->repertoire)) return t; return NULL; } static inline int parser_action(PARSER *parser, char *input) { parser->line++; if(unlikely(parser->flags & PARSER_DEFER_UNTIL_KEYWORD)) { char command[100 + 1]; bool has_keyword = find_first_keyword(input, command, 100, isspace_map_pluginsd); if(!has_keyword || strcmp(command, parser->defer.end_keyword) != 0) { if(parser->defer.response) { buffer_strcat(parser->defer.response, input); if(buffer_strlen(parser->defer.response) > PLUGINSD_MAX_DEFERRED_SIZE) { // more than PLUGINSD_MAX_DEFERRED_SIZE of data, // or a bad plugin that did not send the end_keyword internal_error(true, "PLUGINSD: deferred response is too big (%zu bytes). Stopping this plugin.", buffer_strlen(parser->defer.response)); return 1; } } return 0; } else { // call the action parser->defer.action(parser, parser->defer.action_data); // empty everything parser->defer.action = NULL; parser->defer.action_data = NULL; parser->defer.end_keyword = NULL; parser->defer.response = NULL; parser->flags &= ~PARSER_DEFER_UNTIL_KEYWORD; } return 0; } static __thread char *words[PLUGINSD_MAX_WORDS]; size_t num_words = quoted_strings_splitter_pluginsd(input, words, PLUGINSD_MAX_WORDS); const char *command = get_word(words, num_words, 0); if(unlikely(!command)) return 0; PARSER_RC rc; PARSER_KEYWORD *t = parser_find_keyword(parser, command); if(likely(t)) { worker_is_busy(t->worker_job_id); rc = parser_execute(parser, t, words, num_words); // rc = (*t->func)(words, num_words, parser); worker_is_idle(); } else rc = PARSER_RC_ERROR; if(rc == PARSER_RC_ERROR) { BUFFER *wb = buffer_create(PLUGINSD_LINE_MAX, NULL); for(size_t i = 0; i < num_words ;i++) { if(i) buffer_fast_strcat(wb, " ", 1); buffer_fast_strcat(wb, "\"", 1); const char *s = get_word(words, num_words, i); buffer_strcat(wb, s?s:""); buffer_fast_strcat(wb, "\"", 1); } netdata_log_error("PLUGINSD: parser_action('%s') failed on line %zu: { %s } (quotes added to show parsing)", command, parser->line, buffer_tostring(wb)); buffer_free(wb); } return (rc == PARSER_RC_ERROR || rc == PARSER_RC_STOP); } #endif //NETDATA_PLUGINSD_PARSER_H