embulk-input-script 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/.gitignore +12 -0
- data/LICENSE.txt +21 -0
- data/README.md +32 -0
- data/build.gradle +99 -0
- data/config/checkstyle/checkstyle.xml +128 -0
- data/config/checkstyle/default.xml +108 -0
- data/example/config.yml +7 -0
- data/example/script.rb +39 -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/script.rb +3 -0
- data/src/main/java/org/embulk/input/script/ScriptInputPlugin.java +475 -0
- data/src/test/java/org/embulk/input/script/TestScriptInputPlugin.java +5 -0
- metadata +95 -0
checksums.yaml
ADDED
|
@@ -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
|
data/.gitignore
ADDED
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,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
|
+
```
|
data/build.gradle
ADDED
|
@@ -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>
|
data/example/config.yml
ADDED
data/example/script.rb
ADDED
|
@@ -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
|
+
|
|
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,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
|
+
}
|
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: []
|