embulk-output-mongodb_nest 0.1.0

Sign up to get free protection for your applications and to get access to all the features.
checksums.yaml ADDED
@@ -0,0 +1,7 @@
1
+ ---
2
+ SHA1:
3
+ metadata.gz: 68b073414092445e8f8ed206b5869923cff6dadd
4
+ data.tar.gz: a5e314633418c5531bdd7b82804aff8a93bea542
5
+ SHA512:
6
+ metadata.gz: 62a6bde8df8d222458a9c8b98f1510b0d3a9a74c5a0a6cc722a34812f62074d9e2c4eac998931f54d28125446760c5f5884708e3c569b34dcb5511d1ec9f29fa
7
+ data.tar.gz: e68632825b0249b9fb7197f92af10c538526ca430e261fe11ec11e2ccd9ba14310d096ad9c8cc6981c852feec92eba8f831b7dae334f1ec1123b109e04e1d669
data/.gitignore ADDED
@@ -0,0 +1,12 @@
1
+ *~
2
+ /pkg/
3
+ /tmp/
4
+ *.gemspec
5
+ .gradle/
6
+ /classpath/
7
+ build/
8
+ .idea
9
+ /.settings/
10
+ /.metadata/
11
+ .classpath
12
+ .project
data/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,86 @@
1
+ # Mongodb Nest output plugin for Embulk
2
+
3
+ input source중에 지정된 필드를 subdocument로 만들어서 몽고DB로 저장한다
4
+
5
+ Dumps records to Mongodb with subdocument
6
+
7
+ ## Overview
8
+
9
+ * **Plugin type**: output
10
+ * **Load all or nothing**: no
11
+ * **Resume supported**: no
12
+ * **Cleanup supported**: no
13
+
14
+ ## Configuration
15
+
16
+ - **type**: mongodb_nest (string, required)
17
+ - **host**: host name (string, required)
18
+ - **port**: port number (integer, default: `27017`)
19
+ - **database**: database name (string, required)
20
+ - **user**: mongodb account (string, required)
21
+ - **password**: password (string, required)
22
+ - **collection**: collection name (string, required)
23
+ - **key**: primary key of collection (string, required)
24
+ - **child**: describe subdocument (object list, default: null)
25
+ - **bulk_size**: bulk upsert size at time (integer, default: `10000`)
26
+
27
+
28
+ ## Example
29
+
30
+ * default configuration
31
+ ```yaml
32
+ out:
33
+ type: mongodb_nest
34
+ host: your-host-name
35
+ database: your-database-name
36
+ user: your-account-name
37
+ password: your-password
38
+ collection: your-collection-name
39
+ key: your-key
40
+ child:
41
+ - {name: mychild, field: time}
42
+ - {name: yourchild, field: comment}
43
+ - {name: mychild, field: purchase}
44
+ ```
45
+
46
+ * Embulk input csv sample configuration
47
+ ```yaml
48
+ in:
49
+ type: file
50
+ path_prefix: /Users/focuschange/google/program/embulk/try1/csv/sample_
51
+ decoders:
52
+ - {type: gzip}
53
+ parser:
54
+ charset: UTF-8
55
+ newline: LF
56
+ type: csv
57
+ delimiter: ','
58
+ quote: '"'
59
+ escape: '"'
60
+ null_string: 'NULL'
61
+ trim_if_not_quoted: false
62
+ skip_header_lines: 1
63
+ allow_extra_columns: false
64
+ allow_optional_columns: false
65
+ columns:
66
+ - {name: id, type: long}
67
+ - {name: account, type: long}
68
+ - {name: time, type: timestamp, format: '%Y-%m-%d %H:%M:%S'}
69
+ - {name: purchase, type: timestamp, format: '%Y%m%d'}
70
+ - {name: comment, type: string}
71
+ out:
72
+ type: mongodb_nest
73
+ host: your-host-name
74
+ database: your-database-name
75
+ user: your-account-name
76
+ password: your-password
77
+ collection: your-collection-name
78
+ key: your-key
79
+ child:
80
+ - {name: mychild, field: time}
81
+ - {name: yourchild, field: comment}
82
+ - {name: mychild, field: purchase}
83
+ ```
84
+
85
+
86
+
data/build.gradle ADDED
@@ -0,0 +1,97 @@
1
+ plugins {
2
+ id "com.jfrog.bintray" version "1.1"
3
+ id "com.github.jruby-gradle.base" version "0.1.5"
4
+ id "java"
5
+ 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 "org.embulk:embulk-core:0.9.7"
23
+ provided "org.embulk:embulk-core:0.9.7"
24
+ compile "org.mongodb:mongo-java-driver:3.7.0"
25
+
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 "-rrubygems/gem_runner", "-eGem::GemRunner.new.run(ARGV)", "build"
55
+ script "${project.name}.gemspec"
56
+ doLast { ant.move(file: "${project.name}-${project.version}.gem", todir: "pkg") }
57
+ }
58
+
59
+ task gemPush(type: JRubyExec, dependsOn: ["gem"]) {
60
+ jrubyArgs "-rrubygems/gem_runner", "-eGem::GemRunner.new.run(ARGV)", "push"
61
+ script "pkg/${project.name}-${project.version}.gem"
62
+ }
63
+
64
+ task "package"(dependsOn: ["gemspec", "classpath"]) {
65
+ doLast {
66
+ println "> Build succeeded."
67
+ println "> You can run embulk with '-L ${file(".").absolutePath}' argument."
68
+ }
69
+ }
70
+
71
+ task gemspec {
72
+ ext.gemspecFile = file("${project.name}.gemspec")
73
+ inputs.file "build.gradle"
74
+ outputs.file gemspecFile
75
+ doLast { gemspecFile.write($/
76
+ Gem::Specification.new do |spec|
77
+ spec.name = "${project.name}"
78
+ spec.version = "${project.version}"
79
+ spec.authors = ["focuschange"]
80
+ spec.summary = %[Mongodb Nest output plugin for Embulk]
81
+ spec.description = %[Dumps records to Mongodb with subdocument]
82
+ spec.email = ["focuschange@gmail.com"]
83
+ spec.licenses = ["MIT"]
84
+ spec.homepage = "https://github.com/focuschange/embulk-output-mongodb_nest"
85
+
86
+ spec.files = `git ls-files`.split("\n") + Dir["classpath/*.jar"]
87
+ spec.test_files = spec.files.grep(%r"^(test|spec)/")
88
+ spec.require_paths = ["lib"]
89
+
90
+ #spec.add_dependency 'YOUR_GEM_DEPENDENCY', ['~> YOUR_GEM_DEPENDENCY_VERSION']
91
+ spec.add_development_dependency 'bundler', ['~> 1.0']
92
+ spec.add_development_dependency 'rake', ['>= 10.0']
93
+ end
94
+ /$)
95
+ }
96
+ }
97
+ 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>
Binary file
@@ -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" "$@"
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,3 @@
1
+ Embulk::JavaPlugin.register_output(
2
+ "mongodb_nest", "org.embulk.output.mongodb_nest.MongodbNestOutputPlugin",
3
+ File.expand_path('../../../../classpath', __FILE__))
@@ -0,0 +1,98 @@
1
+ package org.embulk.output.mongodb_nest;
2
+ import java.util.List;
3
+ import com.google.common.base.Optional;
4
+ import org.embulk.config.Config;
5
+ import org.embulk.config.ConfigDefault;
6
+ import org.embulk.config.ConfigDiff;
7
+ import org.embulk.config.ConfigSource;
8
+ import org.embulk.config.Task;
9
+ import org.embulk.config.TaskReport;
10
+ import org.embulk.config.TaskSource;
11
+ import org.embulk.spi.*;
12
+ import org.slf4j.Logger;
13
+
14
+ /**
15
+ * <pre>
16
+ *
17
+ * </pre>
18
+ *
19
+ * @Author : 이상호(focuschange@gmail.com)
20
+ * @Date : 2018. 5. 28
21
+ * @Version : 1.0
22
+ * @see :
23
+ */
24
+ public class MongodbNestOutputPlugin implements OutputPlugin
25
+ {
26
+ private static final Logger logger = Exec.getLogger(MongodbNestOutputPlugin.class);
27
+
28
+ public interface DefineChildDocument extends Task
29
+ {
30
+ @Config("name")
31
+ public String getName();
32
+
33
+ @Config("field")
34
+ public String getField();
35
+ }
36
+
37
+ public interface PluginTask extends Task
38
+ {
39
+ @Config("collection")
40
+ public String getCollection();
41
+
42
+ @Config("host")
43
+ public String getHost();
44
+
45
+ @Config("port")
46
+ @ConfigDefault("27017")
47
+ public int getPort();
48
+
49
+ @Config("database")
50
+ public String getDatabase();
51
+
52
+ @Config("user")
53
+ public String getUser();
54
+
55
+ @Config("password")
56
+ public String getPassword();
57
+
58
+ @Config("key")
59
+ public String getKey();
60
+
61
+ @Config("child")
62
+ @ConfigDefault("null")
63
+ public Optional<List<DefineChildDocument>> getChild();
64
+
65
+ @Config("bulk_size")
66
+ @ConfigDefault("1000")
67
+ public int getBulkSize();
68
+
69
+ }
70
+
71
+ @Override public ConfigDiff transaction(ConfigSource config, Schema schema, int taskCount, OutputPlugin.Control control)
72
+ {
73
+ PluginTask task = config.loadConfig(PluginTask.class);
74
+
75
+ // retryable (idempotent) output:
76
+ // return resume(task.dump(), schema, taskCount, control);
77
+
78
+ // non-retryable (non-idempotent) output:
79
+ control.run(task.dump());
80
+ return Exec.newConfigDiff();
81
+ }
82
+
83
+ @Override public ConfigDiff resume(TaskSource taskSource, Schema schema, int taskCount, OutputPlugin.Control control)
84
+ {
85
+ throw new UnsupportedOperationException("mongodb_nest output plugin does not support resuming");
86
+ }
87
+
88
+ @Override public void cleanup(TaskSource taskSource, Schema schema, int taskCount, List<TaskReport> successTaskReports)
89
+ {
90
+ }
91
+
92
+ @Override public TransactionalPageOutput open(TaskSource taskSource, Schema schema, int taskIndex)
93
+ {
94
+ PluginTask task = taskSource.loadTask(PluginTask.class);
95
+
96
+ return new PluginPageOutput(task, schema);
97
+ }
98
+ }
@@ -0,0 +1,180 @@
1
+ package org.embulk.output.mongodb_nest;
2
+
3
+ import com.mongodb.BasicDBObject;
4
+ import com.mongodb.client.MongoClient;
5
+ import com.mongodb.client.MongoClients;
6
+ import com.mongodb.client.MongoCollection;
7
+ import com.mongodb.client.MongoDatabase;
8
+ import com.mongodb.client.model.ReplaceOneModel;
9
+ import com.mongodb.client.model.ReplaceOptions;
10
+ import com.mongodb.client.model.WriteModel;
11
+ import org.bson.Document;
12
+ import org.embulk.config.TaskReport;
13
+ import org.embulk.spi.*;
14
+ import org.embulk.spi.time.Timestamp;
15
+ import org.slf4j.Logger;
16
+
17
+ import java.io.UnsupportedEncodingException;
18
+ import java.net.URLEncoder;
19
+ import java.util.ArrayList;
20
+ import java.util.Date;
21
+ import java.util.List;
22
+ import java.util.Map;
23
+
24
+ /**
25
+ * <pre>
26
+ *
27
+ * </pre>
28
+ *
29
+ * @Author : 이상호(focuschange@gmail.com)
30
+ * @Date : 2018. 5. 28
31
+ * @Version : 1.0
32
+ * @see :
33
+ */
34
+ public class PluginPageOutput implements TransactionalPageOutput
35
+ {
36
+ private static final Logger logger = Exec.getLogger(MongodbNestOutputPlugin.class);
37
+
38
+ private final MongodbNestOutputPlugin.PluginTask task;
39
+ private final Schema schema;
40
+ private final PageReader pageReader;
41
+
42
+ private MongoClient mongo;
43
+ private MongoDatabase db;
44
+ private MongoCollection<Document> collection;
45
+
46
+ PluginPageOutput(MongodbNestOutputPlugin.PluginTask task, Schema schema)
47
+ {
48
+ this.pageReader = new PageReader(schema);
49
+ this.schema = schema;
50
+ this.task = task;
51
+
52
+ String connectionStr = "mongodb://";
53
+
54
+ if (task.getUser() != null)
55
+ {
56
+ connectionStr += task.getUser();
57
+ try
58
+ {
59
+ connectionStr += ":" + URLEncoder.encode(task.getPassword(), "UTF-8") + "@";
60
+ }
61
+ catch (UnsupportedEncodingException e)
62
+ {
63
+ e.printStackTrace();
64
+ }
65
+ }
66
+
67
+ connectionStr += task.getHost() + ":" + task.getPort() + "/" + task.getDatabase();
68
+
69
+ this.mongo = MongoClients.create(connectionStr);
70
+ this.db = this.mongo.getDatabase(task.getDatabase());
71
+ this.collection = this.db.getCollection(task.getCollection());
72
+ }
73
+
74
+ @Override public void add(Page page)
75
+ {
76
+ pageReader.setPage(page);
77
+ List<WriteModel<Document>> replaceModel = new ArrayList<>();
78
+
79
+ while (pageReader.nextRecord())
80
+ {
81
+ BasicDBObject doc = new BasicDBObject();
82
+
83
+ for (int i = 0; i < schema.getColumnCount(); i++)
84
+ {
85
+ String t = schema.getColumnName(i);
86
+ Class<?> type = schema.getColumnType(i).getJavaType();
87
+
88
+ if (pageReader.isNull(i))
89
+ {
90
+ doc.append(t, null);
91
+ }
92
+ else if (type.equals(boolean.class))
93
+ {
94
+ doc.append(t, pageReader.getBoolean(i));
95
+ }
96
+ else if (type.equals(double.class))
97
+ {
98
+ doc.append(t, pageReader.getDouble(i));
99
+ }
100
+ else if (type.equals(long.class))
101
+ {
102
+ doc.append(t, pageReader.getLong(i));
103
+ }
104
+ else if (type.equals(String.class))
105
+ {
106
+ doc.append(t, pageReader.getString(i));
107
+ }
108
+ else if (type.equals(Timestamp.class))
109
+ {
110
+ doc.append(t, new java.sql.Timestamp(pageReader.getTimestamp(i).toEpochMilli()));
111
+ }
112
+ }
113
+
114
+ if (task.getChild() != null)
115
+ {
116
+ doc = transformDocument(doc);
117
+ }
118
+
119
+ replaceModel.add(new ReplaceOneModel<>(
120
+ new Document(task.getKey(), doc.get(task.getKey())),
121
+ new Document(doc),
122
+ new ReplaceOptions().upsert(true))
123
+ );
124
+
125
+ if(replaceModel.size() % task.getBulkSize() == 0)
126
+ {
127
+ collection.bulkWrite(replaceModel);
128
+ replaceModel.clear();
129
+ }
130
+ }
131
+
132
+ if (replaceModel.size() > 0)
133
+ {
134
+ collection.bulkWrite(replaceModel);
135
+ }
136
+
137
+ }
138
+
139
+ private BasicDBObject transformDocument(BasicDBObject doc)
140
+ {
141
+ for (MongodbNestOutputPlugin.DefineChildDocument cd : task.getChild().get())
142
+ {
143
+ String name = cd.getName();
144
+ String field = cd.getField();
145
+
146
+ BasicDBObject subdoc = new BasicDBObject();
147
+ Object exists = doc.get(name);
148
+
149
+ if (exists != null)
150
+ {
151
+ subdoc.putAll((Map) exists);
152
+ }
153
+
154
+ subdoc.append(field, doc.remove(field));
155
+ doc.append(name, subdoc);
156
+ }
157
+
158
+ return doc;
159
+ }
160
+
161
+ @Override public void finish()
162
+ {
163
+
164
+ }
165
+
166
+ @Override public void close()
167
+ {
168
+ this.mongo.close();
169
+ }
170
+
171
+ @Override public void abort()
172
+ {
173
+
174
+ }
175
+
176
+ @Override public TaskReport commit()
177
+ {
178
+ return null;
179
+ }
180
+ }
@@ -0,0 +1,5 @@
1
+ package org.embulk.output.mongodb_nest;
2
+
3
+ public class TestMongodbNestOutputPlugin
4
+ {
5
+ }
metadata ADDED
@@ -0,0 +1,88 @@
1
+ --- !ruby/object:Gem::Specification
2
+ name: embulk-output-mongodb_nest
3
+ version: !ruby/object:Gem::Version
4
+ version: 0.1.0
5
+ platform: ruby
6
+ authors:
7
+ - focuschange
8
+ autorequire:
9
+ bindir: bin
10
+ cert_chain: []
11
+ date: 2018-05-30 00:00:00.000000000 Z
12
+ dependencies:
13
+ - !ruby/object:Gem::Dependency
14
+ requirement: !ruby/object:Gem::Requirement
15
+ requirements:
16
+ - - ~>
17
+ - !ruby/object:Gem::Version
18
+ version: '1.0'
19
+ name: bundler
20
+ prerelease: false
21
+ type: :development
22
+ version_requirements: !ruby/object:Gem::Requirement
23
+ requirements:
24
+ - - ~>
25
+ - !ruby/object:Gem::Version
26
+ version: '1.0'
27
+ - !ruby/object:Gem::Dependency
28
+ requirement: !ruby/object:Gem::Requirement
29
+ requirements:
30
+ - - '>='
31
+ - !ruby/object:Gem::Version
32
+ version: '10.0'
33
+ name: rake
34
+ prerelease: false
35
+ type: :development
36
+ version_requirements: !ruby/object:Gem::Requirement
37
+ requirements:
38
+ - - '>='
39
+ - !ruby/object:Gem::Version
40
+ version: '10.0'
41
+ description: Dumps records to Mongodb with subdocument
42
+ email:
43
+ - focuschange@gmail.com
44
+ executables: []
45
+ extensions: []
46
+ extra_rdoc_files: []
47
+ files:
48
+ - .gitignore
49
+ - LICENSE.txt
50
+ - README.md
51
+ - build.gradle
52
+ - config/checkstyle/checkstyle.xml
53
+ - config/checkstyle/default.xml
54
+ - gradle/wrapper/gradle-wrapper.jar
55
+ - gradle/wrapper/gradle-wrapper.properties
56
+ - gradlew
57
+ - gradlew.bat
58
+ - lib/embulk/output/mongodb_nest.rb
59
+ - src/main/java/org/embulk/output/mongodb_nest/MongodbNestOutputPlugin.java
60
+ - src/main/java/org/embulk/output/mongodb_nest/PluginPageOutput.java
61
+ - src/test/java/org/embulk/output/mongodb_nest/TestMongodbNestOutputPlugin.java
62
+ - classpath/mongo-java-driver-3.7.0.jar
63
+ - classpath/embulk-output-mongodb_nest-0.1.0.jar
64
+ homepage: https://github.com/focuschange/embulk-output-mongodb_nest
65
+ licenses:
66
+ - MIT
67
+ metadata: {}
68
+ post_install_message:
69
+ rdoc_options: []
70
+ require_paths:
71
+ - lib
72
+ required_ruby_version: !ruby/object:Gem::Requirement
73
+ requirements:
74
+ - - '>='
75
+ - !ruby/object:Gem::Version
76
+ version: '0'
77
+ required_rubygems_version: !ruby/object:Gem::Requirement
78
+ requirements:
79
+ - - '>='
80
+ - !ruby/object:Gem::Version
81
+ version: '0'
82
+ requirements: []
83
+ rubyforge_project:
84
+ rubygems_version: 2.1.9
85
+ signing_key:
86
+ specification_version: 4
87
+ summary: Mongodb Nest output plugin for Embulk
88
+ test_files: []