added host as command-line param to samples

Author: skarlsson

postgres_asio/postgres_asio.cpp

@@ -50,7 +50,7 @@ namespace postgres_asio
     std::string connection::last_error() const                 { return PQerrorMessage(_pg_conn); }
     uint32_t    connection::backend_pid() const                { return (uint32_t)PQbackendPID(_pg_conn); }
     int         connection::socket() const                     { return PQsocket(_pg_conn); }
-    void        connection::set_client_encoding(std::string s) { PQsetClientEncoding(_pg_conn, s.c_str()); }
+    bool        connection::set_client_encoding(std::string s) { return (PQsetClientEncoding(_pg_conn, s.c_str()) == 0); }
     void        connection::set_log_id(std::string id)         { _log_id = id; }
     std::string connection::get_log_id() const                 { return _log_id; }
     void        connection::set_warning_timout(uint32_t ms)    { _warn_timeout = ms; }

postgres_asio/postgres_asio.h

@@ -50,7 +50,7 @@ namespace postgres_asio
         std::string last_error() const;
         uint32_t    backend_pid() const;
 
-        void        set_client_encoding(std::string s);
+        bool        set_client_encoding(std::string s);
         void        set_log_id(std::string id);
         std::string get_log_id() const;
         void        set_warning_timout(uint32_t ms);

samples/cursor_sample/cursor_sample.cpp

@@ -24,11 +24,17 @@ void handle_fetch1000(boost::shared_ptr<postgres_asio::connection> connection, s
     connection->exec("FETCH 1000 in mycursor", [connection, total_count](int ec, boost::shared_ptr<PGresult> res) { handle_fetch1000(connection, total_count, ec, std::move(res)); });
 }
 
-int main(int, char *argv[])
+int main(int argc, char *argv[])
 {
+    std::string host;
     boost::log::core::get()->set_filter(boost::log::trivial::severity >= boost::log::trivial::info);
     std::string connect_string = "user=postgres password=postgres dbname=test";
 
+    if (argc > 1)
+    {
+        connect_string += std::string("host=") + argv[1];
+    }
+   
     boost::asio::io_service fg_ios;
     boost::asio::io_service bg_ios;
     std::auto_ptr<boost::asio::io_service::work> work2(new boost::asio::io_service::work(fg_ios));

samples/insert_sample/insert_sample.cpp

@@ -11,10 +11,15 @@
 #include <boost/uuid/uuid_generators.hpp>
 
 
-int main(int, char *argv[])
+int main(int argc, char *argv[])
 {
     boost::log::core::get()->set_filter(boost::log::trivial::severity >= boost::log::trivial::info);
     std::string connect_string = "user=postgres password=postgres dbname=test";
+    
+    if (argc > 1)
+    {
+        connect_string += std::string("host=") + argv[1];
+    }
 
     boost::asio::io_service fg_ios;
     boost::asio::io_service bg_ios;

samples/query_sample/query_sample.cpp

@@ -7,10 +7,15 @@
 #include <boost/log/trivial.hpp>
 #include <boost/log/expressions.hpp>
 
-int main(int, char *argv[])
+int main(int argc, char *argv[])
 {
     boost::log::core::get()->set_filter(boost::log::trivial::severity >= boost::log::trivial::info);
     std::string connect_string = "user=postgres password=postgres dbname=test";
+    
+    if (argc > 1)
+    {
+        connect_string += std::string("host=") + argv[1];
+    }
 
     boost::asio::io_service fg_ios;
     boost::asio::io_service bg_ios;