diff options
author | Loo Rong Jie <loorongjie@gmail.com> | 2018-07-04 11:33:45 +0800 |
---|---|---|
committer | Loo Rong Jie <loorongjie@gmail.com> | 2018-07-20 10:02:45 +0800 |
commit | d4fc213ebbfef8a30113f7b0905138685bcb7592 (patch) | |
tree | ff8b05b3918482f89917d6959126159fa35e911e | |
parent | 77c7b1112210beb3f0752f206bfa519f22aaf5c6 (diff) |
[Bazel/MSVC] Make Kafka builds on Windows
-rw-r--r-- | tensorflow/contrib/kafka/kernels/kafka_dataset_ops.cc | 2 | ||||
-rw-r--r-- | third_party/kafka/BUILD | 36 |
2 files changed, 27 insertions, 11 deletions
diff --git a/tensorflow/contrib/kafka/kernels/kafka_dataset_ops.cc b/tensorflow/contrib/kafka/kernels/kafka_dataset_ops.cc index 2638b25ec4..588e456bd6 100644 --- a/tensorflow/contrib/kafka/kernels/kafka_dataset_ops.cc +++ b/tensorflow/contrib/kafka/kernels/kafka_dataset_ops.cc @@ -15,7 +15,7 @@ limitations under the License. #include "tensorflow/core/framework/dataset.h" -#include "src-cpp/rdkafkacpp.h" +#include "rdkafkacpp.h" namespace tensorflow { diff --git a/third_party/kafka/BUILD b/third_party/kafka/BUILD index 75792b0d87..7a60edb483 100644 --- a/third_party/kafka/BUILD +++ b/third_party/kafka/BUILD @@ -15,6 +15,7 @@ cc_library( "src-cpp/KafkaConsumerImpl.cpp", "src-cpp/MessageImpl.cpp", "src-cpp/MetadataImpl.cpp", + "src-cpp/ProducerImpl.cpp", "src-cpp/QueueImpl.cpp", "src-cpp/RdKafka.cpp", "src-cpp/TopicImpl.cpp", @@ -130,7 +131,15 @@ cc_library( "src/tinycthread.h", "src/xxhash.c", "src/xxhash.h", - ], + ] + select({ + "@org_tensorflow//tensorflow:windows": [ + "src/rdkafka_sasl_win32.c", + "src/rdwin32.h", + "src/regexp.c", + "src/regexp.h", + ], + "//conditions:default": [], + }), hdrs = [ "config.h", "src-cpp/rdkafkacpp.h", @@ -138,15 +147,22 @@ cc_library( "src/lz4.c", "src/snappy_compat.h", ], - copts = [ - "-Iexternal/kafka/src", - "-Iexternal/kafka/src-cpp", - ], - defines = [ - ], - linkopts = [ - "-lpthread", - ], + includes = ["src", "src-cpp"], + defines = ["LIBRDKAFKA_STATICLIB"], + copts = select({ + "@org_tensorflow//tensorflow:windows": [ + "-DWIN32_LEAN_AND_MEAN", + "-DWITHOUT_WIN32_CONFIG", + "-DWITH_ZLIB=1", + "-DWITH_SSL=1", + "-DWITH_SNAPPY=1", + ], + "//conditions:default": [], + }), + linkopts = select({ + "@org_tensorflow//tensorflow:windows": ["-defaultlib:crypt32.lib"], + "//conditions:default": ["-lpthread"], + }), visibility = ["//visibility:public"], deps = [ "@boringssl//:ssl", |