embulk-filter-base58 0.1.0

Sign up to get free protection for your applications and to get access to all the features.
checksums.yaml ADDED
@@ -0,0 +1,7 @@
1
+ ---
2
+ SHA1:
3
+ metadata.gz: 00cfc6ad1057cdbf2dc12e61707eae5b5dbdcb6b
4
+ data.tar.gz: a0a1bbe1f72006006ecbbfd241691ddfaa485364
5
+ SHA512:
6
+ metadata.gz: 5a2b21c12d7381b0ac0263ceaef7da002f65c36081b6f8cd628b715e8907ff37461a1808d01162b8207bb14d6fa3131818b37d4b105f1bc71d86ca66d3a24fe4
7
+ data.tar.gz: a95d96b3422597fa70a357081b67f885f3a9aa0622948a75c4888b859652c4c6f6f484083e641036bf9dc279579c3ff4c55c106ae78b8ba3f618a5834b64d9fb
data/.gitignore ADDED
@@ -0,0 +1,12 @@
1
+ *~
2
+ /pkg/
3
+ /tmp/
4
+ *.gemspec
5
+ .gradle/
6
+ /classpath/
7
+ build/
8
+ .idea
9
+ /.settings/
10
+ /.metadata/
11
+ .classpath
12
+ .project
data/.travis.yml ADDED
@@ -0,0 +1,7 @@
1
+ language: java
2
+ jdk:
3
+ - oraclejdk8
4
+ script:
5
+ - ./gradlew test
6
+ after_success:
7
+ - ./gradlew jacocoTestReport coveralls
data/LICENSE.txt ADDED
@@ -0,0 +1,21 @@
1
+
2
+ MIT License
3
+
4
+ Permission is hereby granted, free of charge, to any person obtaining
5
+ a copy of this software and associated documentation files (the
6
+ "Software"), to deal in the Software without restriction, including
7
+ without limitation the rights to use, copy, modify, merge, publish,
8
+ distribute, sublicense, and/or sell copies of the Software, and to
9
+ permit persons to whom the Software is furnished to do so, subject to
10
+ the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be
13
+ included in all copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
16
+ EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
17
+ MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
18
+ NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
19
+ LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
20
+ OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
21
+ WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
data/README.md ADDED
@@ -0,0 +1,33 @@
1
+ # Base58 filter plugin for Embulk
2
+
3
+ Embulk filter plugin to convert hex to base58 and vice versa.
4
+
5
+ ## Overview
6
+
7
+ * **Plugin type**: filter
8
+
9
+ ## Configuration
10
+
11
+ - **columns**: Columns to encode/decode (array, required)
12
+ - **name**: Name of input column (string, required)
13
+ - **encode**: Whether to encode or decode the value. (boolean, default:`true`)
14
+ - **prefix**: Adds a prefix when encoding, or strips the prefix when decoding. (string, default:`""`)
15
+ - **new_name**: New column name if you want to rename (string, default: `null`)
16
+
17
+ ## Example
18
+
19
+ ```yaml
20
+ filters:
21
+ - type: base58
22
+ columns:
23
+ - { name: _id }
24
+ - { name: account_id, encode: true, prefix: account_, new_name: public_account_id }
25
+ ```
26
+
27
+ ## Build
28
+
29
+ ```
30
+ $ ./gradlew gem # -t to watch change of files and rebuild continuously
31
+ ```
32
+
33
+ *Hat tip to [kamatama41/embulk-filter-hash](https://github.com/kamatama41/embulk-filter-hash)*
data/build.gradle ADDED
@@ -0,0 +1,73 @@
1
+ plugins {
2
+ id "com.jfrog.bintray" version "1.1"
3
+ id "com.github.jruby-gradle.base" version "0.1.5"
4
+ id "java"
5
+ }
6
+ import com.github.jrubygradle.JRubyExec
7
+ repositories {
8
+ mavenCentral()
9
+ jcenter()
10
+ }
11
+ configurations {
12
+ provided
13
+ }
14
+
15
+ version = "0.1.0"
16
+
17
+ dependencies {
18
+ compile "org.embulk:embulk-core:0.8.8"
19
+ provided "org.embulk:embulk-core:0.8.8"
20
+ // compile "YOUR_JAR_DEPENDENCY_GROUP:YOUR_JAR_DEPENDENCY_MODULE:YOUR_JAR_DEPENDENCY_VERSION"
21
+ testCompile "junit:junit:4.+"
22
+ }
23
+
24
+ task classpath(type: Copy, dependsOn: ["jar"]) {
25
+ doFirst { file("classpath").deleteDir() }
26
+ from (configurations.runtime - configurations.provided + files(jar.archivePath))
27
+ into "classpath"
28
+ }
29
+ clean { delete "classpath" }
30
+
31
+ task gem(type: JRubyExec, dependsOn: ["gemspec", "classpath"]) {
32
+ jrubyArgs "-rrubygems/gem_runner", "-eGem::GemRunner.new.run(ARGV)", "build"
33
+ script "${project.name}.gemspec"
34
+ doLast { ant.move(file: "${project.name}-${project.version}.gem", todir: "pkg") }
35
+ }
36
+
37
+ task gemPush(type: JRubyExec, dependsOn: ["gem"]) {
38
+ jrubyArgs "-rrubygems/gem_runner", "-eGem::GemRunner.new.run(ARGV)", "push"
39
+ script "pkg/${project.name}-${project.version}.gem"
40
+ }
41
+
42
+ task "package"(dependsOn: ["gemspec", "classpath"]) << {
43
+ println "> Build succeeded."
44
+ println "> You can run embulk with '-L ${file(".").absolutePath}' argument."
45
+ }
46
+
47
+ task gemspec {
48
+ ext.gemspecFile = file("${project.name}.gemspec")
49
+ inputs.file "build.gradle"
50
+ outputs.file gemspecFile
51
+ doLast { gemspecFile.write($/
52
+ Gem::Specification.new do |spec|
53
+ spec.name = "${project.name}"
54
+ spec.version = "${project.version}"
55
+ spec.authors = ["Kevin Fitzgerald"]
56
+ spec.summary = %[Base58 filter plugin for Embulk]
57
+ spec.description = %[Embulk filter plugin to convert a hex value to and from base58.]
58
+ spec.email = ["kevin@okanjo.com"]
59
+ spec.licenses = ["MIT"]
60
+ spec.homepage = "https://github.com/kfitzgerald/embulk-filter-base58"
61
+
62
+ spec.files = `git ls-files`.split("\n") + Dir["classpath/*.jar"]
63
+ spec.test_files = spec.files.grep(%r"^(test|spec)/")
64
+ spec.require_paths = ["lib"]
65
+
66
+ #spec.add_dependency 'YOUR_GEM_DEPENDENCY', ['~> YOUR_GEM_DEPENDENCY_VERSION']
67
+ spec.add_development_dependency 'bundler', ['~> 1.0']
68
+ spec.add_development_dependency 'rake', ['>= 10.0']
69
+ end
70
+ /$)
71
+ }
72
+ }
73
+ clean { delete "${project.name}.gemspec" }
Binary file
@@ -0,0 +1,6 @@
1
+ #Tue Aug 11 00:26:20 PDT 2015
2
+ distributionBase=GRADLE_USER_HOME
3
+ distributionPath=wrapper/dists
4
+ zipStoreBase=GRADLE_USER_HOME
5
+ zipStorePath=wrapper/dists
6
+ distributionUrl=https\://services.gradle.org/distributions/gradle-2.6-bin.zip
data/gradlew ADDED
@@ -0,0 +1,164 @@
1
+ #!/usr/bin/env bash
2
+
3
+ ##############################################################################
4
+ ##
5
+ ## Gradle start up script for UN*X
6
+ ##
7
+ ##############################################################################
8
+
9
+ # Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
10
+ DEFAULT_JVM_OPTS=""
11
+
12
+ APP_NAME="Gradle"
13
+ APP_BASE_NAME=`basename "$0"`
14
+
15
+ # Use the maximum available, or set MAX_FD != -1 to use that value.
16
+ MAX_FD="maximum"
17
+
18
+ warn ( ) {
19
+ echo "$*"
20
+ }
21
+
22
+ die ( ) {
23
+ echo
24
+ echo "$*"
25
+ echo
26
+ exit 1
27
+ }
28
+
29
+ # OS specific support (must be 'true' or 'false').
30
+ cygwin=false
31
+ msys=false
32
+ darwin=false
33
+ case "`uname`" in
34
+ CYGWIN* )
35
+ cygwin=true
36
+ ;;
37
+ Darwin* )
38
+ darwin=true
39
+ ;;
40
+ MINGW* )
41
+ msys=true
42
+ ;;
43
+ esac
44
+
45
+ # For Cygwin, ensure paths are in UNIX format before anything is touched.
46
+ if $cygwin ; then
47
+ [ -n "$JAVA_HOME" ] && JAVA_HOME=`cygpath --unix "$JAVA_HOME"`
48
+ fi
49
+
50
+ # Attempt to set APP_HOME
51
+ # Resolve links: $0 may be a link
52
+ PRG="$0"
53
+ # Need this for relative symlinks.
54
+ while [ -h "$PRG" ] ; do
55
+ ls=`ls -ld "$PRG"`
56
+ link=`expr "$ls" : '.*-> \(.*\)$'`
57
+ if expr "$link" : '/.*' > /dev/null; then
58
+ PRG="$link"
59
+ else
60
+ PRG=`dirname "$PRG"`"/$link"
61
+ fi
62
+ done
63
+ SAVED="`pwd`"
64
+ cd "`dirname \"$PRG\"`/" >&-
65
+ APP_HOME="`pwd -P`"
66
+ cd "$SAVED" >&-
67
+
68
+ CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar
69
+
70
+ # Determine the Java command to use to start the JVM.
71
+ if [ -n "$JAVA_HOME" ] ; then
72
+ if [ -x "$JAVA_HOME/jre/sh/java" ] ; then
73
+ # IBM's JDK on AIX uses strange locations for the executables
74
+ JAVACMD="$JAVA_HOME/jre/sh/java"
75
+ else
76
+ JAVACMD="$JAVA_HOME/bin/java"
77
+ fi
78
+ if [ ! -x "$JAVACMD" ] ; then
79
+ die "ERROR: JAVA_HOME is set to an invalid directory: $JAVA_HOME
80
+
81
+ Please set the JAVA_HOME variable in your environment to match the
82
+ location of your Java installation."
83
+ fi
84
+ else
85
+ JAVACMD="java"
86
+ which java >/dev/null 2>&1 || die "ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
87
+
88
+ Please set the JAVA_HOME variable in your environment to match the
89
+ location of your Java installation."
90
+ fi
91
+
92
+ # Increase the maximum file descriptors if we can.
93
+ if [ "$cygwin" = "false" -a "$darwin" = "false" ] ; then
94
+ MAX_FD_LIMIT=`ulimit -H -n`
95
+ if [ $? -eq 0 ] ; then
96
+ if [ "$MAX_FD" = "maximum" -o "$MAX_FD" = "max" ] ; then
97
+ MAX_FD="$MAX_FD_LIMIT"
98
+ fi
99
+ ulimit -n $MAX_FD
100
+ if [ $? -ne 0 ] ; then
101
+ warn "Could not set maximum file descriptor limit: $MAX_FD"
102
+ fi
103
+ else
104
+ warn "Could not query maximum file descriptor limit: $MAX_FD_LIMIT"
105
+ fi
106
+ fi
107
+
108
+ # For Darwin, add options to specify how the application appears in the dock
109
+ if $darwin; then
110
+ GRADLE_OPTS="$GRADLE_OPTS \"-Xdock:name=$APP_NAME\" \"-Xdock:icon=$APP_HOME/media/gradle.icns\""
111
+ fi
112
+
113
+ # For Cygwin, switch paths to Windows format before running java
114
+ if $cygwin ; then
115
+ APP_HOME=`cygpath --path --mixed "$APP_HOME"`
116
+ CLASSPATH=`cygpath --path --mixed "$CLASSPATH"`
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
+ # Split up the JVM_OPTS And GRADLE_OPTS values into an array, following the shell quoting and substitution rules
158
+ function splitJvmOpts() {
159
+ JVM_OPTS=("$@")
160
+ }
161
+ eval splitJvmOpts $DEFAULT_JVM_OPTS $JAVA_OPTS $GRADLE_OPTS
162
+ JVM_OPTS[${#JVM_OPTS[*]}]="-Dorg.gradle.appname=$APP_BASE_NAME"
163
+
164
+ exec "$JAVACMD" "${JVM_OPTS[@]}" -classpath "$CLASSPATH" org.gradle.wrapper.GradleWrapperMain "$@"
data/gradlew.bat ADDED
@@ -0,0 +1,90 @@
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
+ @rem Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
12
+ set DEFAULT_JVM_OPTS=
13
+
14
+ set DIRNAME=%~dp0
15
+ if "%DIRNAME%" == "" set DIRNAME=.
16
+ set APP_BASE_NAME=%~n0
17
+ set APP_HOME=%DIRNAME%
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 Windowz variants
50
+
51
+ if not "%OS%" == "Windows_NT" goto win9xME_args
52
+ if "%@eval[2+2]" == "4" goto 4NT_args
53
+
54
+ :win9xME_args
55
+ @rem Slurp the command line arguments.
56
+ set CMD_LINE_ARGS=
57
+ set _SKIP=2
58
+
59
+ :win9xME_args_slurp
60
+ if "x%~1" == "x" goto execute
61
+
62
+ set CMD_LINE_ARGS=%*
63
+ goto execute
64
+
65
+ :4NT_args
66
+ @rem Get arguments from the 4NT Shell from JP Software
67
+ set CMD_LINE_ARGS=%$
68
+
69
+ :execute
70
+ @rem Setup the command line
71
+
72
+ set CLASSPATH=%APP_HOME%\gradle\wrapper\gradle-wrapper.jar
73
+
74
+ @rem Execute Gradle
75
+ "%JAVA_EXE%" %DEFAULT_JVM_OPTS% %JAVA_OPTS% %GRADLE_OPTS% "-Dorg.gradle.appname=%APP_BASE_NAME%" -classpath "%CLASSPATH%" org.gradle.wrapper.GradleWrapperMain %CMD_LINE_ARGS%
76
+
77
+ :end
78
+ @rem End local scope for the variables with windows NT shell
79
+ if "%ERRORLEVEL%"=="0" goto mainEnd
80
+
81
+ :fail
82
+ rem Set variable GRADLE_EXIT_CONSOLE if you need the _script_ return code instead of
83
+ rem the _cmd.exe /c_ return code!
84
+ if not "" == "%GRADLE_EXIT_CONSOLE%" exit 1
85
+ exit /b 1
86
+
87
+ :mainEnd
88
+ if "%OS%"=="Windows_NT" endlocal
89
+
90
+ :omega
@@ -0,0 +1,3 @@
1
+ Embulk::JavaPlugin.register_filter(
2
+ "base58", "org.embulk.filter.base58.Base58FilterPlugin",
3
+ File.expand_path('../../../../classpath', __FILE__))
@@ -0,0 +1,76 @@
1
+ package org.embulk.filter.base58;
2
+
3
+ import java.math.BigInteger;
4
+
5
+ public class Base58
6
+ {
7
+ private Base58()
8
+ {
9
+ }
10
+
11
+ private static final String base58Chars = "123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz"; // Base-58 char library
12
+
13
+ public static String encode(String hex)
14
+ {
15
+ String originalHex = hex;
16
+
17
+ BigInteger numeric = new BigInteger(hex, 16);
18
+ BigInteger zero = new BigInteger("0");
19
+ BigInteger fifty58 = new BigInteger("58");
20
+ String output = "";
21
+
22
+ while (numeric.compareTo(zero) == 1) {
23
+ BigInteger remainder = numeric.mod(fifty58);
24
+ numeric = numeric.divide(fifty58);
25
+ output = base58Chars.charAt(Integer.parseInt(remainder.toString())) + output;
26
+ }
27
+
28
+ //leading zeros
29
+ for (int i = 0; i < originalHex.length() && originalHex.substring(i, 2).equals("00"); i += 2) {
30
+ output = "1" + output;
31
+ }
32
+
33
+ return output;
34
+ }
35
+
36
+ public static String decode(String base58Value)
37
+ {
38
+ String originalBase58 = base58Value;
39
+
40
+ // Ignore bogus base58 strings
41
+ if (base58Value.matches("[^1-9A-HJ-NP-Za-km-z]")) {
42
+ return "";
43
+ }
44
+
45
+ BigInteger output = new BigInteger("0");
46
+ BigInteger fifty58 = new BigInteger("58");
47
+
48
+ for (int i = 0; i < base58Value.length(); i++) {
49
+ int current = base58Chars.indexOf(base58Value.charAt(i));
50
+ output = output.multiply(fifty58).add(new BigInteger(current + ""));
51
+ }
52
+
53
+ String hex = output.toString(16);
54
+
55
+ // Leading zeros
56
+ for (int ii = 0; ii < originalBase58.length() && originalBase58.charAt(ii) == '1'; ii++) {
57
+ hex = "00" + hex;
58
+ }
59
+
60
+ if (hex.length() % 2 != 0) {
61
+ hex = "0" + hex;
62
+ }
63
+
64
+ return hex.toLowerCase();
65
+ }
66
+
67
+ public static String encodeWithPrefix(String hex, String prefix)
68
+ {
69
+ return prefix + encode(hex);
70
+ }
71
+
72
+ public static String decodeWithPrefix(String baseValue, String prefix)
73
+ {
74
+ return decode(baseValue.substring(prefix.length()));
75
+ }
76
+ }
@@ -0,0 +1,229 @@
1
+ package org.embulk.filter.base58;
2
+
3
+ import com.google.common.base.Optional;
4
+ import com.google.common.base.Throwables;
5
+ import com.google.common.collect.ImmutableList;
6
+ import org.embulk.config.Config;
7
+ import org.embulk.config.ConfigDefault;
8
+ import org.embulk.config.ConfigSource;
9
+ import org.embulk.config.Task;
10
+ import org.embulk.config.TaskSource;
11
+ import org.embulk.spi.Column;
12
+ import org.embulk.spi.DataException;
13
+ import org.embulk.spi.Exec;
14
+ import org.embulk.spi.FilterPlugin;
15
+ import org.embulk.spi.Page;
16
+ import org.embulk.spi.PageBuilder;
17
+ import org.embulk.spi.PageOutput;
18
+ import org.embulk.spi.PageReader;
19
+ import org.embulk.spi.Schema;
20
+ import org.embulk.spi.type.Types;
21
+ import org.slf4j.Logger;
22
+
23
+ import java.util.HashMap;
24
+ import java.util.List;
25
+ import java.util.Map;
26
+
27
+ public class Base58FilterPlugin implements FilterPlugin {
28
+
29
+ private final Logger logger = Exec.getLogger(Base58FilterPlugin.class);
30
+
31
+ public interface PluginTask extends Task {
32
+ @Config("columns")
33
+ List<Base58Column> getColumns();
34
+ }
35
+
36
+ public interface Base58Column extends Task {
37
+ @Config("name")
38
+ String getName();
39
+
40
+ @Config("encode")
41
+ @ConfigDefault("true")
42
+ Optional<Boolean> getIsEncode();
43
+
44
+ @Config("prefix")
45
+ @ConfigDefault("")
46
+ Optional<String> getPrefix();
47
+
48
+ @Config("new_name")
49
+ @ConfigDefault("null")
50
+ Optional<String> getNewName();
51
+ }
52
+
53
+ @Override
54
+ public void transaction(ConfigSource config, Schema inputSchema, FilterPlugin.Control control) {
55
+
56
+ PluginTask task = config.loadConfig(PluginTask.class);
57
+ ImmutableList.Builder<Column> builder = ImmutableList.builder();
58
+
59
+ // Roll through original columns
60
+ int i = 0;
61
+ for (Column column : inputSchema.getColumns()) {
62
+ builder.add(new Column(i++, column.getName(), column.getType()));
63
+ }
64
+
65
+ // Append new columns, if base58 columns desire it
66
+ for (Base58Column column : task.getColumns()) {
67
+ if (column.getNewName().isPresent()) {
68
+ logger.info("added column: name: {}, type: {}, index: {}",
69
+ column.getNewName().get(),
70
+ Types.STRING,
71
+ i);
72
+ builder.add(new Column(i++, column.getNewName().get(), Types.STRING));
73
+ } else {
74
+ logger.info("overriding column: name: {}", column.getName());
75
+ }
76
+ }
77
+
78
+ Schema outputSchema = new Schema(builder.build());
79
+ control.run(task.dump(), outputSchema);
80
+ }
81
+
82
+ @Override
83
+ public PageOutput open(final TaskSource taskSource, final Schema inputSchema,
84
+ final Schema outputSchema, final PageOutput output) {
85
+
86
+ final PluginTask task = taskSource.loadTask(PluginTask.class);
87
+ final Map<String, Base58Column> base58ColumnMap = convertBase58ColumnListToMap(task.getColumns());
88
+ final Map<String, Column> outputColumnMap = convertColumnListToMap(outputSchema.getColumns());
89
+
90
+ return new PageOutput() {
91
+ private PageReader reader = new PageReader(inputSchema);
92
+ private PageBuilder builder = new PageBuilder(Exec.getBufferAllocator(), outputSchema, output);
93
+
94
+ @Override
95
+ public void add(Page page) {
96
+ reader.setPage(page);
97
+ while (reader.nextRecord()) {
98
+ setValue();
99
+ builder.addRecord();
100
+ }
101
+ }
102
+
103
+ private void setValue() {
104
+
105
+ // Do base58 conversions ahead of iterating output columns
106
+ final Map<String, String> base58OutputMap = new HashMap<>();;
107
+ for (Base58Column column : base58ColumnMap.values()) {
108
+ String inputValue, convertedValue = null;
109
+
110
+ // If the original column is not a string, then forget about it (it should be hex)
111
+ Column originalColumn = outputColumnMap.get(column.getName());
112
+ if (Types.STRING.equals(originalColumn.getType())) {
113
+ inputValue = reader.getString(originalColumn);
114
+ } else {
115
+ logger.error("cannot convert base58 value of non-string values. name: {}, type: {}, index: {}",
116
+ originalColumn.getName(),
117
+ originalColumn.getType(),
118
+ originalColumn.getIndex());
119
+ throw new DataException("Unexpected string type in column `"+originalColumn.getName()+"`. Got: " + originalColumn.getType());
120
+ }
121
+
122
+ // Convert the value
123
+ try {
124
+ convertedValue = convertValue(inputValue, column.getIsEncode().get(), column.getPrefix().get());
125
+ } catch (Exception e) {
126
+ // Failed to do the conversion. Probably misconfigured or malformed value
127
+ logger.error("failed to encode/decode base58 column value. name: {}, type: {}, index: {}, value: {}, method: {}, prefix: {}, target_name: {}",
128
+ originalColumn.getName(),
129
+ originalColumn.getType(),
130
+ originalColumn.getIndex(),
131
+ inputValue,
132
+ column.getIsEncode().get() ? "encode" : "decode",
133
+ column.getPrefix(),
134
+ column.getNewName().or(column.getName()));
135
+ // Don't crash the import if a single value is screwed up. Just log it for now
136
+ }
137
+
138
+ // Add it to the output mappings
139
+ if (column.getNewName().isPresent()) {
140
+ base58OutputMap.put(column.getNewName().get(), convertedValue);
141
+ } else {
142
+ base58OutputMap.put(column.getName(), convertedValue);
143
+ }
144
+ }
145
+
146
+ List<Column> columns = outputSchema.getColumns();
147
+ for (Column outputColumn : columns) {
148
+
149
+ // Did we convert the value for this column?
150
+ if (base58OutputMap.containsKey(outputColumn.getName())) {
151
+ String value = base58OutputMap.get(outputColumn.getName());
152
+ if (value == null) {
153
+ builder.setNull(outputColumn);
154
+ } else {
155
+ builder.setString(outputColumn, base58OutputMap.get(outputColumn.getName()));
156
+ }
157
+ continue;
158
+ }
159
+
160
+ // No value?
161
+ if (reader.isNull(outputColumn)) {
162
+ builder.setNull(outputColumn);
163
+ continue;
164
+ }
165
+
166
+ // Inherit value
167
+ if (Types.STRING.equals(outputColumn.getType())) {
168
+ builder.setString(outputColumn, reader.getString(outputColumn));
169
+ }
170
+ else if (Types.BOOLEAN.equals(outputColumn.getType())) {
171
+ builder.setBoolean(outputColumn, reader.getBoolean(outputColumn));
172
+ }
173
+ else if (Types.DOUBLE.equals(outputColumn.getType())) {
174
+ builder.setDouble(outputColumn, reader.getDouble(outputColumn));
175
+ }
176
+ else if (Types.LONG.equals(outputColumn.getType())) {
177
+ builder.setLong(outputColumn, reader.getLong(outputColumn));
178
+ }
179
+ else if (Types.TIMESTAMP.equals(outputColumn.getType())) {
180
+ builder.setTimestamp(outputColumn, reader.getTimestamp(outputColumn));
181
+ }
182
+ else if (Types.JSON.equals(outputColumn.getType())) {
183
+ builder.setJson(outputColumn, reader.getJson(outputColumn));
184
+ }
185
+ }
186
+ }
187
+
188
+ private String convertValue(String value, Boolean isEncode, String prefix) {
189
+ String result = null;
190
+ try {
191
+ if (isEncode) {
192
+ result = Base58.encodeWithPrefix(value, prefix);
193
+ } else {
194
+ result = Base58.decodeWithPrefix(value, prefix);
195
+ }
196
+ } catch (Exception e) {
197
+ Throwables.propagate(e);
198
+ }
199
+ return result;
200
+ }
201
+
202
+ @Override
203
+ public void finish() {
204
+ builder.finish();
205
+ }
206
+
207
+ @Override
208
+ public void close() {
209
+ builder.close();
210
+ }
211
+ };
212
+ }
213
+
214
+ private static Map<String, Base58Column> convertBase58ColumnListToMap(List<Base58Column> base58Columns) {
215
+ Map<String, Base58Column> result = new HashMap<>();
216
+ for (Base58Column base58Column : base58Columns) {
217
+ result.put(base58Column.getName(), base58Column);
218
+ }
219
+ return result;
220
+ }
221
+
222
+ private static Map<String, Column> convertColumnListToMap(List<Column> columns) {
223
+ Map<String, Column> result = new HashMap<>();
224
+ for (Column column : columns) {
225
+ result.put(column.getName(), column);
226
+ }
227
+ return result;
228
+ }
229
+ }
@@ -0,0 +1,27 @@
1
+ package org.embulk.filter.base58;
2
+
3
+ //import com.fasterxml.jackson.databind.ObjectMapper;
4
+ //import com.google.inject.AbstractModule;
5
+ //import com.google.inject.Guice;
6
+ //import org.embulk.EmbulkEmbed;
7
+ //import org.embulk.config.ConfigLoader;
8
+ //import org.embulk.config.ModelManager;
9
+ //import org.junit.Test;
10
+ //
11
+ //import java.io.File;
12
+ //import java.io.IOException;
13
+
14
+ public class TestBase58FilterPlugin
15
+ {
16
+ // @Test
17
+ // public void test() throws IOException {
18
+ // EmbulkEmbed embed = new EmbulkEmbed.Bootstrap().initialize();
19
+ // ConfigLoader loader = new ConfigLoader(new ModelManager(Guice.createInjector(new AbstractModule() {
20
+ // @Override
21
+ // protected void configure() {}
22
+ // }),
23
+ // new ObjectMapper()
24
+ // ));
25
+ // embed.run(loader.fromYamlFile(new File("./test.yml")));
26
+ // }
27
+ }
metadata ADDED
@@ -0,0 +1,86 @@
1
+ --- !ruby/object:Gem::Specification
2
+ name: embulk-filter-base58
3
+ version: !ruby/object:Gem::Version
4
+ version: 0.1.0
5
+ platform: ruby
6
+ authors:
7
+ - Kevin Fitzgerald
8
+ autorequire:
9
+ bindir: bin
10
+ cert_chain: []
11
+ date: 2016-07-16 00:00:00.000000000 Z
12
+ dependencies:
13
+ - !ruby/object:Gem::Dependency
14
+ requirement: !ruby/object:Gem::Requirement
15
+ requirements:
16
+ - - ~>
17
+ - !ruby/object:Gem::Version
18
+ version: '1.0'
19
+ name: bundler
20
+ prerelease: false
21
+ type: :development
22
+ version_requirements: !ruby/object:Gem::Requirement
23
+ requirements:
24
+ - - ~>
25
+ - !ruby/object:Gem::Version
26
+ version: '1.0'
27
+ - !ruby/object:Gem::Dependency
28
+ requirement: !ruby/object:Gem::Requirement
29
+ requirements:
30
+ - - '>='
31
+ - !ruby/object:Gem::Version
32
+ version: '10.0'
33
+ name: rake
34
+ prerelease: false
35
+ type: :development
36
+ version_requirements: !ruby/object:Gem::Requirement
37
+ requirements:
38
+ - - '>='
39
+ - !ruby/object:Gem::Version
40
+ version: '10.0'
41
+ description: Embulk filter plugin to convert a hex value to and from base58.
42
+ email:
43
+ - kevin@okanjo.com
44
+ executables: []
45
+ extensions: []
46
+ extra_rdoc_files: []
47
+ files:
48
+ - .gitignore
49
+ - .travis.yml
50
+ - LICENSE.txt
51
+ - README.md
52
+ - build.gradle
53
+ - gradle/wrapper/gradle-wrapper.jar
54
+ - gradle/wrapper/gradle-wrapper.properties
55
+ - gradlew
56
+ - gradlew.bat
57
+ - lib/embulk/filter/base58.rb
58
+ - src/main/java/org/embulk/filter/base58/Base58.java
59
+ - src/main/java/org/embulk/filter/base58/Base58FilterPlugin.java
60
+ - src/test/java/org/embulk/filter/base58/TestBase58FilterPlugin.java
61
+ - classpath/embulk-filter-base58-0.1.0.jar
62
+ homepage: https://github.com/kfitzgerald/embulk-filter-base58
63
+ licenses:
64
+ - MIT
65
+ metadata: {}
66
+ post_install_message:
67
+ rdoc_options: []
68
+ require_paths:
69
+ - lib
70
+ required_ruby_version: !ruby/object:Gem::Requirement
71
+ requirements:
72
+ - - '>='
73
+ - !ruby/object:Gem::Version
74
+ version: '0'
75
+ required_rubygems_version: !ruby/object:Gem::Requirement
76
+ requirements:
77
+ - - '>='
78
+ - !ruby/object:Gem::Version
79
+ version: '0'
80
+ requirements: []
81
+ rubyforge_project:
82
+ rubygems_version: 2.1.9
83
+ signing_key:
84
+ specification_version: 4
85
+ summary: Base58 filter plugin for Embulk
86
+ test_files: []