| /** |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include "core/conf.h" |
| #include "core/htrace.h" |
| #include "core/htracer.h" |
| #include "core/span.h" |
| #include "sampler/sampler.h" |
| #include "test/span_util.h" |
| #include "test/test.h" |
| #include "util/htable.h" |
| #include "util/log.h" |
| |
| #include <errno.h> |
| #include <inttypes.h> |
| #include <stdarg.h> |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <string.h> |
| |
| #define MAX_SPAN_JSON_LEN 100000 |
| |
| static struct htrace_conf *g_test_conf; |
| |
| static struct htrace_log *g_test_lg; |
| |
| static struct htracer *g_test_tracer; |
| |
| static struct htrace_span *create_span(const char *desc, |
| uint64_t begin_ms, uint64_t end_ms, uint64_t span_id, |
| const char *prid, ...) __attribute__((sentinel)); |
| |
| static int test_span_to_json(const char *expected, |
| struct htrace_span *span) |
| { |
| int buf_len; |
| char *buf; |
| struct htrace_span *rspan = NULL; |
| char err[128]; |
| size_t err_len = sizeof(err); |
| |
| htrace_span_sort_and_dedupe_parents(span); |
| buf_len = span_json_size(span); |
| if ((0 > buf_len) || (buf_len > MAX_SPAN_JSON_LEN)) { |
| fprintf(stderr, "invalid span_json_size %d.\n", buf_len); |
| return EXIT_FAILURE; |
| } |
| buf = malloc(buf_len); |
| EXPECT_NONNULL(buf); |
| span_json_sprintf(span, buf_len, buf); |
| EXPECT_STR_EQ(expected, buf); |
| span_json_parse(buf, &rspan, err, err_len); |
| if (err[0]) { |
| fprintf(stderr, "Failed to parse span json %s: %s\n", buf, err); |
| return EXIT_FAILURE; |
| } |
| EXPECT_NONNULL(rspan); |
| if (span_compare(span, rspan) != 0) { |
| htrace_span_free(rspan); |
| fprintf(stderr, "Failed to parse the span json back into a span " |
| "which was identical to the input. JSON: %s\n", buf); |
| return EXIT_FAILURE; |
| } |
| free(buf); |
| htrace_span_free(rspan); |
| return EXIT_SUCCESS; |
| } |
| |
| static struct htrace_span *create_span(const char *desc, |
| uint64_t begin_ms, uint64_t end_ms, uint64_t span_id, |
| const char *prid, ...) |
| { |
| struct htrace_span* span = NULL; |
| uint64_t *parents, parent; |
| int i, num_parents = 0; |
| va_list ap, ap2; |
| |
| va_start(ap, prid); |
| va_copy(ap2, ap); |
| while (1) { |
| parent = va_arg(ap2, uint64_t); |
| if (!parent) { |
| break; |
| } |
| num_parents++; |
| } while (parent); |
| va_end(ap2); |
| if (num_parents > 0) { |
| parents = xcalloc(sizeof(uint64_t) * num_parents); |
| for (i = 0; i < num_parents; i++) { |
| parents[i] = va_arg(ap, uint64_t); |
| } |
| } |
| va_end(ap); |
| span = htrace_span_alloc(desc, begin_ms, span_id); |
| span->end_ms = end_ms; |
| span->span_id = span_id; |
| span->prid = xstrdup(prid); |
| span->num_parents = num_parents; |
| if (num_parents == 1) { |
| span->parent.single = parents[0]; |
| free(parents); |
| } else if (num_parents > 1) { |
| span->parent.list = parents; |
| } |
| return span; |
| } |
| |
| static int test_spans_to_str(void) |
| { |
| struct htrace_span *span; |
| |
| span = create_span("foo", 123LLU, 456LLU, 789LLU, "span-unit", |
| NULL); |
| EXPECT_INT_ZERO(test_span_to_json( |
| "{\"s\":\"0000000000000315\",\"b\":123,\"e\":456," |
| "\"d\":\"foo\",\"r\":\"span-unit\"" |
| ",\"p\":[]}", span)); |
| htrace_span_free(span); |
| |
| span = create_span("myspan", 34359738368LLU, |
| 34359739368LLU, 68719476736LLU, "span-unit2", |
| 1LLU, 2LLU, 3LLU, NULL); |
| EXPECT_INT_ZERO(test_span_to_json( |
| "{\"s\":\"0000001000000000\",\"b\":34359738368,\"e\":34359739368," |
| "\"d\":\"myspan\",\"r\":\"span-unit2\"," "\"p\":[\"0000000000000001\"," |
| "\"0000000000000002\",\"0000000000000003\"]}", span)); |
| htrace_span_free(span); |
| |
| span = create_span("nextSpan", 14359739368LLU, 18719476736LLU, |
| 0x8000001000000000LLU, "span-unit3", |
| 1LLU, 1LLU, 1LLU, NULL); |
| EXPECT_INT_ZERO(test_span_to_json( |
| "{\"s\":\"8000001000000000\",\"b\":14359739368,\"e\":18719476736," |
| "\"d\":\"nextSpan\",\"r\":\"span-unit3\"," "\"p\":[\"0000000000000001\"]" |
| "}", span)); |
| htrace_span_free(span); |
| return EXIT_SUCCESS; |
| } |
| |
| int main(void) |
| { |
| g_test_conf = htrace_conf_from_strs("", HTRACE_PROCESS_ID"=span-unit"); |
| EXPECT_NONNULL(g_test_conf); |
| g_test_lg = htrace_log_alloc(g_test_conf); |
| EXPECT_NONNULL(g_test_lg); |
| g_test_tracer = htracer_create("span-unit", g_test_conf); |
| EXPECT_NONNULL(g_test_tracer); |
| |
| EXPECT_INT_ZERO(test_spans_to_str()); |
| |
| htracer_free(g_test_tracer); |
| htrace_log_free(g_test_lg); |
| htrace_conf_free(g_test_conf); |
| return EXIT_SUCCESS; |
| } |
| |
| // vim: ts=4:sw=4:tw=79:et |