From 2231b92a89ea560934be92987c27068be398c6fd Mon Sep 17 00:00:00 2001
From: Helin Wang <ustc.harry@gmail.com>
Date: Wed, 12 Jul 2017 23:20:06 +0000
Subject: [PATCH 1/2] go_binary: remove hardcoded library link path

---
 cmake/generic.cmake          | 5 +----
 go/cmd/master/CMakeLists.txt | 2 +-
 2 files changed, 2 insertions(+), 5 deletions(-)

diff --git a/cmake/generic.cmake b/cmake/generic.cmake
index b13400d125..71ee266611 100644
--- a/cmake/generic.cmake
+++ b/cmake/generic.cmake
@@ -320,14 +320,11 @@ function(go_binary TARGET_NAME)
   cmake_parse_arguments(go_binary "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
   string(REPLACE "${PADDLE_GO_PATH}/" "" CMAKE_CURRENT_SOURCE_REL_DIR ${CMAKE_CURRENT_SOURCE_DIR})
 
-  # FIXME: link path
   add_custom_command(OUTPUT ${TARGET_NAME}_timestamp
-    COMMAND env LIBRARY_PATH=${CMAKE_BINARY_DIR}/go/pserver/client/c/:$ENV{LIBRARY_PATH}
-    GOPATH=${GOPATH} ${CMAKE_Go_COMPILER} build
+    COMMAND env GOPATH=${GOPATH} ${CMAKE_Go_COMPILER} build
     -o "${CMAKE_CURRENT_BINARY_DIR}/${TARGET_NAME}"
     "./${CMAKE_CURRENT_SOURCE_REL_DIR}/${go_binary_SRCS}"
     WORKING_DIRECTORY "${PADDLE_IN_GOPATH}/go")
-  # TODO: don't know what ${TARGET_NAME}_link does
   add_custom_target(${TARGET_NAME} ALL DEPENDS go_vendor ${TARGET_NAME}_timestamp ${go_binary_DEPS})
   install(PROGRAMS ${CMAKE_CURRENT_BINARY_DIR}/${TARGET_NAME} DESTINATION bin)
 endfunction(go_binary)
diff --git a/go/cmd/master/CMakeLists.txt b/go/cmd/master/CMakeLists.txt
index 1058ffa86b..9e149967e7 100644
--- a/go/cmd/master/CMakeLists.txt
+++ b/go/cmd/master/CMakeLists.txt
@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-go_binary(master SRC master.go DEPS paddle_go_optimizer)
+go_binary(master SRC master.go)

From b04986da9f57cfba0657194c7e35b7e9229a6676 Mon Sep 17 00:00:00 2001
From: Helin Wang <ustc.harry@gmail.com>
Date: Wed, 12 Jul 2017 23:48:06 +0000
Subject: [PATCH 2/2] add pserver client test

---
 go/CMakeLists.txt                | 1 +
 go/pserver/client/CMakeLists.txt | 3 +++
 2 files changed, 4 insertions(+)
 create mode 100644 go/pserver/client/CMakeLists.txt

diff --git a/go/CMakeLists.txt b/go/CMakeLists.txt
index 18fee46d19..29ce909c64 100644
--- a/go/CMakeLists.txt
+++ b/go/CMakeLists.txt
@@ -19,4 +19,5 @@ add_subdirectory(cmd/master)
 add_subdirectory(master/c)
 add_subdirectory(master)
 add_subdirectory(pserver)
+add_subdirectory(pserver/client)
 add_subdirectory(utils/networkhelper)
diff --git a/go/pserver/client/CMakeLists.txt b/go/pserver/client/CMakeLists.txt
new file mode 100644
index 0000000000..0052bb460b
--- /dev/null
+++ b/go/pserver/client/CMakeLists.txt
@@ -0,0 +1,3 @@
+if(WITH_TESTING)
+  go_test(pserver_client_test DEPS paddle_go_optimizer)
+endif()