Line data Source code
1 : /**
2 : * @file KafkaStream.cpp KafkaStream class implementation
3 : *
4 : * This is part of the DUNE DAQ software, copyright 2020.
5 : * Licensing/copyright details are in the COPYING file that you should have
6 : * received with this code.
7 : */
8 :
9 : #include "KafkaStream.hpp"
10 : #include <boost/crc.hpp>
11 : #include <chrono>
12 : #include <ers/StreamFactory.hpp>
13 : #include <iostream>
14 : #include <string>
15 : #include <vector>
16 :
17 0 : ERS_REGISTER_OUTPUT_STREAM(erskafka::KafkaStream, "erskafka", param)
18 :
19 : /** Constructor that creates a new instance of the erskafka stream with the given configuration.
20 : * \param format elastic search connection string.
21 : */
22 : namespace erskafka {
23 0 : erskafka::KafkaStream::KafkaStream(const std::string& param)
24 : {
25 :
26 0 : if (const char* env_p = std::getenv("DUNEDAQ_PARTITION"))
27 0 : m_partition = env_p;
28 :
29 : // Kafka server settings
30 0 : std::string brokers = param;
31 0 : std::string errstr;
32 :
33 0 : RdKafka::Conf* conf = RdKafka::Conf::create(RdKafka::Conf::CONF_GLOBAL);
34 0 : conf->set("bootstrap.servers", brokers, errstr);
35 0 : if (errstr != "") {
36 0 : std::cout << "Bootstrap server error : " << errstr << '\n';
37 : }
38 0 : if (const char* env_p = std::getenv("DUNEDAQ_APPLICATION_NAME"))
39 0 : conf->set("client.id", env_p, errstr);
40 : else
41 0 : conf->set("client.id", "erskafkaproducerdefault", errstr);
42 0 : if (errstr != "") {
43 0 : std::cout << "Producer configuration error : " << errstr << '\n';
44 : }
45 : // Create producer instance
46 0 : m_producer = RdKafka::Producer::create(conf, errstr);
47 :
48 0 : if (errstr != "") {
49 0 : std::cout << "Producer creation error : " << errstr << '\n';
50 : }
51 0 : }
52 :
53 : void
54 0 : erskafka::KafkaStream::ers_to_json(const ers::Issue& issue, size_t chain, std::vector<nlohmann::json>& j_objs)
55 : {
56 0 : try {
57 0 : nlohmann::json message;
58 0 : message["partition"] = m_partition.c_str();
59 0 : message["issue_name"] = issue.get_class_name();
60 0 : message["message"] = issue.message().c_str();
61 0 : message["severity"] = ers::to_string(issue.severity());
62 0 : message["usecs_since_epoch"] =
63 0 : std::chrono::duration_cast<std::chrono::microseconds>(issue.ptime().time_since_epoch()).count();
64 0 : message["time"] = std::chrono::duration_cast<std::chrono::milliseconds>(issue.ptime().time_since_epoch()).count();
65 :
66 0 : message["qualifiers"] = issue.qualifiers();
67 0 : message["params"] = nlohmann::json::array({});
68 0 : for (auto p : issue.parameters()) {
69 0 : message["params"].push_back(p.first + ": " + p.second);
70 0 : }
71 0 : message["cwd"] = issue.context().cwd();
72 0 : message["file_name"] = issue.context().file_name();
73 0 : message["function_name"] = issue.context().function_name();
74 0 : message["host_name"] = issue.context().host_name();
75 0 : message["package_name"] = issue.context().package_name();
76 0 : message["user_name"] = issue.context().user_name();
77 0 : message["application_name"] = issue.context().application_name();
78 0 : message["user_id"] = issue.context().user_id();
79 0 : message["process_id"] = issue.context().process_id();
80 0 : message["thread_id"] = issue.context().thread_id();
81 0 : message["line_number"] = issue.context().line_number();
82 0 : message["chain"] = chain;
83 :
84 0 : if (issue.cause()) {
85 0 : ers_to_json(*issue.cause(), 2, j_objs);
86 : }
87 0 : j_objs.push_back(message);
88 0 : } catch (const std::exception& e) {
89 0 : std::cout << "Conversion from json error : " << e.what() << '\n';
90 0 : }
91 0 : }
92 :
93 : void
94 0 : erskafka::KafkaStream::kafka_exporter(std::string input, std::string topic)
95 : {
96 0 : try {
97 : // RdKafka::Producer::RK_MSG_COPY to be investigated
98 0 : RdKafka::ErrorCode err = m_producer->produce(topic,
99 : RdKafka::Topic::PARTITION_UA,
100 : RdKafka::Producer::RK_MSG_COPY,
101 0 : const_cast<char*>(input.c_str()),
102 : input.size(),
103 : nullptr,
104 : 0,
105 : 0,
106 : nullptr,
107 : nullptr);
108 0 : if (err != RdKafka::ERR_NO_ERROR) {
109 0 : std::cout << "% Failed to produce to topic " << topic << ": " << RdKafka::err2str(err) << std::endl;
110 : }
111 0 : } catch (const std::exception& e) {
112 0 : std::cout << "Producer error : " << e.what() << '\n';
113 0 : }
114 0 : }
115 :
116 : /** Write method
117 : * \param issue issue to be sent.
118 : */
119 : void
120 0 : erskafka::KafkaStream::write(const ers::Issue& issue)
121 : {
122 0 : try {
123 0 : std::vector<nlohmann::json> j_objs;
124 0 : if (issue.cause() == nullptr) {
125 0 : ers_to_json(issue, 0, j_objs);
126 : } else {
127 0 : ers_to_json(issue, 1, j_objs);
128 : }
129 :
130 : // build a unique hash for a group of nested issues
131 0 : std::ostringstream tmpstream(issue.message());
132 0 : tmpstream << issue.context().process_id() << issue.time_t() << issue.context().application_name()
133 0 : << issue.context().host_name() << rand();
134 0 : std::string tmp = tmpstream.str();
135 0 : boost::crc_32_type crc32;
136 0 : crc32.process_bytes(tmp.c_str(), tmp.length());
137 :
138 0 : for (auto j : j_objs) {
139 0 : j["group_hash"] = crc32.checksum();
140 :
141 0 : erskafka::KafkaStream::kafka_exporter(j.dump(), "erskafka-reporting");
142 0 : }
143 0 : chained().write(issue);
144 0 : } catch (const std::exception& e) {
145 0 : std::cout << "Producer error : " << e.what() << '\n';
146 0 : }
147 0 : }
148 : } // namespace erskafka
|
