From c4b18d0c7ea8b487606555513a25c32015b802e8 Mon Sep 17 00:00:00 2001
From: quicksilver <zhifeng.zhang@zilliz.com>
Date: Thu, 1 Aug 2019 22:44:34 +0800
Subject: [PATCH] add snappy cache

Former-commit-id: 73b5ea17d939b406564233fd3bfa19d9b46e43c1
---
 cpp/cmake/ThirdPartyPackages.cmake | 68 ++++++++++++++++++++++--------
 1 file changed, 51 insertions(+), 17 deletions(-)

diff --git a/cpp/cmake/ThirdPartyPackages.cmake b/cpp/cmake/ThirdPartyPackages.cmake
index 8df5065387..52eb2ceda6 100644
--- a/cpp/cmake/ThirdPartyPackages.cmake
+++ b/cpp/cmake/ThirdPartyPackages.cmake
@@ -1563,6 +1563,7 @@ endif()
 macro(build_snappy)
     message(STATUS "Building snappy-${SNAPPY_VERSION} from source")
     set(SNAPPY_PREFIX "${CMAKE_CURRENT_BINARY_DIR}/snappy_ep-prefix/src/snappy_ep")
+    set(SNAPPY_INCLUDE_DIRS "${SNAPPY_PREFIX}/include")
     set(SNAPPY_STATIC_LIB_NAME snappy)
     set(SNAPPY_STATIC_LIB
             "${SNAPPY_PREFIX}/lib/${CMAKE_STATIC_LIBRARY_PREFIX}${SNAPPY_STATIC_LIB_NAME}${CMAKE_STATIC_LIBRARY_SUFFIX}"
@@ -1574,29 +1575,62 @@ macro(build_snappy)
             -DSNAPPY_BUILD_TESTS=OFF
             "-DCMAKE_INSTALL_PREFIX=${SNAPPY_PREFIX}")
 
-    externalproject_add(snappy_ep
-                        ${EP_LOG_OPTIONS}
-                        BUILD_COMMAND
-                        ${MAKE}
-                        ${MAKE_BUILD_ARGS}
-                        BUILD_IN_SOURCE
-                        1
-                        INSTALL_DIR
-                        ${SNAPPY_PREFIX}
-                        URL
-                        ${SNAPPY_SOURCE_URL}
-                        CMAKE_ARGS
-                        ${SNAPPY_CMAKE_ARGS}
-                        BUILD_BYPRODUCTS
-                        "${SNAPPY_STATIC_LIB}")
+    if(USE_JFROG_CACHE STREQUAL "ON")
+        set(SNAPPY_CACHE_PACKAGE_NAME "snappy_${SNAPPY_MD5}.tar.gz")
+        set(SNAPPY_CACHE_URL "${JFROG_ARTFACTORY_CACHE_URL}/${SNAPPY_CACHE_PACKAGE_NAME}")
+        set(SNAPPY_CACHE_PACKAGE_PATH "${THIRDPARTY_PACKAGE_CACHE}/${SNAPPY_CACHE_PACKAGE_NAME}")
 
-    file(MAKE_DIRECTORY "${SNAPPY_PREFIX}/include")
+        file(DOWNLOAD ${SNAPPY_CACHE_URL} ${SNAPPY_CACHE_PACKAGE_PATH} STATUS status)
+        list(GET status 0 status_code)
+        message(STATUS "DOWNLOADING FROM ${SNAPPY_CACHE_URL} TO ${SNAPPY_CACHE_PACKAGE_PATH}. STATUS = ${status_code}")
+        if (NOT status_code EQUAL 0)
+            externalproject_add(snappy_ep
+                    ${EP_LOG_OPTIONS}
+                    BUILD_COMMAND
+                    ${MAKE}
+                    ${MAKE_BUILD_ARGS}
+                    BUILD_IN_SOURCE
+                    1
+                    INSTALL_DIR
+                    ${SNAPPY_PREFIX}
+                    URL
+                    ${SNAPPY_SOURCE_URL}
+                    CMAKE_ARGS
+                    ${SNAPPY_CMAKE_ARGS}
+                    BUILD_BYPRODUCTS
+                    "${SNAPPY_STATIC_LIB}")
+
+            ExternalProject_Create_Cache(snappy_ep ${SNAPPY_CACHE_PACKAGE_PATH} "${CMAKE_CURRENT_BINARY_DIR}/snappy_ep-prefix" ${JFROG_USER_NAME} ${JFROG_PASSWORD} ${SNAPPY_CACHE_URL})
+
+            file(MAKE_DIRECTORY "${SNAPPY_INCLUDE_DIR}")
+        else()
+            ExternalProject_Use_Cache(snappy_ep ${SNAPPY_CACHE_PACKAGE_PATH} ${CMAKE_CURRENT_BINARY_DIR})
+        endif()
+    else()
+        externalproject_add(snappy_ep
+                ${EP_LOG_OPTIONS}
+                BUILD_COMMAND
+                ${MAKE}
+                ${MAKE_BUILD_ARGS}
+                BUILD_IN_SOURCE
+                1
+                INSTALL_DIR
+                ${SNAPPY_PREFIX}
+                URL
+                ${SNAPPY_SOURCE_URL}
+                CMAKE_ARGS
+                ${SNAPPY_CMAKE_ARGS}
+                BUILD_BYPRODUCTS
+                "${SNAPPY_STATIC_LIB}")
+
+        file(MAKE_DIRECTORY "${SNAPPY_INCLUDE_DIR}")
+    endif()
 
     add_library(snappy STATIC IMPORTED)
     set_target_properties(snappy
                         PROPERTIES IMPORTED_LOCATION "${SNAPPY_STATIC_LIB}"
                         INTERFACE_INCLUDE_DIRECTORIES
-                        "${SNAPPY_PREFIX}/include")
+                        "${SNAPPY_INCLUDE_DIRS}")
     add_dependencies(snappy snappy_ep)
 endmacro()