implement RestClient::init() and RestClient::disable()

mrtazz · mrtazz · commit b88ab3b9e7bf · 2016-01-26T07:31:58.000-05:00
diff --git a/include/restclient-cpp/restclient.h b/include/restclient-cpp/restclient.h
@@ -41,8 +41,8 @@ typedef struct {
   HeaderFields headers;
 } Response;
 
-// init function
-void init();
+// init and disable functions
+int init();
 void disable();
 
 /**
diff --git a/source/restclient.cc b/source/restclient.cc
@@ -12,10 +12,32 @@
  */
 
 #include "restclient-cpp/restclient.h"
-#include "restclient-cpp/version.h"
 
+#include <curl/curl.h>
+
+#include "restclient-cpp/version.h"
 #include "restclient-cpp/connection.h"
 
+/**
+ * @brief global init function. Call this before you start any threads.
+ */
+int RestClient::init() {
+  CURLcode res = curl_global_init(CURL_GLOBAL_ALL);
+  if (res == CURLE_OK) {
+    return 0;
+  } else {
+    return 1;
+  }
+}
+
+/**
+ * @brief global disable function. Call this before you terminate your
+ * program.
+ */
+void RestClient::disable() {
+  curl_global_cleanup();
+}
+
 /**
  * @brief HTTP GET method
  *
diff --git a/test/test_restclient.cc b/test/test_restclient.cc
@@ -17,10 +17,12 @@ class RestClientTest : public ::testing::Test
 
     virtual void SetUp()
     {
+      RestClient::init();
     }
 
     virtual void TearDown()
     {
+      RestClient::disable();
     }
 
 };

Original file line number	Diff line number	Diff line change
`@@ -17,10 +17,12 @@ class RestClientTest : public ::testing::Test`
`17`	`17`
`18`	`18`	`virtual void SetUp()`
`19`	`19`	`{`
	`20`	`+ RestClient::init();`
`20`	`21`	`}`
`21`	`22`
`22`	`23`	`virtual void TearDown()`
`23`	`24`	`{`
	`25`	`+ RestClient::disable();`
`24`	`26`	`}`
`25`	`27`
`26`	`28`	`};`