embulk-input-script 0.1.0

Sign up to get free protection for your applications and to get access to all the features.
@@ -0,0 +1,7 @@
1
+ ---
2
+ SHA1:
3
+ metadata.gz: 85ab6da285a00bff398fefa24574109c4198d538
4
+ data.tar.gz: 2700badaf9e775511d8ef98286e01bdf9af9ae30
5
+ SHA512:
6
+ metadata.gz: 22ad260aa3842d50b2ace4947a65ddba1b1cb5d10792167ae3d7342102e54fbd9db540c54572e1be1dc967675b0ba4ad7196dc2830c635979c4cd8d3efe02db5
7
+ data.tar.gz: cfb24bbe52846dd278a51ade46b3f924e2dd2fa18d5e60c07be1060cf64228f688dcf52ec074e9d4bd1b3e4a67ba6e220a7dc06214aea9268e51fd2fde59b354
@@ -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
@@ -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.
@@ -0,0 +1,32 @@
1
+ # Script input plugin for Embulk
2
+
3
+ TODO: Write short description here and build.gradle file.
4
+
5
+ ## Overview
6
+
7
+ * **Plugin type**: input
8
+ * **Resume supported**: yes
9
+ * **Cleanup supported**: yes
10
+ * **Guess supported**: no
11
+
12
+ ## Configuration
13
+
14
+ - **option1**: description (integer, required)
15
+ - **option2**: description (string, default: `"myvalue"`)
16
+ - **option3**: description (string, default: `null`)
17
+
18
+ ## Example
19
+
20
+ ```yaml
21
+ in:
22
+ type: script
23
+ option1: example1
24
+ option2: example2
25
+ ```
26
+
27
+
28
+ ## Build
29
+
30
+ ```
31
+ $ ./gradlew gem # -t to watch change of files and rebuild continuously
32
+ ```
@@ -0,0 +1,99 @@
1
+ plugins {
2
+ id "com.jfrog.bintray" version "1.1"
3
+ id "com.github.jruby-gradle.base" version "1.5.0"
4
+ id "java"
5
+ id "checkstyle"
6
+ }
7
+ import com.github.jrubygradle.JRubyExec
8
+ repositories {
9
+ mavenCentral()
10
+ jcenter()
11
+ }
12
+ configurations {
13
+ provided
14
+ }
15
+
16
+ version = "0.1.0"
17
+
18
+ sourceCompatibility = 1.8
19
+ targetCompatibility = 1.8
20
+
21
+ dependencies {
22
+ compile "com.opencsv:opencsv:4.5"
23
+ compile "org.embulk:embulk-core:0.9.14"
24
+ provided "org.embulk:embulk-core:0.9.14"
25
+ // compile "YOUR_JAR_DEPENDENCY_GROUP:YOUR_JAR_DEPENDENCY_MODULE:YOUR_JAR_DEPENDENCY_VERSION"
26
+ testCompile "junit:junit:4.+"
27
+ }
28
+
29
+ task classpath(type: Copy, dependsOn: ["jar"]) {
30
+ doFirst { file("classpath").deleteDir() }
31
+ from (configurations.runtime - configurations.provided + files(jar.archivePath))
32
+ into "classpath"
33
+ }
34
+ clean { delete "classpath" }
35
+
36
+ checkstyle {
37
+ configFile = file("${project.rootDir}/config/checkstyle/checkstyle.xml")
38
+ toolVersion = '6.14.1'
39
+ }
40
+ checkstyleMain {
41
+ configFile = file("${project.rootDir}/config/checkstyle/default.xml")
42
+ ignoreFailures = true
43
+ }
44
+ checkstyleTest {
45
+ configFile = file("${project.rootDir}/config/checkstyle/default.xml")
46
+ ignoreFailures = true
47
+ }
48
+ task checkstyle(type: Checkstyle) {
49
+ classpath = sourceSets.main.output + sourceSets.test.output
50
+ source = sourceSets.main.allJava + sourceSets.test.allJava
51
+ }
52
+
53
+ task gem(type: JRubyExec, dependsOn: ["gemspec", "classpath"]) {
54
+ jrubyArgs "-S"
55
+ script "gem"
56
+ scriptArgs "build", "${project.name}.gemspec"
57
+ doLast { ant.move(file: "${project.name}-${project.version}.gem", todir: "pkg") }
58
+ }
59
+
60
+ task gemPush(type: JRubyExec, dependsOn: ["gem"]) {
61
+ jrubyArgs "-S"
62
+ script "gem"
63
+ scriptArgs "push", "pkg/${project.name}-${project.version}.gem"
64
+ }
65
+
66
+ task "package"(dependsOn: ["gemspec", "classpath"]) {
67
+ doLast {
68
+ println "> Build succeeded."
69
+ println "> You can run embulk with '-L ${file(".").absolutePath}' argument."
70
+ }
71
+ }
72
+
73
+ task gemspec {
74
+ ext.gemspecFile = file("${project.name}.gemspec")
75
+ inputs.file "build.gradle"
76
+ outputs.file gemspecFile
77
+ doLast { gemspecFile.write($/
78
+ Gem::Specification.new do |spec|
79
+ spec.name = "${project.name}"
80
+ spec.version = "${project.version}"
81
+ spec.authors = ["Sadayuki Furuhashi"]
82
+ spec.summary = %[Script input plugin for Embulk]
83
+ spec.description = %[Loads records from Script.]
84
+ spec.email = ["frsyuki@gmail.com"]
85
+ spec.licenses = ["MIT"]
86
+ # TODO set this: spec.homepage = "https://github.com/frsyuki/embulk-input-script"
87
+
88
+ spec.files = `git ls-files`.split("\n") + Dir["classpath/*.jar"]
89
+ spec.test_files = spec.files.grep(%r"^(test|spec)/")
90
+ spec.require_paths = ["lib"]
91
+
92
+ #spec.add_dependency 'YOUR_GEM_DEPENDENCY', ['~> YOUR_GEM_DEPENDENCY_VERSION']
93
+ spec.add_development_dependency 'bundler', ['~> 1.0']
94
+ spec.add_development_dependency 'rake', ['~> 12.0']
95
+ end
96
+ /$)
97
+ }
98
+ }
99
+ clean { delete "${project.name}.gemspec" }
@@ -0,0 +1,128 @@
1
+ <?xml version="1.0" encoding="UTF-8"?>
2
+ <!DOCTYPE module PUBLIC
3
+ "-//Puppy Crawl//DTD Check Configuration 1.3//EN"
4
+ "http://www.puppycrawl.com/dtds/configuration_1_3.dtd">
5
+ <module name="Checker">
6
+ <!-- https://github.com/facebook/presto/blob/master/src/checkstyle/checks.xml -->
7
+ <module name="FileTabCharacter"/>
8
+ <module name="NewlineAtEndOfFile">
9
+ <property name="lineSeparator" value="lf"/>
10
+ </module>
11
+ <module name="RegexpMultiline">
12
+ <property name="format" value="\r"/>
13
+ <property name="message" value="Line contains carriage return"/>
14
+ </module>
15
+ <module name="RegexpMultiline">
16
+ <property name="format" value=" \n"/>
17
+ <property name="message" value="Line has trailing whitespace"/>
18
+ </module>
19
+ <module name="RegexpMultiline">
20
+ <property name="format" value="\{\n\n"/>
21
+ <property name="message" value="Blank line after opening brace"/>
22
+ </module>
23
+ <module name="RegexpMultiline">
24
+ <property name="format" value="\n\n\s*\}"/>
25
+ <property name="message" value="Blank line before closing brace"/>
26
+ </module>
27
+ <module name="RegexpMultiline">
28
+ <property name="format" value="\n\n\n"/>
29
+ <property name="message" value="Multiple consecutive blank lines"/>
30
+ </module>
31
+ <module name="RegexpMultiline">
32
+ <property name="format" value="\n\n\Z"/>
33
+ <property name="message" value="Blank line before end of file"/>
34
+ </module>
35
+ <module name="RegexpMultiline">
36
+ <property name="format" value="Preconditions\.checkNotNull"/>
37
+ <property name="message" value="Use of checkNotNull"/>
38
+ </module>
39
+
40
+ <module name="TreeWalker">
41
+ <module name="EmptyBlock">
42
+ <property name="option" value="text"/>
43
+ <property name="tokens" value="
44
+ LITERAL_DO, LITERAL_ELSE, LITERAL_FINALLY, LITERAL_IF,
45
+ LITERAL_FOR, LITERAL_TRY, LITERAL_WHILE, INSTANCE_INIT, STATIC_INIT"/>
46
+ </module>
47
+ <module name="EmptyStatement"/>
48
+ <module name="EmptyForInitializerPad"/>
49
+ <module name="EmptyForIteratorPad">
50
+ <property name="option" value="space"/>
51
+ </module>
52
+ <module name="MethodParamPad">
53
+ <property name="allowLineBreaks" value="true"/>
54
+ <property name="option" value="nospace"/>
55
+ </module>
56
+ <module name="ParenPad"/>
57
+ <module name="TypecastParenPad"/>
58
+ <module name="NeedBraces"/>
59
+ <module name="LeftCurly">
60
+ <property name="option" value="nl"/>
61
+ <property name="tokens" value="CLASS_DEF, CTOR_DEF, INTERFACE_DEF, METHOD_DEF"/>
62
+ </module>
63
+ <module name="LeftCurly">
64
+ <property name="option" value="eol"/>
65
+ <property name="tokens" value="
66
+ LITERAL_CATCH, LITERAL_DO, LITERAL_ELSE, LITERAL_FINALLY, LITERAL_FOR,
67
+ LITERAL_IF, LITERAL_SWITCH, LITERAL_SYNCHRONIZED, LITERAL_TRY, LITERAL_WHILE"/>
68
+ </module>
69
+ <module name="RightCurly">
70
+ <property name="option" value="alone"/>
71
+ </module>
72
+ <module name="GenericWhitespace"/>
73
+ <module name="WhitespaceAfter"/>
74
+ <module name="NoWhitespaceBefore"/>
75
+
76
+ <module name="UpperEll"/>
77
+ <module name="DefaultComesLast"/>
78
+ <module name="ArrayTypeStyle"/>
79
+ <module name="MultipleVariableDeclarations"/>
80
+ <module name="ModifierOrder"/>
81
+ <module name="OneStatementPerLine"/>
82
+ <module name="StringLiteralEquality"/>
83
+ <module name="MutableException"/>
84
+ <module name="EqualsHashCode"/>
85
+ <module name="InnerAssignment"/>
86
+ <module name="InterfaceIsType"/>
87
+ <module name="HideUtilityClassConstructor"/>
88
+
89
+ <module name="MemberName"/>
90
+ <module name="LocalVariableName"/>
91
+ <module name="LocalFinalVariableName"/>
92
+ <module name="TypeName"/>
93
+ <module name="PackageName"/>
94
+ <module name="ParameterName"/>
95
+ <module name="StaticVariableName"/>
96
+ <module name="ClassTypeParameterName">
97
+ <property name="format" value="^[A-Z][0-9]?$"/>
98
+ </module>
99
+ <module name="MethodTypeParameterName">
100
+ <property name="format" value="^[A-Z][0-9]?$"/>
101
+ </module>
102
+
103
+ <module name="AvoidStarImport"/>
104
+ <module name="RedundantImport"/>
105
+ <module name="UnusedImports"/>
106
+ <module name="ImportOrder">
107
+ <property name="groups" value="*,javax,java"/>
108
+ <property name="separated" value="true"/>
109
+ <property name="option" value="bottom"/>
110
+ <property name="sortStaticImportsAlphabetically" value="true"/>
111
+ </module>
112
+
113
+ <module name="WhitespaceAround">
114
+ <property name="allowEmptyConstructors" value="true"/>
115
+ <property name="allowEmptyMethods" value="true"/>
116
+ <property name="ignoreEnhancedForColon" value="false"/>
117
+ <property name="tokens" value="
118
+ ASSIGN, BAND, BAND_ASSIGN, BOR, BOR_ASSIGN, BSR, BSR_ASSIGN,
119
+ BXOR, BXOR_ASSIGN, COLON, DIV, DIV_ASSIGN, EQUAL, GE, GT, LAND, LE,
120
+ LITERAL_ASSERT, LITERAL_CATCH, LITERAL_DO, LITERAL_ELSE,
121
+ LITERAL_FINALLY, LITERAL_FOR, LITERAL_IF, LITERAL_RETURN,
122
+ LITERAL_SYNCHRONIZED, LITERAL_TRY, LITERAL_WHILE,
123
+ LOR, LT, MINUS, MINUS_ASSIGN, MOD, MOD_ASSIGN, NOT_EQUAL,
124
+ PLUS, PLUS_ASSIGN, QUESTION, SL, SLIST, SL_ASSIGN, SR, SR_ASSIGN,
125
+ STAR, STAR_ASSIGN, TYPE_EXTENSION_AND"/>
126
+ </module>
127
+ </module>
128
+ </module>
@@ -0,0 +1,108 @@
1
+ <?xml version="1.0" encoding="UTF-8"?>
2
+ <!DOCTYPE module PUBLIC
3
+ "-//Puppy Crawl//DTD Check Configuration 1.3//EN"
4
+ "http://www.puppycrawl.com/dtds/configuration_1_3.dtd">
5
+ <!--
6
+ This is a subset of ./checkstyle.xml which allows some loose styles
7
+ -->
8
+ <module name="Checker">
9
+ <module name="FileTabCharacter"/>
10
+ <module name="NewlineAtEndOfFile">
11
+ <property name="lineSeparator" value="lf"/>
12
+ </module>
13
+ <module name="RegexpMultiline">
14
+ <property name="format" value="\r"/>
15
+ <property name="message" value="Line contains carriage return"/>
16
+ </module>
17
+ <module name="RegexpMultiline">
18
+ <property name="format" value=" \n"/>
19
+ <property name="message" value="Line has trailing whitespace"/>
20
+ </module>
21
+ <module name="RegexpMultiline">
22
+ <property name="format" value="\n\n\n"/>
23
+ <property name="message" value="Multiple consecutive blank lines"/>
24
+ </module>
25
+ <module name="RegexpMultiline">
26
+ <property name="format" value="\n\n\Z"/>
27
+ <property name="message" value="Blank line before end of file"/>
28
+ </module>
29
+
30
+ <module name="TreeWalker">
31
+ <module name="EmptyBlock">
32
+ <property name="option" value="text"/>
33
+ <property name="tokens" value="
34
+ LITERAL_DO, LITERAL_ELSE, LITERAL_FINALLY, LITERAL_IF,
35
+ LITERAL_FOR, LITERAL_TRY, LITERAL_WHILE, INSTANCE_INIT, STATIC_INIT"/>
36
+ </module>
37
+ <module name="EmptyStatement"/>
38
+ <module name="EmptyForInitializerPad"/>
39
+ <module name="EmptyForIteratorPad">
40
+ <property name="option" value="space"/>
41
+ </module>
42
+ <module name="MethodParamPad">
43
+ <property name="allowLineBreaks" value="true"/>
44
+ <property name="option" value="nospace"/>
45
+ </module>
46
+ <module name="ParenPad"/>
47
+ <module name="TypecastParenPad"/>
48
+ <module name="NeedBraces"/>
49
+ <module name="LeftCurly">
50
+ <property name="option" value="nl"/>
51
+ <property name="tokens" value="CLASS_DEF, CTOR_DEF, INTERFACE_DEF, METHOD_DEF"/>
52
+ </module>
53
+ <module name="LeftCurly">
54
+ <property name="option" value="eol"/>
55
+ <property name="tokens" value="
56
+ LITERAL_CATCH, LITERAL_DO, LITERAL_ELSE, LITERAL_FINALLY, LITERAL_FOR,
57
+ LITERAL_IF, LITERAL_SWITCH, LITERAL_SYNCHRONIZED, LITERAL_TRY, LITERAL_WHILE"/>
58
+ </module>
59
+ <module name="RightCurly">
60
+ <property name="option" value="alone"/>
61
+ </module>
62
+ <module name="GenericWhitespace"/>
63
+ <module name="WhitespaceAfter"/>
64
+ <module name="NoWhitespaceBefore"/>
65
+
66
+ <module name="UpperEll"/>
67
+ <module name="DefaultComesLast"/>
68
+ <module name="ArrayTypeStyle"/>
69
+ <module name="MultipleVariableDeclarations"/>
70
+ <module name="ModifierOrder"/>
71
+ <module name="OneStatementPerLine"/>
72
+ <module name="StringLiteralEquality"/>
73
+ <module name="MutableException"/>
74
+ <module name="EqualsHashCode"/>
75
+ <module name="InnerAssignment"/>
76
+ <module name="InterfaceIsType"/>
77
+ <module name="HideUtilityClassConstructor"/>
78
+
79
+ <module name="MemberName"/>
80
+ <module name="LocalVariableName"/>
81
+ <module name="LocalFinalVariableName"/>
82
+ <module name="TypeName"/>
83
+ <module name="PackageName"/>
84
+ <module name="ParameterName"/>
85
+ <module name="StaticVariableName"/>
86
+ <module name="ClassTypeParameterName">
87
+ <property name="format" value="^[A-Z][0-9]?$"/>
88
+ </module>
89
+ <module name="MethodTypeParameterName">
90
+ <property name="format" value="^[A-Z][0-9]?$"/>
91
+ </module>
92
+
93
+ <module name="WhitespaceAround">
94
+ <property name="allowEmptyConstructors" value="true"/>
95
+ <property name="allowEmptyMethods" value="true"/>
96
+ <property name="ignoreEnhancedForColon" value="false"/>
97
+ <property name="tokens" value="
98
+ ASSIGN, BAND, BAND_ASSIGN, BOR, BOR_ASSIGN, BSR, BSR_ASSIGN,
99
+ BXOR, BXOR_ASSIGN, COLON, DIV, DIV_ASSIGN, EQUAL, GE, GT, LAND, LE,
100
+ LITERAL_ASSERT, LITERAL_CATCH, LITERAL_DO, LITERAL_ELSE,
101
+ LITERAL_FINALLY, LITERAL_FOR, LITERAL_IF, LITERAL_RETURN,
102
+ LITERAL_SYNCHRONIZED, LITERAL_TRY, LITERAL_WHILE,
103
+ LOR, LT, MINUS, MINUS_ASSIGN, MOD, MOD_ASSIGN, NOT_EQUAL,
104
+ PLUS, PLUS_ASSIGN, QUESTION, SL, SLIST, SL_ASSIGN, SR, SR_ASSIGN,
105
+ STAR, STAR_ASSIGN, TYPE_EXTENSION_AND"/>
106
+ </module>
107
+ </module>
108
+ </module>
@@ -0,0 +1,7 @@
1
+ in:
2
+ type: script
3
+ run: ruby script.rb
4
+ config:
5
+ any: param
6
+ out:
7
+ type: stdout
@@ -0,0 +1,39 @@
1
+ require 'yaml'
2
+
3
+ case ARGV[0]
4
+ when "setup"
5
+ config = YAML.load_file(ARGV[1])
6
+ setup_path = ARGV[2]
7
+ puts "config: #{config.inspect}"
8
+
9
+ setup = {
10
+ "columns" => [
11
+ {"name" => "s", "type" => "string"},
12
+ {"name" => "i", "type" => "long"},
13
+ {"name" => "f", "type" => "double"},
14
+ {"name" => "t", "type" => "timestamp", "format" => "%Y-%m-%d %H:%M:%S"},
15
+ {"name" => "j", "type" => "json"},
16
+ ],
17
+ "tasks" => 2,
18
+ }
19
+ File.write(setup_path, YAML.dump(setup))
20
+
21
+ when "run"
22
+ setup = YAML.load_file(ARGV[1])
23
+ output_path = ARGV[2]
24
+ task_index = ARGV[3].to_i
25
+
26
+ File.open(output_path, "w") do |out|
27
+ [
28
+ ["a", task_index, 0.1, "2019-01-01 00:00:00", ""],
29
+ ["b", task_index, 0.2, "2019-01-01 00:00:00", ""],
30
+ ["c", task_index, 0.3, "2019-01-01 00:00:00", ""],
31
+ ].each do |r|
32
+ out.puts(r.join(","))
33
+ end
34
+ end
35
+
36
+ when "finish"
37
+ puts "Done."
38
+ end
39
+
@@ -0,0 +1,5 @@
1
+ distributionBase=GRADLE_USER_HOME
2
+ distributionPath=wrapper/dists
3
+ zipStoreBase=GRADLE_USER_HOME
4
+ zipStorePath=wrapper/dists
5
+ distributionUrl=https\://services.gradle.org/distributions/gradle-4.1-bin.zip
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" "$@"
@@ -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,3 @@
1
+ Embulk::JavaPlugin.register_input(
2
+ "script", "org.embulk.input.script.ScriptInputPlugin",
3
+ File.expand_path('../../../../classpath', __FILE__))
@@ -0,0 +1,475 @@
1
+ package org.embulk.input.script;
2
+
3
+ import com.google.common.base.Optional;
4
+ import com.google.common.collect.ImmutableList;
5
+ import com.google.common.collect.ImmutableSet;
6
+ import com.google.common.escape.CharEscaperBuilder;
7
+ import com.google.common.escape.Escaper;
8
+ import com.opencsv.CSVReader;
9
+ import com.opencsv.CSVReaderBuilder;
10
+ import com.opencsv.enums.CSVReaderNullFieldIndicator;
11
+ import java.io.BufferedWriter;
12
+ import java.io.File;
13
+ import java.io.IOException;
14
+ import java.io.InputStream;
15
+ import java.io.InputStreamReader;
16
+ import java.nio.file.Files;
17
+ import java.nio.file.Path;
18
+ import java.nio.file.Paths;
19
+ import java.time.Instant;
20
+ import java.time.format.DateTimeFormatter;
21
+ import java.time.format.DateTimeParseException;
22
+ import java.util.ArrayList;
23
+ import java.util.Iterator;
24
+ import java.util.List;
25
+ import java.util.Map;
26
+ import org.embulk.config.Config;
27
+ import org.embulk.config.ConfigDefault;
28
+ import org.embulk.config.ConfigDiff;
29
+ import org.embulk.config.ConfigException;
30
+ import org.embulk.config.ConfigInject;
31
+ import org.embulk.config.ConfigLoader;
32
+ import org.embulk.config.ConfigSource;
33
+ import org.embulk.config.ModelManager;
34
+ import org.embulk.config.Task;
35
+ import org.embulk.config.TaskReport;
36
+ import org.embulk.config.TaskSource;
37
+ import org.embulk.spi.Column;
38
+ import org.embulk.spi.ColumnVisitor;
39
+ import org.embulk.spi.DataException;
40
+ import org.embulk.spi.Exec;
41
+ import org.embulk.spi.InputPlugin;
42
+ import org.embulk.spi.PageBuilder;
43
+ import org.embulk.spi.PageOutput;
44
+ import org.embulk.spi.Schema;
45
+ import org.embulk.spi.SchemaConfig;
46
+ import org.embulk.spi.json.JsonParseException;
47
+ import org.embulk.spi.json.JsonParser;
48
+ import org.embulk.spi.time.Timestamp;
49
+ import org.embulk.spi.time.TimestampParseException;
50
+ import org.embulk.spi.time.TimestampParser;
51
+ import org.embulk.spi.util.Timestamps;
52
+ import org.slf4j.Logger;
53
+ import org.yaml.snakeyaml.Yaml;
54
+ import static java.nio.charset.StandardCharsets.UTF_8;
55
+
56
+ public class ScriptInputPlugin
57
+ implements InputPlugin
58
+ {
59
+ public interface PluginTask extends Task {
60
+ @Config("run")
61
+ public String getCommand();
62
+
63
+ @Config("config")
64
+ @ConfigDefault("{}")
65
+ public ConfigSource getConfig();
66
+
67
+ @Config("cwd")
68
+ @ConfigDefault("\"\"")
69
+ public String getCwd();
70
+
71
+ @Config("env")
72
+ @ConfigDefault("{}")
73
+ public Map<String, String> getEnv();
74
+
75
+ @Config("try_named_pipe")
76
+ @ConfigDefault("true")
77
+ public boolean getTryNamedPipe();
78
+
79
+ @ConfigInject
80
+ public ConfigLoader getConfigLoader();
81
+
82
+ public String getTempDir();
83
+ public void setTempDir(String path);
84
+
85
+ public ConfigSource getSetupConfig();
86
+ public void setSetupConfig(ConfigSource setupConfig);
87
+ }
88
+
89
+ public interface SetupTask extends Task, TimestampParser.Task {
90
+ @Config("columns")
91
+ SchemaConfig getSchemaConfig();
92
+
93
+ @Config("tasks")
94
+ @ConfigDefault("1")
95
+ int getTasks();
96
+ }
97
+
98
+ public static final List<String> SHELL = ImmutableList.of(
99
+ // TODO use ["PowerShell.exe", "-Command"] on windows?
100
+ "sh", "-c"
101
+ );
102
+
103
+ public static final Escaper SHELL_ESCAPER = new CharEscaperBuilder()
104
+ .addEscape('\'', "'\"'\"'")
105
+ .addEscape(' ', "\\ ")
106
+ .toEscaper();
107
+
108
+ private static Path resolveOutPath(Path tempDir, int taskIndex) {
109
+ return tempDir.resolve("output-" + taskIndex + ".csv");
110
+ }
111
+
112
+ private static Path resolveConfigPath(Path tempDir) {
113
+ return tempDir.resolve("config.yml");
114
+ }
115
+
116
+ private static Path resolveSetupPath(Path tempDir) {
117
+ return tempDir.resolve("setup.yml");
118
+ }
119
+
120
+ private final Logger logger = Exec.getLogger(getClass());
121
+
122
+ @Override
123
+ public ConfigDiff transaction(ConfigSource config,
124
+ InputPlugin.Control control) {
125
+ PluginTask task = config.loadConfig(PluginTask.class);
126
+ Path tempDir;
127
+ Path configPath;
128
+ try {
129
+ tempDir = Files.createTempDirectory("embulk-input-script-");
130
+ task.setTempDir(tempDir.toString());
131
+
132
+ // create /tmp/config.yml
133
+ configPath = resolveConfigPath(tempDir);
134
+ writeConfigFile(configPath, task.getConfig());
135
+ } catch (IOException ex) {
136
+ throw new ConfigException("Failed to create config.yml file", ex);
137
+ }
138
+
139
+ // $ <command> setup /tmp/config.yml /tmp/setup.yml
140
+ Path setupPath = resolveSetupPath(tempDir);
141
+ try {
142
+ int ecode = runCommand(task.getCwd(), task.getEnv(), task.getCommand(),
143
+ ImmutableList.of(
144
+ "setup",
145
+ configPath.toString(),
146
+ setupPath.toString()));
147
+ if (ecode != 0) {
148
+ throw new ConfigException("Setup command exited with error code " + ecode);
149
+ }
150
+ } catch (IOException ex) {
151
+ throw new ConfigException("Failed to start setup command", ex);
152
+ }
153
+
154
+ // read /tmp/setup.yml
155
+ ConfigSource setupConfig;
156
+ try {
157
+ setupConfig = task.getConfigLoader().fromYamlFile(setupPath.toFile());
158
+ } catch (IOException ex) {
159
+ throw new ConfigException("Setup command didn't create appropriate setup.yml", ex);
160
+ }
161
+ task.setSetupConfig(setupConfig);
162
+ SetupTask setup = setupConfig.loadConfig(SetupTask.class);
163
+
164
+ Schema schema = setup.getSchemaConfig().toSchema();
165
+ int tasks = setup.getTasks();
166
+ return resume(task.dump(), schema, tasks, control);
167
+ }
168
+
169
+ @Override
170
+ public ConfigDiff resume(TaskSource taskSource,
171
+ Schema schema, int taskCount,
172
+ InputPlugin.Control control) {
173
+ PluginTask task = taskSource.loadTask(PluginTask.class);
174
+ Path tempDir = Paths.get(task.getTempDir());
175
+
176
+ // Repeat run for taskCount times
177
+ control.run(taskSource, schema, taskCount);
178
+
179
+ // $ <command> finish /tmp/setup.yml
180
+ Path setupPath = resolveSetupPath(tempDir);
181
+ try {
182
+ int ecode = runCommand(task.getCwd(), task.getEnv(), task.getCommand(),
183
+ ImmutableList.of(
184
+ "finish",
185
+ setupPath.toString()));
186
+ if (ecode != 0) {
187
+ throw new ConfigException("Finish command exited with error code " + ecode);
188
+ }
189
+ } catch (IOException ex) {
190
+ throw new ConfigException("Failed to start finish command", ex);
191
+ }
192
+
193
+ return Exec.newConfigDiff();
194
+ }
195
+
196
+ @Override
197
+ public void cleanup(TaskSource taskSource,
198
+ Schema schema, int taskCount,
199
+ List<TaskReport> successTaskReports) {
200
+ PluginTask task = taskSource.loadTask(PluginTask.class);
201
+ Path tempDir = Paths.get(task.getTempDir());
202
+ for (int i = 0; i < taskCount; i++) {
203
+ Path outPath = resolveOutPath(tempDir, i);
204
+ try {
205
+ Files.deleteIfExists(outPath);
206
+ }
207
+ catch (IOException ex) {
208
+ // ignore
209
+ }
210
+ }
211
+ }
212
+
213
+ @Override
214
+ public TaskReport run(TaskSource taskSource,
215
+ Schema schema, int taskIndex,
216
+ PageOutput output)
217
+ {
218
+ PluginTask task = taskSource.loadTask(PluginTask.class);
219
+ SetupTask setup = task.getSetupConfig().loadConfig(SetupTask.class);
220
+
221
+ Path tempDir = Paths.get(task.getTempDir());
222
+ Path setupPath = resolveSetupPath(tempDir);
223
+
224
+ TimestampParser[] timestampParsers =
225
+ Timestamps.newTimestampColumnParsers(setup, setup.getSchemaConfig());
226
+
227
+ Path outPath = resolveOutPath(tempDir, taskIndex);
228
+ int ecode;
229
+ try {
230
+ // $ <command> run /tmp/setup.yml <output-taskIndex.csv> <taskIndex>
231
+ if (task.getTryNamedPipe() && tryToCreateNamedPipe(outPath)) {
232
+ Process proc = startCommand(task.getCwd(), task.getEnv(), task.getCommand(),
233
+ ImmutableList.of(
234
+ "run",
235
+ setupPath.toString(),
236
+ outPath.toString(),
237
+ Integer.toString(taskIndex)));
238
+ try (InputStream in = Files.newInputStream(outPath)) {
239
+ readRecords(in, output, schema, taskIndex, timestampParsers);
240
+ } finally {
241
+ ecode = waitFor(proc);
242
+ }
243
+ } else {
244
+ Files.deleteIfExists(outPath);
245
+ ecode = runCommand(task.getCwd(), task.getEnv(), task.getCommand(),
246
+ ImmutableList.of(
247
+ "run",
248
+ setupPath.toString(),
249
+ outPath.toString(),
250
+ Integer.toString(taskIndex)));
251
+ try (InputStream in = Files.newInputStream(outPath)) {
252
+ readRecords(in, output, schema, taskIndex, timestampParsers);
253
+ }
254
+ }
255
+
256
+ if (ecode != 0) {
257
+ throw new ConfigException("Run command exited with error code " + ecode);
258
+ }
259
+ }
260
+ catch (IOException ex) {
261
+ throw new DataException("Failed to start run command", ex);
262
+ }
263
+
264
+ return Exec.newTaskReport();
265
+ }
266
+
267
+ @Override
268
+ public ConfigDiff guess(ConfigSource config)
269
+ {
270
+ return Exec.newConfigDiff();
271
+ }
272
+
273
+ private void writeConfigFile(Path path, ConfigSource config)
274
+ throws IOException
275
+ {
276
+ ModelManager modelManager = Exec.getModelManager();
277
+ Object object = modelManager.readObject(Object.class, modelManager.writeObject(config));
278
+ String yamlString = new Yaml().dump(object);
279
+ try (BufferedWriter writer = Files.newBufferedWriter(path, UTF_8)) {
280
+ writer.write(yamlString);
281
+ }
282
+ }
283
+
284
+ private ConfigSource readSetupFile(ConfigLoader configLoader, Path setupPath)
285
+ throws IOException
286
+ {
287
+ return configLoader.fromYamlFile(setupPath.toFile());
288
+ }
289
+
290
+ private Process startCommand(String cwd, Map<String, String> env,
291
+ String shellCommand, List<String> args)
292
+ throws IOException
293
+ {
294
+ logger.info("Running command: {} {}", shellCommand, args);
295
+
296
+ // $ sh -c "shellCommand \"@args\""
297
+ List<String> sh = new ArrayList<>();
298
+ sh.addAll(SHELL);
299
+ {
300
+ StringBuilder esc = new StringBuilder();
301
+ esc.append(shellCommand);
302
+ for (String cmd : args) {
303
+ esc.append(" ");
304
+ esc.append(SHELL_ESCAPER.escape(cmd));
305
+ }
306
+ sh.add(esc.toString());
307
+ }
308
+
309
+ ProcessBuilder pb = new ProcessBuilder(sh);
310
+ if (!cwd.equals("")) {
311
+ pb.directory(new File(cwd));
312
+ }
313
+ pb.environment().putAll(env);
314
+ pb.redirectErrorStream(true);
315
+ pb.redirectOutput(ProcessBuilder.Redirect.INHERIT);
316
+ pb.redirectInput(ProcessBuilder.Redirect.PIPE);
317
+
318
+ Process proc = pb.start();
319
+ proc.getOutputStream().close();
320
+
321
+ return proc;
322
+ }
323
+
324
+ private int runCommand(String cwd, Map<String, String> env,
325
+ String shellCommand, List<String> cmdline)
326
+ throws IOException
327
+ {
328
+ return waitFor(startCommand(cwd, env, shellCommand, cmdline));
329
+ }
330
+
331
+ private boolean tryToCreateNamedPipe(Path path)
332
+ {
333
+ try {
334
+ ProcessBuilder pb = new ProcessBuilder("mkfifo", path.toString());
335
+ pb.inheritIO();
336
+ Process proc = pb.start();
337
+ int ecode = waitFor(proc);
338
+ return ecode == 0;
339
+ }
340
+ catch (IOException ex) {
341
+ return false;
342
+ }
343
+ }
344
+
345
+ private int waitFor(Process proc)
346
+ {
347
+ try {
348
+ return proc.waitFor();
349
+ }
350
+ catch (InterruptedException ex) {
351
+ Thread.currentThread().interrupt();
352
+ throw new RuntimeException(ex);
353
+ }
354
+ }
355
+
356
+ private static final ImmutableSet<String> TRUE_STRINGS =
357
+ ImmutableSet.of(
358
+ "true", "True", "TRUE",
359
+ "yes", "Yes", "YES",
360
+ "t", "T", "y", "Y",
361
+ "on", "On",
362
+ "ON", "1");
363
+
364
+ private static DateTimeFormatter TIMESTAMP_PARSER = DateTimeFormatter.ISO_INSTANT;
365
+
366
+ static class CsvRecordValidateException extends DataException {
367
+ CsvRecordValidateException(String message) {
368
+ super(message);
369
+ }
370
+
371
+ CsvRecordValidateException(Throwable cause) {
372
+ super(cause);
373
+ }
374
+ }
375
+
376
+ private void readRecords(InputStream in, PageOutput output, Schema schema, int taskIndex, TimestampParser[] timestampParsers)
377
+ {
378
+ JsonParser jsonParser = new JsonParser();
379
+
380
+ try (InputStreamReader reader = new InputStreamReader(in, UTF_8); CSVReader csv = new CSVReaderBuilder(reader).withFieldAsNull(CSVReaderNullFieldIndicator.EMPTY_SEPARATORS).build()) {
381
+ Iterator<String[]> ite = csv.iterator();
382
+ try (PageBuilder pageBuilder = new PageBuilder(Exec.getBufferAllocator(), schema, output)) {
383
+ while (ite.hasNext()) {
384
+ readRecord(ite.next(), pageBuilder, schema, jsonParser, timestampParsers);
385
+ pageBuilder.addRecord();
386
+ }
387
+ pageBuilder.finish();
388
+ }
389
+
390
+ }
391
+ catch (IOException ex) {
392
+ throw new DataException("Failed to parse output file " + taskIndex, ex);
393
+ }
394
+ }
395
+
396
+ private void readRecord(String[] record, PageBuilder pageBuilder, Schema schema,
397
+ JsonParser jsonParser, TimestampParser[] timestampParsers)
398
+ {
399
+ if (record.length != schema.getColumnCount()) {
400
+ throw new CsvRecordValidateException("Invalid number of columns (expected " + schema.getColumnCount() + " but got " + record.length + ")");
401
+ }
402
+
403
+ schema.visitColumns(new ColumnVisitor() {
404
+ public void booleanColumn(Column column) {
405
+ String v = record[column.getIndex()];
406
+ if (v == null) {
407
+ pageBuilder.setNull(column);
408
+ } else {
409
+ pageBuilder.setBoolean(column, TRUE_STRINGS.contains(v));
410
+ }
411
+ }
412
+
413
+ public void longColumn(Column column) {
414
+ String v = record[column.getIndex()];
415
+ if (v == null) {
416
+ pageBuilder.setNull(column);
417
+ } else {
418
+ try {
419
+ pageBuilder.setLong(column, Long.parseLong(v));
420
+ } catch (NumberFormatException e) {
421
+ throw new CsvRecordValidateException(e);
422
+ }
423
+ }
424
+ }
425
+
426
+ public void doubleColumn(Column column) {
427
+ String v = record[column.getIndex()];
428
+ if (v == null) {
429
+ pageBuilder.setNull(column);
430
+ } else {
431
+ try {
432
+ pageBuilder.setDouble(column, Double.parseDouble(v));
433
+ } catch (NumberFormatException e) {
434
+ throw new CsvRecordValidateException(e);
435
+ }
436
+ }
437
+ }
438
+
439
+ public void stringColumn(Column column) {
440
+ String v = record[column.getIndex()];
441
+ if (v == null) {
442
+ pageBuilder.setNull(column);
443
+ } else {
444
+ pageBuilder.setString(column, v);
445
+ }
446
+ }
447
+
448
+ public void timestampColumn(Column column) {
449
+ String v = record[column.getIndex()];
450
+ if (v == null) {
451
+ pageBuilder.setNull(column);
452
+ } else {
453
+ try {
454
+ pageBuilder.setTimestamp(column, timestampParsers[column.getIndex()].parse(v));
455
+ } catch (TimestampParseException e) {
456
+ throw new CsvRecordValidateException(e);
457
+ }
458
+ }
459
+ }
460
+
461
+ public void jsonColumn(Column column) {
462
+ String v = record[column.getIndex()];
463
+ if (v == null) {
464
+ pageBuilder.setNull(column);
465
+ } else {
466
+ try {
467
+ pageBuilder.setJson(column, jsonParser.parse(v));
468
+ } catch (JsonParseException e) {
469
+ throw new CsvRecordValidateException(e);
470
+ }
471
+ }
472
+ }
473
+ });
474
+ }
475
+ }
@@ -0,0 +1,5 @@
1
+ package org.embulk.input.script;
2
+
3
+ public class TestScriptInputPlugin
4
+ {
5
+ }
metadata ADDED
@@ -0,0 +1,95 @@
1
+ --- !ruby/object:Gem::Specification
2
+ name: embulk-input-script
3
+ version: !ruby/object:Gem::Version
4
+ version: 0.1.0
5
+ platform: ruby
6
+ authors:
7
+ - Sadayuki Furuhashi
8
+ autorequire:
9
+ bindir: bin
10
+ cert_chain: []
11
+ date: 2019-02-15 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: '12.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: '12.0'
41
+ description: Loads records from Script.
42
+ email:
43
+ - frsyuki@gmail.com
44
+ executables: []
45
+ extensions: []
46
+ extra_rdoc_files: []
47
+ files:
48
+ - ".gitignore"
49
+ - LICENSE.txt
50
+ - README.md
51
+ - build.gradle
52
+ - classpath/commons-beanutils-1.9.3.jar
53
+ - classpath/commons-collections-3.2.2.jar
54
+ - classpath/commons-collections4-4.2.jar
55
+ - classpath/commons-lang3-3.8.1.jar
56
+ - classpath/commons-logging-1.2.jar
57
+ - classpath/commons-text-1.3.jar
58
+ - classpath/embulk-input-script-0.1.0.jar
59
+ - classpath/opencsv-4.5.jar
60
+ - config/checkstyle/checkstyle.xml
61
+ - config/checkstyle/default.xml
62
+ - example/config.yml
63
+ - example/script.rb
64
+ - gradle/wrapper/gradle-wrapper.jar
65
+ - gradle/wrapper/gradle-wrapper.properties
66
+ - gradlew
67
+ - gradlew.bat
68
+ - lib/embulk/input/script.rb
69
+ - src/main/java/org/embulk/input/script/ScriptInputPlugin.java
70
+ - src/test/java/org/embulk/input/script/TestScriptInputPlugin.java
71
+ homepage:
72
+ licenses:
73
+ - MIT
74
+ metadata: {}
75
+ post_install_message:
76
+ rdoc_options: []
77
+ require_paths:
78
+ - lib
79
+ required_ruby_version: !ruby/object:Gem::Requirement
80
+ requirements:
81
+ - - ">="
82
+ - !ruby/object:Gem::Version
83
+ version: '0'
84
+ required_rubygems_version: !ruby/object:Gem::Requirement
85
+ requirements:
86
+ - - ">="
87
+ - !ruby/object:Gem::Version
88
+ version: '0'
89
+ requirements: []
90
+ rubyforge_project:
91
+ rubygems_version: 2.6.8
92
+ signing_key:
93
+ specification_version: 4
94
+ summary: Script input plugin for Embulk
95
+ test_files: []