Skip to content

Commit 607f743

Browse files
committed
Issue 170 Upgrade to parquet-cpp 1.3
1 parent 4c8407a commit 607f743

File tree

3 files changed

+13
-10
lines changed

3 files changed

+13
-10
lines changed

parquet/CMakeLists.txt

+4-3
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,13 @@
33
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
44

55
cmake_minimum_required(VERSION 3.5)
6-
project(parquet VERSION 0.0.8 LANGUAGES C CXX)
6+
project(parquet VERSION 0.0.9 LANGUAGES C CXX)
77
set(CPACK_PACKAGE_DESCRIPTION_SUMMARY "Parquet Lua Module")
88

9-
find_package(parquet-cpp 0.0.1 REQUIRED CONFIG)
9+
find_package(parquet-cpp 1.3.1 REQUIRED CONFIG)
1010

1111
set(MODULE_SRCS parquet.cpp parquet.def)
12-
set(CPACK_DEBIAN_PACKAGE_DEPENDS "parquet-cpp (>= 0.0.1), ${PACKAGE_PREFIX}-lpeg (>= 1.0), libc6 (>= 2.14), libgcc1 (>= 1:4.1.1), libstdc++6 (>= 4.1.1)")
12+
set(INSTALL_MODULE_PATH ${INSTALL_IOMODULE_PATH})
13+
set(CPACK_DEBIAN_PACKAGE_DEPENDS "parquet-cpp (>= 1.3.1), ${PACKAGE_PREFIX}-lpeg (>= 1.0)")
1314
include(sandbox_module)
1415
target_link_libraries(parquet ${PARQUET-CPP_LIBRARIES})

parquet/parquet.cpp

+8-6
Original file line numberDiff line numberDiff line change
@@ -11,10 +11,10 @@
1111
#include <string>
1212
#include <vector>
1313

14-
#include <parquet/column/writer.h>
14+
#include <parquet/column_writer.h>
1515
#include <parquet/file/writer.h>
1616
#include <parquet/types.h>
17-
#include <parquet/util/output.h>
17+
#include <arrow/io/file.h>
1818

1919
extern "C"
2020
{
@@ -31,6 +31,7 @@ int luaopen_parquet(lua_State *lua);
3131
#endif
3232

3333
using namespace std;
34+
using FileClass = arrow::io::FileOutputStream;
3435

3536
namespace pq = parquet;
3637

@@ -784,7 +785,8 @@ static int pq_new_writer(lua_State *lua)
784785
pw->w->num_records = 0;
785786
add_columns(pw->w, pw->w->node);
786787

787-
shared_ptr<pq::LocalFileOutputStream> sink(new pq::LocalFileOutputStream(name));
788+
shared_ptr<FileClass> sink;
789+
PARQUET_THROW_NOT_OK(FileClass::Open(name, &sink));
788790
if (t == LUA_TTABLE) {
789791
pw->w->writer = pq::ParquetFileWriter::Open(sink, static_pointer_cast<pq::schema::GroupNode>(ud->n->node),
790792
setup_properties(lua));
@@ -1455,7 +1457,7 @@ static void dissect_field(pq_writer *pw, lua_State *lua, pq_node *n, int16_t r,
14551457
if (n->node->is_group()) {
14561458
if (n->node->is_repeated() && lua_objlen(lua, -1) > 0) { // array of groups
14571459
size_t ol = lua_objlen(lua, -1);
1458-
for (int j = 1; j <= static_cast<int>(ol); ++j) {
1460+
for (size_t j = 1; j <= ol; ++j) {
14591461
lua_rawgeti(lua, -1, j);
14601462
if (lua_type(lua, -1) != LUA_TTABLE) {
14611463
stringstream ss;
@@ -1488,7 +1490,7 @@ static void dissect_field(pq_writer *pw, lua_State *lua, pq_node *n, int16_t r,
14881490
if (ol == 0) {
14891491
add_null(c, r, d);
14901492
} else {
1491-
for (int j = 1; j <= static_cast<int>(ol); ++j) {
1493+
for (size_t j = 1; j <= ol; ++j) {
14921494
lua_rawgeti(lua, -1, j);
14931495
add_value(lua, n, c, r, n->dl);
14941496
lua_pop(lua, 1);
@@ -1563,7 +1565,7 @@ static void dissect_list(pq_writer *pw, lua_State *lua, pq_node *n, int16_t r, i
15631565
static void dissect_tuple(pq_writer *pw, lua_State *lua, pq_node *n, int16_t r, int16_t d)
15641566
{
15651567
size_t len = n->group->fields.size();
1566-
for (int i = 0; i < static_cast<int>(len); ++i) {
1568+
for (size_t i = 0; i < len; ++i) {
15671569
pq_node *cn = n->group->fields[i];
15681570
lua_checkstack(lua, 2);
15691571
lua_rawgeti(lua, -1, i + 1);

parquet/tests/test.lua

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55
require "string"
66
require "parquet"
7-
assert(parquet.version() == "0.0.8", parquet.version())
7+
assert(parquet.version() == "0.0.9", parquet.version())
88
local parser = require "lpeg.parquet"
99
local r1 = {
1010
DocId = 10,

0 commit comments

Comments
 (0)