embulk-input-kafka 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +7 -0
- data/.circleci/config.yml +44 -0
- data/.gitignore +12 -0
- data/LICENSE.txt +21 -0
- data/README.md +53 -0
- data/build.gradle +149 -0
- data/config/checkstyle/checkstyle.xml +128 -0
- data/config/checkstyle/default.xml +108 -0
- data/gradle/wrapper/gradle-wrapper.jar +0 -0
- data/gradle/wrapper/gradle-wrapper.properties +5 -0
- data/gradlew +172 -0
- data/gradlew.bat +84 -0
- data/lib/embulk/input/kafka.rb +3 -0
- data/src/main/java/org/embulk/input/kafka/AbstractKafkaInputColumnVisitor.java +135 -0
- data/src/main/java/org/embulk/input/kafka/AvroFormatColumnVisitor.java +207 -0
- data/src/main/java/org/embulk/input/kafka/JsonFormatColumnVisitor.java +168 -0
- data/src/main/java/org/embulk/input/kafka/KafkaInputPlugin.java +513 -0
- data/src/main/java/org/embulk/input/kafka/KafkaJsonDeserializer.java +38 -0
- data/src/test/avro/ComplexRecordAvro.avsc +18 -0
- data/src/test/avro/SimpleRecordAvro.avsc +11 -0
- data/src/test/java/org/embulk/input/kafka/ComplexRecord.java +75 -0
- data/src/test/java/org/embulk/input/kafka/SimpleRecord.java +39 -0
- data/src/test/java/org/embulk/input/kafka/TestKafkaInputPlugin.java +353 -0
- data/src/test/resources/config_complex.yml +16 -0
- data/src/test/resources/config_complex_avro.yml +17 -0
- data/src/test/resources/config_simple.yml +11 -0
- data/src/test/resources/config_simple_avro.yml +12 -0
- metadata +118 -0
Binary file
|
data/gradlew
ADDED
@@ -0,0 +1,172 @@
|
|
1
|
+
#!/usr/bin/env sh
|
2
|
+
|
3
|
+
##############################################################################
|
4
|
+
##
|
5
|
+
## Gradle start up script for UN*X
|
6
|
+
##
|
7
|
+
##############################################################################
|
8
|
+
|
9
|
+
# Attempt to set APP_HOME
|
10
|
+
# Resolve links: $0 may be a link
|
11
|
+
PRG="$0"
|
12
|
+
# Need this for relative symlinks.
|
13
|
+
while [ -h "$PRG" ] ; do
|
14
|
+
ls=`ls -ld "$PRG"`
|
15
|
+
link=`expr "$ls" : '.*-> \(.*\)$'`
|
16
|
+
if expr "$link" : '/.*' > /dev/null; then
|
17
|
+
PRG="$link"
|
18
|
+
else
|
19
|
+
PRG=`dirname "$PRG"`"/$link"
|
20
|
+
fi
|
21
|
+
done
|
22
|
+
SAVED="`pwd`"
|
23
|
+
cd "`dirname \"$PRG\"`/" >/dev/null
|
24
|
+
APP_HOME="`pwd -P`"
|
25
|
+
cd "$SAVED" >/dev/null
|
26
|
+
|
27
|
+
APP_NAME="Gradle"
|
28
|
+
APP_BASE_NAME=`basename "$0"`
|
29
|
+
|
30
|
+
# Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
|
31
|
+
DEFAULT_JVM_OPTS=""
|
32
|
+
|
33
|
+
# Use the maximum available, or set MAX_FD != -1 to use that value.
|
34
|
+
MAX_FD="maximum"
|
35
|
+
|
36
|
+
warn () {
|
37
|
+
echo "$*"
|
38
|
+
}
|
39
|
+
|
40
|
+
die () {
|
41
|
+
echo
|
42
|
+
echo "$*"
|
43
|
+
echo
|
44
|
+
exit 1
|
45
|
+
}
|
46
|
+
|
47
|
+
# OS specific support (must be 'true' or 'false').
|
48
|
+
cygwin=false
|
49
|
+
msys=false
|
50
|
+
darwin=false
|
51
|
+
nonstop=false
|
52
|
+
case "`uname`" in
|
53
|
+
CYGWIN* )
|
54
|
+
cygwin=true
|
55
|
+
;;
|
56
|
+
Darwin* )
|
57
|
+
darwin=true
|
58
|
+
;;
|
59
|
+
MINGW* )
|
60
|
+
msys=true
|
61
|
+
;;
|
62
|
+
NONSTOP* )
|
63
|
+
nonstop=true
|
64
|
+
;;
|
65
|
+
esac
|
66
|
+
|
67
|
+
CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar
|
68
|
+
|
69
|
+
# Determine the Java command to use to start the JVM.
|
70
|
+
if [ -n "$JAVA_HOME" ] ; then
|
71
|
+
if [ -x "$JAVA_HOME/jre/sh/java" ] ; then
|
72
|
+
# IBM's JDK on AIX uses strange locations for the executables
|
73
|
+
JAVACMD="$JAVA_HOME/jre/sh/java"
|
74
|
+
else
|
75
|
+
JAVACMD="$JAVA_HOME/bin/java"
|
76
|
+
fi
|
77
|
+
if [ ! -x "$JAVACMD" ] ; then
|
78
|
+
die "ERROR: JAVA_HOME is set to an invalid directory: $JAVA_HOME
|
79
|
+
|
80
|
+
Please set the JAVA_HOME variable in your environment to match the
|
81
|
+
location of your Java installation."
|
82
|
+
fi
|
83
|
+
else
|
84
|
+
JAVACMD="java"
|
85
|
+
which java >/dev/null 2>&1 || die "ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
|
86
|
+
|
87
|
+
Please set the JAVA_HOME variable in your environment to match the
|
88
|
+
location of your Java installation."
|
89
|
+
fi
|
90
|
+
|
91
|
+
# Increase the maximum file descriptors if we can.
|
92
|
+
if [ "$cygwin" = "false" -a "$darwin" = "false" -a "$nonstop" = "false" ] ; then
|
93
|
+
MAX_FD_LIMIT=`ulimit -H -n`
|
94
|
+
if [ $? -eq 0 ] ; then
|
95
|
+
if [ "$MAX_FD" = "maximum" -o "$MAX_FD" = "max" ] ; then
|
96
|
+
MAX_FD="$MAX_FD_LIMIT"
|
97
|
+
fi
|
98
|
+
ulimit -n $MAX_FD
|
99
|
+
if [ $? -ne 0 ] ; then
|
100
|
+
warn "Could not set maximum file descriptor limit: $MAX_FD"
|
101
|
+
fi
|
102
|
+
else
|
103
|
+
warn "Could not query maximum file descriptor limit: $MAX_FD_LIMIT"
|
104
|
+
fi
|
105
|
+
fi
|
106
|
+
|
107
|
+
# For Darwin, add options to specify how the application appears in the dock
|
108
|
+
if $darwin; then
|
109
|
+
GRADLE_OPTS="$GRADLE_OPTS \"-Xdock:name=$APP_NAME\" \"-Xdock:icon=$APP_HOME/media/gradle.icns\""
|
110
|
+
fi
|
111
|
+
|
112
|
+
# For Cygwin, switch paths to Windows format before running java
|
113
|
+
if $cygwin ; then
|
114
|
+
APP_HOME=`cygpath --path --mixed "$APP_HOME"`
|
115
|
+
CLASSPATH=`cygpath --path --mixed "$CLASSPATH"`
|
116
|
+
JAVACMD=`cygpath --unix "$JAVACMD"`
|
117
|
+
|
118
|
+
# We build the pattern for arguments to be converted via cygpath
|
119
|
+
ROOTDIRSRAW=`find -L / -maxdepth 1 -mindepth 1 -type d 2>/dev/null`
|
120
|
+
SEP=""
|
121
|
+
for dir in $ROOTDIRSRAW ; do
|
122
|
+
ROOTDIRS="$ROOTDIRS$SEP$dir"
|
123
|
+
SEP="|"
|
124
|
+
done
|
125
|
+
OURCYGPATTERN="(^($ROOTDIRS))"
|
126
|
+
# Add a user-defined pattern to the cygpath arguments
|
127
|
+
if [ "$GRADLE_CYGPATTERN" != "" ] ; then
|
128
|
+
OURCYGPATTERN="$OURCYGPATTERN|($GRADLE_CYGPATTERN)"
|
129
|
+
fi
|
130
|
+
# Now convert the arguments - kludge to limit ourselves to /bin/sh
|
131
|
+
i=0
|
132
|
+
for arg in "$@" ; do
|
133
|
+
CHECK=`echo "$arg"|egrep -c "$OURCYGPATTERN" -`
|
134
|
+
CHECK2=`echo "$arg"|egrep -c "^-"` ### Determine if an option
|
135
|
+
|
136
|
+
if [ $CHECK -ne 0 ] && [ $CHECK2 -eq 0 ] ; then ### Added a condition
|
137
|
+
eval `echo args$i`=`cygpath --path --ignore --mixed "$arg"`
|
138
|
+
else
|
139
|
+
eval `echo args$i`="\"$arg\""
|
140
|
+
fi
|
141
|
+
i=$((i+1))
|
142
|
+
done
|
143
|
+
case $i in
|
144
|
+
(0) set -- ;;
|
145
|
+
(1) set -- "$args0" ;;
|
146
|
+
(2) set -- "$args0" "$args1" ;;
|
147
|
+
(3) set -- "$args0" "$args1" "$args2" ;;
|
148
|
+
(4) set -- "$args0" "$args1" "$args2" "$args3" ;;
|
149
|
+
(5) set -- "$args0" "$args1" "$args2" "$args3" "$args4" ;;
|
150
|
+
(6) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" ;;
|
151
|
+
(7) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" ;;
|
152
|
+
(8) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" ;;
|
153
|
+
(9) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" "$args8" ;;
|
154
|
+
esac
|
155
|
+
fi
|
156
|
+
|
157
|
+
# Escape application args
|
158
|
+
save () {
|
159
|
+
for i do printf %s\\n "$i" | sed "s/'/'\\\\''/g;1s/^/'/;\$s/\$/' \\\\/" ; done
|
160
|
+
echo " "
|
161
|
+
}
|
162
|
+
APP_ARGS=$(save "$@")
|
163
|
+
|
164
|
+
# Collect all arguments for the java command, following the shell quoting and substitution rules
|
165
|
+
eval set -- $DEFAULT_JVM_OPTS $JAVA_OPTS $GRADLE_OPTS "\"-Dorg.gradle.appname=$APP_BASE_NAME\"" -classpath "\"$CLASSPATH\"" org.gradle.wrapper.GradleWrapperMain "$APP_ARGS"
|
166
|
+
|
167
|
+
# by default we should be in the correct project dir, but when run from Finder on Mac, the cwd is wrong
|
168
|
+
if [ "$(uname)" = "Darwin" ] && [ "$HOME" = "$PWD" ]; then
|
169
|
+
cd "$(dirname "$0")"
|
170
|
+
fi
|
171
|
+
|
172
|
+
exec "$JAVACMD" "$@"
|
data/gradlew.bat
ADDED
@@ -0,0 +1,84 @@
|
|
1
|
+
@if "%DEBUG%" == "" @echo off
|
2
|
+
@rem ##########################################################################
|
3
|
+
@rem
|
4
|
+
@rem Gradle startup script for Windows
|
5
|
+
@rem
|
6
|
+
@rem ##########################################################################
|
7
|
+
|
8
|
+
@rem Set local scope for the variables with windows NT shell
|
9
|
+
if "%OS%"=="Windows_NT" setlocal
|
10
|
+
|
11
|
+
set DIRNAME=%~dp0
|
12
|
+
if "%DIRNAME%" == "" set DIRNAME=.
|
13
|
+
set APP_BASE_NAME=%~n0
|
14
|
+
set APP_HOME=%DIRNAME%
|
15
|
+
|
16
|
+
@rem Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
|
17
|
+
set DEFAULT_JVM_OPTS=
|
18
|
+
|
19
|
+
@rem Find java.exe
|
20
|
+
if defined JAVA_HOME goto findJavaFromJavaHome
|
21
|
+
|
22
|
+
set JAVA_EXE=java.exe
|
23
|
+
%JAVA_EXE% -version >NUL 2>&1
|
24
|
+
if "%ERRORLEVEL%" == "0" goto init
|
25
|
+
|
26
|
+
echo.
|
27
|
+
echo ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
|
28
|
+
echo.
|
29
|
+
echo Please set the JAVA_HOME variable in your environment to match the
|
30
|
+
echo location of your Java installation.
|
31
|
+
|
32
|
+
goto fail
|
33
|
+
|
34
|
+
:findJavaFromJavaHome
|
35
|
+
set JAVA_HOME=%JAVA_HOME:"=%
|
36
|
+
set JAVA_EXE=%JAVA_HOME%/bin/java.exe
|
37
|
+
|
38
|
+
if exist "%JAVA_EXE%" goto init
|
39
|
+
|
40
|
+
echo.
|
41
|
+
echo ERROR: JAVA_HOME is set to an invalid directory: %JAVA_HOME%
|
42
|
+
echo.
|
43
|
+
echo Please set the JAVA_HOME variable in your environment to match the
|
44
|
+
echo location of your Java installation.
|
45
|
+
|
46
|
+
goto fail
|
47
|
+
|
48
|
+
:init
|
49
|
+
@rem Get command-line arguments, handling Windows variants
|
50
|
+
|
51
|
+
if not "%OS%" == "Windows_NT" goto win9xME_args
|
52
|
+
|
53
|
+
:win9xME_args
|
54
|
+
@rem Slurp the command line arguments.
|
55
|
+
set CMD_LINE_ARGS=
|
56
|
+
set _SKIP=2
|
57
|
+
|
58
|
+
:win9xME_args_slurp
|
59
|
+
if "x%~1" == "x" goto execute
|
60
|
+
|
61
|
+
set CMD_LINE_ARGS=%*
|
62
|
+
|
63
|
+
:execute
|
64
|
+
@rem Setup the command line
|
65
|
+
|
66
|
+
set CLASSPATH=%APP_HOME%\gradle\wrapper\gradle-wrapper.jar
|
67
|
+
|
68
|
+
@rem Execute Gradle
|
69
|
+
"%JAVA_EXE%" %DEFAULT_JVM_OPTS% %JAVA_OPTS% %GRADLE_OPTS% "-Dorg.gradle.appname=%APP_BASE_NAME%" -classpath "%CLASSPATH%" org.gradle.wrapper.GradleWrapperMain %CMD_LINE_ARGS%
|
70
|
+
|
71
|
+
:end
|
72
|
+
@rem End local scope for the variables with windows NT shell
|
73
|
+
if "%ERRORLEVEL%"=="0" goto mainEnd
|
74
|
+
|
75
|
+
:fail
|
76
|
+
rem Set variable GRADLE_EXIT_CONSOLE if you need the _script_ return code instead of
|
77
|
+
rem the _cmd.exe /c_ return code!
|
78
|
+
if not "" == "%GRADLE_EXIT_CONSOLE%" exit 1
|
79
|
+
exit /b 1
|
80
|
+
|
81
|
+
:mainEnd
|
82
|
+
if "%OS%"=="Windows_NT" endlocal
|
83
|
+
|
84
|
+
:omega
|
@@ -0,0 +1,135 @@
|
|
1
|
+
package org.embulk.input.kafka;
|
2
|
+
|
3
|
+
import org.apache.kafka.clients.consumer.ConsumerRecord;
|
4
|
+
import org.apache.kafka.common.serialization.DoubleDeserializer;
|
5
|
+
import org.apache.kafka.common.serialization.LongDeserializer;
|
6
|
+
import org.apache.kafka.common.serialization.StringDeserializer;
|
7
|
+
import org.apache.kafka.common.utils.Bytes;
|
8
|
+
import org.embulk.input.kafka.KafkaInputPlugin.PluginTask;
|
9
|
+
import org.embulk.spi.Column;
|
10
|
+
import org.embulk.spi.ColumnVisitor;
|
11
|
+
import org.embulk.spi.PageBuilder;
|
12
|
+
import org.embulk.spi.time.Timestamp;
|
13
|
+
import org.embulk.spi.time.TimestampParser;
|
14
|
+
import org.slf4j.Logger;
|
15
|
+
import org.slf4j.LoggerFactory;
|
16
|
+
|
17
|
+
public abstract class AbstractKafkaInputColumnVisitor<V> implements ColumnVisitor
|
18
|
+
{
|
19
|
+
protected final PluginTask task;
|
20
|
+
protected final PageBuilder pageBuilder;
|
21
|
+
protected final TimestampParser[] timestampParsers;
|
22
|
+
|
23
|
+
protected ConsumerRecord<Bytes, V> consumerRecord;
|
24
|
+
protected V recordValue;
|
25
|
+
|
26
|
+
private static final LongDeserializer longDeserializer = new LongDeserializer();
|
27
|
+
private static final StringDeserializer stringDeserializer = new StringDeserializer();
|
28
|
+
private static final DoubleDeserializer doubleDeserializer = new DoubleDeserializer();
|
29
|
+
|
30
|
+
private static final Logger logger = LoggerFactory.getLogger(AbstractKafkaInputColumnVisitor.class);
|
31
|
+
|
32
|
+
public AbstractKafkaInputColumnVisitor(PluginTask task, PageBuilder pageBuilder, TimestampParser[] timestampParsers)
|
33
|
+
{
|
34
|
+
this.task = task;
|
35
|
+
this.pageBuilder = pageBuilder;
|
36
|
+
this.timestampParsers = timestampParsers;
|
37
|
+
}
|
38
|
+
|
39
|
+
public void reset(ConsumerRecord<Bytes, V> consumerRecord)
|
40
|
+
{
|
41
|
+
this.consumerRecord = consumerRecord;
|
42
|
+
this.recordValue = consumerRecord.value();
|
43
|
+
}
|
44
|
+
|
45
|
+
protected boolean isKeyColumn(Column column)
|
46
|
+
{
|
47
|
+
return column.getName().equals(task.getKeyColumnName());
|
48
|
+
}
|
49
|
+
|
50
|
+
protected boolean isPartitionColumn(Column column)
|
51
|
+
{
|
52
|
+
return column.getName().equals(task.getPartitionColumnName());
|
53
|
+
}
|
54
|
+
|
55
|
+
protected void longColumnForKey(Column column)
|
56
|
+
{
|
57
|
+
if (consumerRecord.serializedKeySize() == 0) {
|
58
|
+
pageBuilder.setNull(column);
|
59
|
+
return;
|
60
|
+
}
|
61
|
+
try {
|
62
|
+
long key = longDeserializer.deserialize("dummy", consumerRecord.key().get());
|
63
|
+
pageBuilder.setLong(column, key);
|
64
|
+
}
|
65
|
+
catch (Exception e) {
|
66
|
+
logDeserializeFailure(consumerRecord);
|
67
|
+
pageBuilder.setNull(column);
|
68
|
+
}
|
69
|
+
}
|
70
|
+
|
71
|
+
protected void longColumnForPartition(Column column)
|
72
|
+
{
|
73
|
+
pageBuilder.setLong(column, Integer.valueOf(consumerRecord.partition()).longValue());
|
74
|
+
}
|
75
|
+
|
76
|
+
public void doubleColumnForKey(Column column)
|
77
|
+
{
|
78
|
+
if (consumerRecord.serializedKeySize() == 0) {
|
79
|
+
pageBuilder.setNull(column);
|
80
|
+
return;
|
81
|
+
}
|
82
|
+
try {
|
83
|
+
double key = doubleDeserializer.deserialize("dummy", consumerRecord.key().get());
|
84
|
+
pageBuilder.setDouble(column, key);
|
85
|
+
}
|
86
|
+
catch (Exception e) {
|
87
|
+
logDeserializeFailure(consumerRecord);
|
88
|
+
pageBuilder.setNull(column);
|
89
|
+
}
|
90
|
+
}
|
91
|
+
|
92
|
+
public void stringColumnForKey(Column column)
|
93
|
+
{
|
94
|
+
if (consumerRecord.serializedKeySize() == 0) {
|
95
|
+
pageBuilder.setNull(column);
|
96
|
+
return;
|
97
|
+
}
|
98
|
+
try {
|
99
|
+
String key = stringDeserializer.deserialize("dummy", consumerRecord.key().get());
|
100
|
+
pageBuilder.setString(column, key);
|
101
|
+
}
|
102
|
+
catch (Exception e) {
|
103
|
+
logDeserializeFailure(consumerRecord);
|
104
|
+
pageBuilder.setNull(column);
|
105
|
+
}
|
106
|
+
}
|
107
|
+
|
108
|
+
public void timestampColumnForKey(Column column)
|
109
|
+
{
|
110
|
+
if (consumerRecord.serializedKeySize() == 0) {
|
111
|
+
pageBuilder.setNull(column);
|
112
|
+
return;
|
113
|
+
}
|
114
|
+
try {
|
115
|
+
String key = stringDeserializer.deserialize("dummy", consumerRecord.key().get());
|
116
|
+
try {
|
117
|
+
Timestamp timestamp = timestampParsers[column.getIndex()].parse(key);
|
118
|
+
pageBuilder.setTimestamp(column, timestamp);
|
119
|
+
}
|
120
|
+
catch (Exception e) {
|
121
|
+
logDeserializeFailure(consumerRecord);
|
122
|
+
pageBuilder.setNull(column);
|
123
|
+
}
|
124
|
+
}
|
125
|
+
catch (Exception e) {
|
126
|
+
logDeserializeFailure(consumerRecord);
|
127
|
+
pageBuilder.setNull(column);
|
128
|
+
}
|
129
|
+
}
|
130
|
+
|
131
|
+
private void logDeserializeFailure(ConsumerRecord<Bytes, V> consumerRecord)
|
132
|
+
{
|
133
|
+
logger.warn("Failed to deserialize record key: key={} value={}", consumerRecord.key(), consumerRecord.value());
|
134
|
+
}
|
135
|
+
}
|
@@ -0,0 +1,207 @@
|
|
1
|
+
package org.embulk.input.kafka;
|
2
|
+
|
3
|
+
import com.google.common.collect.ImmutableMap;
|
4
|
+
import org.apache.avro.generic.GenericRecord;
|
5
|
+
import org.embulk.input.kafka.KafkaInputPlugin.PluginTask;
|
6
|
+
import org.embulk.spi.Column;
|
7
|
+
import org.embulk.spi.ColumnVisitor;
|
8
|
+
import org.embulk.spi.PageBuilder;
|
9
|
+
import org.embulk.spi.time.Timestamp;
|
10
|
+
import org.embulk.spi.time.TimestampParser;
|
11
|
+
import org.msgpack.value.Value;
|
12
|
+
import org.msgpack.value.ValueFactory;
|
13
|
+
|
14
|
+
import java.util.List;
|
15
|
+
import java.util.Map;
|
16
|
+
import java.util.stream.Collectors;
|
17
|
+
|
18
|
+
public class AvroFormatColumnVisitor extends AbstractKafkaInputColumnVisitor<Object> implements ColumnVisitor
|
19
|
+
{
|
20
|
+
static class UnsupportedValueTypeException extends RuntimeException
|
21
|
+
{
|
22
|
+
public UnsupportedValueTypeException(String message)
|
23
|
+
{
|
24
|
+
super(message);
|
25
|
+
}
|
26
|
+
}
|
27
|
+
|
28
|
+
public AvroFormatColumnVisitor(PluginTask task, PageBuilder pageBuilder, TimestampParser[] timestampParsers)
|
29
|
+
{
|
30
|
+
super(task, pageBuilder, timestampParsers);
|
31
|
+
}
|
32
|
+
|
33
|
+
private GenericRecord genericRecord()
|
34
|
+
{
|
35
|
+
return (GenericRecord) recordValue;
|
36
|
+
}
|
37
|
+
|
38
|
+
@Override
|
39
|
+
public void booleanColumn(Column column)
|
40
|
+
{
|
41
|
+
Object value = genericRecord().get(column.getName());
|
42
|
+
if (value == null) {
|
43
|
+
pageBuilder.setNull(column);
|
44
|
+
return;
|
45
|
+
}
|
46
|
+
pageBuilder.setBoolean(column, (boolean) value);
|
47
|
+
}
|
48
|
+
|
49
|
+
@Override
|
50
|
+
public void longColumn(Column column)
|
51
|
+
{
|
52
|
+
if (super.isKeyColumn(column)) {
|
53
|
+
super.longColumnForKey(column);
|
54
|
+
return;
|
55
|
+
}
|
56
|
+
|
57
|
+
if (isPartitionColumn(column)) {
|
58
|
+
super.longColumnForPartition(column);
|
59
|
+
return;
|
60
|
+
}
|
61
|
+
|
62
|
+
Object value = genericRecord().get(column.getName());
|
63
|
+
if (value == null) {
|
64
|
+
pageBuilder.setNull(column);
|
65
|
+
return;
|
66
|
+
}
|
67
|
+
|
68
|
+
long longValue;
|
69
|
+
if (value instanceof Long) {
|
70
|
+
longValue = (Long) value;
|
71
|
+
}
|
72
|
+
else if (value instanceof Integer) {
|
73
|
+
longValue = ((Integer) value).longValue();
|
74
|
+
}
|
75
|
+
else if (value instanceof Double) {
|
76
|
+
longValue = ((Double) value).longValue();
|
77
|
+
}
|
78
|
+
else if (value instanceof Float) {
|
79
|
+
longValue = ((Float) value).longValue();
|
80
|
+
}
|
81
|
+
else {
|
82
|
+
throw new RuntimeException(String.format("Cannot cast value to long: column=%s value=%s", column, value));
|
83
|
+
}
|
84
|
+
pageBuilder.setLong(column, longValue);
|
85
|
+
}
|
86
|
+
|
87
|
+
@Override
|
88
|
+
public void doubleColumn(Column column)
|
89
|
+
{
|
90
|
+
if (super.isKeyColumn(column)) {
|
91
|
+
super.doubleColumnForKey(column);
|
92
|
+
return;
|
93
|
+
}
|
94
|
+
|
95
|
+
Object value = genericRecord().get(column.getName());
|
96
|
+
if (value == null) {
|
97
|
+
pageBuilder.setNull(column);
|
98
|
+
return;
|
99
|
+
}
|
100
|
+
|
101
|
+
double doubleValue;
|
102
|
+
if (value instanceof Long) {
|
103
|
+
doubleValue = ((Long) value).doubleValue();
|
104
|
+
}
|
105
|
+
else if (value instanceof Integer) {
|
106
|
+
doubleValue = ((Integer) value).doubleValue();
|
107
|
+
}
|
108
|
+
else if (value instanceof Double) {
|
109
|
+
doubleValue = (Double) value;
|
110
|
+
}
|
111
|
+
else if (value instanceof Float) {
|
112
|
+
doubleValue = ((Float) value).doubleValue();
|
113
|
+
}
|
114
|
+
else {
|
115
|
+
throw new RuntimeException(String.format("Cannot cast value to double: column=%s value=%s", column, value));
|
116
|
+
}
|
117
|
+
pageBuilder.setDouble(column, doubleValue);
|
118
|
+
}
|
119
|
+
|
120
|
+
@Override
|
121
|
+
public void stringColumn(Column column)
|
122
|
+
{
|
123
|
+
if (super.isKeyColumn(column)) {
|
124
|
+
super.stringColumnForKey(column);
|
125
|
+
return;
|
126
|
+
}
|
127
|
+
|
128
|
+
Object value = genericRecord().get(column.getName());
|
129
|
+
if (value == null) {
|
130
|
+
pageBuilder.setNull(column);
|
131
|
+
return;
|
132
|
+
}
|
133
|
+
|
134
|
+
pageBuilder.setString(column, value.toString());
|
135
|
+
}
|
136
|
+
|
137
|
+
@Override
|
138
|
+
public void timestampColumn(Column column)
|
139
|
+
{
|
140
|
+
if (super.isKeyColumn(column)) {
|
141
|
+
super.timestampColumnForKey(column);
|
142
|
+
return;
|
143
|
+
}
|
144
|
+
|
145
|
+
Object value = genericRecord().get(column.getName());
|
146
|
+
if (value == null) {
|
147
|
+
pageBuilder.setNull(column);
|
148
|
+
return;
|
149
|
+
}
|
150
|
+
|
151
|
+
Timestamp timestamp = timestampParsers[column.getIndex()].parse(value.toString());
|
152
|
+
pageBuilder.setTimestamp(column, timestamp);
|
153
|
+
}
|
154
|
+
|
155
|
+
@Override
|
156
|
+
public void jsonColumn(Column column)
|
157
|
+
{
|
158
|
+
Object value = genericRecord().get(column.getName());
|
159
|
+
Value jsonValue = convertAvroToValue(value);
|
160
|
+
pageBuilder.setJson(column, jsonValue);
|
161
|
+
}
|
162
|
+
|
163
|
+
private Value convertAvroToValue(Object avroValue)
|
164
|
+
{
|
165
|
+
if (avroValue instanceof GenericRecord) {
|
166
|
+
ImmutableMap.Builder<Value, Value> builder = ImmutableMap.builder();
|
167
|
+
GenericRecord record = (GenericRecord) avroValue;
|
168
|
+
record.getSchema().getFields().forEach(field -> builder
|
169
|
+
.put(convertAvroToValue(field.name()), convertAvroToValue(record.get(field.name()))));
|
170
|
+
return ValueFactory.newMap(builder.build());
|
171
|
+
}
|
172
|
+
else if (avroValue instanceof List<?>) {
|
173
|
+
List<Value> values = ((List<?>) avroValue).stream().map(this::convertAvroToValue)
|
174
|
+
.collect(Collectors.toList());
|
175
|
+
return ValueFactory.newArray(values);
|
176
|
+
}
|
177
|
+
else if (avroValue instanceof Map<?, ?>) {
|
178
|
+
ImmutableMap.Builder<Value, Value> builder = ImmutableMap.builder();
|
179
|
+
((Map<?, ?>) avroValue).forEach((k, v) -> builder.put(convertAvroToValue(k), convertAvroToValue(v)));
|
180
|
+
return ValueFactory.newMap(builder.build());
|
181
|
+
}
|
182
|
+
else if (avroValue instanceof Boolean) {
|
183
|
+
return ValueFactory.newBoolean((Boolean) avroValue);
|
184
|
+
}
|
185
|
+
else if (avroValue instanceof Integer) {
|
186
|
+
return ValueFactory.newInteger((Integer) avroValue);
|
187
|
+
}
|
188
|
+
else if (avroValue instanceof Long) {
|
189
|
+
return ValueFactory.newInteger((Long) avroValue);
|
190
|
+
}
|
191
|
+
else if (avroValue instanceof Float) {
|
192
|
+
return ValueFactory.newFloat((Float) avroValue);
|
193
|
+
}
|
194
|
+
else if (avroValue instanceof Double) {
|
195
|
+
return ValueFactory.newFloat((Double) avroValue);
|
196
|
+
}
|
197
|
+
else if (avroValue instanceof String) {
|
198
|
+
return ValueFactory.newString((String) avroValue);
|
199
|
+
}
|
200
|
+
else if (avroValue == null) {
|
201
|
+
return ValueFactory.newNil();
|
202
|
+
}
|
203
|
+
else {
|
204
|
+
throw new UnsupportedValueTypeException(String.format("%s is not supported", avroValue));
|
205
|
+
}
|
206
|
+
}
|
207
|
+
}
|