Protobuf refactoring

[thirdparty/pdns.git] / pdns / dnsdist-protobuf.hh

diff --git a/pdns/dnsdist-protobuf.hh b/pdns/dnsdist-protobuf.hh
index bbac0b20c95bc0ee617e3b192044174b7e1c4b9c..222b6616d43c10f7b58f65a472829614c54080ad 100644 (file)
--- a/pdns/dnsdist-protobuf.hh
+++ b/pdns/dnsdist-protobuf.hh
@@ -1,9 +1,9 @@
-
 #pragma once
-#include "config.h"
 
-#ifdef HAVE_PROTOBUF
-void protobufMessageFromQuestion(const DNSQuestion& dq, std::string& data);
-void protobufMessageFromResponse(const DNSQuestion& dr, std::string& data);
+#include "protobuf.hh"
 
-#endif /* HAVE_PROTOBUF */
+class DNSDistProtoBufMessage: public DNSProtoBufMessage
+{
+public:
+  DNSDistProtoBufMessage(DNSProtoBufMessageType type, const DNSQuestion& dq);
+};