commit 1733c32c41df9ed66facce2ae248cb6c718451af Author: Torma Kristóf Date: Fri May 22 02:40:51 2020 +0200 initial commit diff --git a/.drone.yml b/.drone.yml new file mode 100644 index 0000000..a53f2fe --- /dev/null +++ b/.drone.yml @@ -0,0 +1,58 @@ +kind: pipeline +type: docker +name: default + +steps: +- name: restore-cache-with-filesystem + image: meltwater/drone-cache + settings: + backend: "filesystem" + restore: true + cache_key: "{{ .Repo.Name }}" + archive_format: "gzip" + filesystem_cache_root: "/tmp/cache" + mount: + - 'build' + - '.gradle' + volumes: + - name: cache + path: /tmp/cache + +- name: build_application + image: openjdk:11-jdk + commands: + - ./gradlew build -x test + +- name: rebuild-cache-with-filesystem + image: meltwater/drone-cache:dev + pull: true + settings: + backend: "filesystem" + rebuild: true + cache_key: "{{ .Repo.Name }}" + archive_format: "gzip" + filesystem_cache_root: "/tmp/cache" + mount: + - 'build' + - '.gradle' + volumes: + - name: cache + path: /tmp/cache + +- name: kaniko + image: banzaicloud/drone-kaniko + settings: + registry: registry.kmlabz.com + repo: tormakris/${DRONE_REPO_NAME} + username: + from_secret: DOCKER_USERNAME + password: + from_secret: DOCKER_PASSWORD + tags: + - latest + - ${DRONE_BUILD_NUMBER} + +volumes: +- name: cache + host: + path: "/tmp/cache" diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..9affc3d --- /dev/null +++ b/.gitignore @@ -0,0 +1,8 @@ +/.gradle +/.idea +/out +/build +*.iml +*.ipr +*.iws +*.log \ No newline at end of file diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..739dfc9 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,14 @@ +FROM openjdk:11-jre + +ENV APPLICATION_USER ktor +RUN useradd $APPLICATION_USER + +RUN mkdir /app +RUN chown -R $APPLICATION_USER /app + +USER $APPLICATION_USER + +COPY ./build/libs/messagequeue-glue-1.0-SNAPSHOT.jar /app/messagequeue-glue-1.0-SNAPSHOT.jar +WORKDIR /app + +CMD ["java", "-server", "-XX:+UnlockExperimentalVMOptions", "-XX:InitialRAMFraction=2", "-XX:MinRAMFraction=2", "-XX:MaxRAMFraction=2", "-XX:+UseG1GC", "-XX:MaxGCPauseMillis=100", "-XX:+UseStringDeduplication", "-jar", "input-service.jar"] \ No newline at end of file diff --git a/build.gradle.kts b/build.gradle.kts new file mode 100644 index 0000000..4c1b196 --- /dev/null +++ b/build.gradle.kts @@ -0,0 +1,38 @@ +buildscript { + repositories { + gradlePluginPortal() + } + dependencies { + classpath("com.github.jengelman.gradle.plugins:shadow:5.2.0") + } +} + +plugins { + kotlin("jvm") version "1.3.72" +} + +group = "com.kmlabz" +version = "1.0-SNAPSHOT" + +repositories { + mavenCentral() + jcenter() +} + +apply(plugin= "com.github.johnrengelman.shadow") + +dependencies { + implementation(kotlin("stdlib-jdk8")) + implementation("com.rabbitmq:amqp-client:2.7.1") + implementation("com.viartemev:the-white-rabbit:0.0.5") + implementation("com.google.code.gson:gson:2.8.6") +} + +tasks { + compileKotlin { + kotlinOptions.jvmTarget = "11" + } + compileTestKotlin { + kotlinOptions.jvmTarget = "11" + } +} \ No newline at end of file diff --git a/docker-compose.yml b/docker-compose.yml new file mode 100644 index 0000000..692d819 --- /dev/null +++ b/docker-compose.yml @@ -0,0 +1,75 @@ +version: '3' + +services: + + output-service-postgres: + image: "postgres:12" + restart: "always" + volumes: + - "ktor-data:/var/lib/postgresql/data" + ports: + - "127.0.0.1:5433:5432" + environment: + POSTGRES_USER: "output-service-rdb" + POSTGRES_PASSWORD: "output-service-rdb" + POSTGRES_DB: "output-service-rdb" + + output-service-rdb: + image: "registry.kmlabz.com/birbnetes/output-service-rdb" + restart: "always" + ports: + - "127.0.0.1:8081:8080" + environment: + DB_USER: "output-service-rdb" + DB_PASSWORD: "output-service-rdb" + DB_URL: "jdbc:postgresql://output-service-postgres:5432/output-service-rdb" + MQ_HOST: rabbitmq + MQ_USERNAME: rabbitmq + MQ_PASSWORD: rabbitmq + depends_on: + - rabbitmq + - output-service-postgres + + input-service-postgres: + image: "postgres:12" + restart: "always" + volumes: + - "input-ktor-data:/var/lib/postgresql/data" + ports: + - "127.0.0.1:5432:5432" + environment: + POSTGRES_USER: "input-service" + POSTGRES_PASSWORD: "input-service" + POSTGRES_DB: "input-service" + + input-service: + image: "registry.kmlabz.com/tormakris/input-service" + restart: "always" + ports: + - "127.0.0.1:8080:8080" + environment: + DB_USER: "input-service" + DB_PASSWORD: "input-service" + DB_URL: "jdbc:postgresql://input-service-postgres:5432/input-service" + MQ_HOST: rabbitmq + MQ_USERNAME: rabbitmq + MQ_PASSWORD: rabbitmq + depends_on: + - rabbitmq + - input-service-postgres + + rabbitmq: + image: "rabbitmq:3-management" + hostname: "rabbitmq" + environment: + RABBITMQ_ERLANG_COOKIE: "SWQOKODSQALRPCLNMEQG" + RABBITMQ_DEFAULT_USER: "rabbitmq" + RABBITMQ_DEFAULT_PASS: "rabbitmq" + RABBITMQ_DEFAULT_VHOST: "/" + ports: + - "127.0.0.1:15672:15672" + - "127.0.0.1:5672:5672" + +volumes: + input-ktor-data: + ktor-data: \ No newline at end of file diff --git a/gradle.properties b/gradle.properties new file mode 100644 index 0000000..29e08e8 --- /dev/null +++ b/gradle.properties @@ -0,0 +1 @@ +kotlin.code.style=official \ No newline at end of file diff --git a/gradle/wrapper/gradle-wrapper.jar b/gradle/wrapper/gradle-wrapper.jar new file mode 100644 index 0000000..f3d88b1 Binary files /dev/null and b/gradle/wrapper/gradle-wrapper.jar differ diff --git a/gradle/wrapper/gradle-wrapper.properties b/gradle/wrapper/gradle-wrapper.properties new file mode 100644 index 0000000..ba94df8 --- /dev/null +++ b/gradle/wrapper/gradle-wrapper.properties @@ -0,0 +1,5 @@ +distributionBase=GRADLE_USER_HOME +distributionPath=wrapper/dists +distributionUrl=https\://services.gradle.org/distributions/gradle-6.1-bin.zip +zipStoreBase=GRADLE_USER_HOME +zipStorePath=wrapper/dists diff --git a/gradlew b/gradlew new file mode 100755 index 0000000..2fe81a7 --- /dev/null +++ b/gradlew @@ -0,0 +1,183 @@ +#!/usr/bin/env sh + +# +# Copyright 2015 the original author or authors. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# https://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# + +############################################################################## +## +## Gradle start up script for UN*X +## +############################################################################## + +# Attempt to set APP_HOME +# Resolve links: $0 may be a link +PRG="$0" +# Need this for relative symlinks. +while [ -h "$PRG" ] ; do + ls=`ls -ld "$PRG"` + link=`expr "$ls" : '.*-> \(.*\)$'` + if expr "$link" : '/.*' > /dev/null; then + PRG="$link" + else + PRG=`dirname "$PRG"`"/$link" + fi +done +SAVED="`pwd`" +cd "`dirname \"$PRG\"`/" >/dev/null +APP_HOME="`pwd -P`" +cd "$SAVED" >/dev/null + +APP_NAME="Gradle" +APP_BASE_NAME=`basename "$0"` + +# Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script. +DEFAULT_JVM_OPTS='"-Xmx64m" "-Xms64m"' + +# Use the maximum available, or set MAX_FD != -1 to use that value. +MAX_FD="maximum" + +warn () { + echo "$*" +} + +die () { + echo + echo "$*" + echo + exit 1 +} + +# OS specific support (must be 'true' or 'false'). +cygwin=false +msys=false +darwin=false +nonstop=false +case "`uname`" in + CYGWIN* ) + cygwin=true + ;; + Darwin* ) + darwin=true + ;; + MINGW* ) + msys=true + ;; + NONSTOP* ) + nonstop=true + ;; +esac + +CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar + +# Determine the Java command to use to start the JVM. +if [ -n "$JAVA_HOME" ] ; then + if [ -x "$JAVA_HOME/jre/sh/java" ] ; then + # IBM's JDK on AIX uses strange locations for the executables + JAVACMD="$JAVA_HOME/jre/sh/java" + else + JAVACMD="$JAVA_HOME/bin/java" + fi + if [ ! -x "$JAVACMD" ] ; then + die "ERROR: JAVA_HOME is set to an invalid directory: $JAVA_HOME + +Please set the JAVA_HOME variable in your environment to match the +location of your Java installation." + fi +else + JAVACMD="java" + which java >/dev/null 2>&1 || die "ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH. + +Please set the JAVA_HOME variable in your environment to match the +location of your Java installation." +fi + +# Increase the maximum file descriptors if we can. +if [ "$cygwin" = "false" -a "$darwin" = "false" -a "$nonstop" = "false" ] ; then + MAX_FD_LIMIT=`ulimit -H -n` + if [ $? -eq 0 ] ; then + if [ "$MAX_FD" = "maximum" -o "$MAX_FD" = "max" ] ; then + MAX_FD="$MAX_FD_LIMIT" + fi + ulimit -n $MAX_FD + if [ $? -ne 0 ] ; then + warn "Could not set maximum file descriptor limit: $MAX_FD" + fi + else + warn "Could not query maximum file descriptor limit: $MAX_FD_LIMIT" + fi +fi + +# For Darwin, add options to specify how the application appears in the dock +if $darwin; then + GRADLE_OPTS="$GRADLE_OPTS \"-Xdock:name=$APP_NAME\" \"-Xdock:icon=$APP_HOME/media/gradle.icns\"" +fi + +# For Cygwin or MSYS, switch paths to Windows format before running java +if [ "$cygwin" = "true" -o "$msys" = "true" ] ; then + APP_HOME=`cygpath --path --mixed "$APP_HOME"` + CLASSPATH=`cygpath --path --mixed "$CLASSPATH"` + JAVACMD=`cygpath --unix "$JAVACMD"` + + # We build the pattern for arguments to be converted via cygpath + ROOTDIRSRAW=`find -L / -maxdepth 1 -mindepth 1 -type d 2>/dev/null` + SEP="" + for dir in $ROOTDIRSRAW ; do + ROOTDIRS="$ROOTDIRS$SEP$dir" + SEP="|" + done + OURCYGPATTERN="(^($ROOTDIRS))" + # Add a user-defined pattern to the cygpath arguments + if [ "$GRADLE_CYGPATTERN" != "" ] ; then + OURCYGPATTERN="$OURCYGPATTERN|($GRADLE_CYGPATTERN)" + fi + # Now convert the arguments - kludge to limit ourselves to /bin/sh + i=0 + for arg in "$@" ; do + CHECK=`echo "$arg"|egrep -c "$OURCYGPATTERN" -` + CHECK2=`echo "$arg"|egrep -c "^-"` ### Determine if an option + + if [ $CHECK -ne 0 ] && [ $CHECK2 -eq 0 ] ; then ### Added a condition + eval `echo args$i`=`cygpath --path --ignore --mixed "$arg"` + else + eval `echo args$i`="\"$arg\"" + fi + i=`expr $i + 1` + done + case $i in + 0) set -- ;; + 1) set -- "$args0" ;; + 2) set -- "$args0" "$args1" ;; + 3) set -- "$args0" "$args1" "$args2" ;; + 4) set -- "$args0" "$args1" "$args2" "$args3" ;; + 5) set -- "$args0" "$args1" "$args2" "$args3" "$args4" ;; + 6) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" ;; + 7) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" ;; + 8) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" ;; + 9) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" "$args8" ;; + esac +fi + +# Escape application args +save () { + for i do printf %s\\n "$i" | sed "s/'/'\\\\''/g;1s/^/'/;\$s/\$/' \\\\/" ; done + echo " " +} +APP_ARGS=`save "$@"` + +# Collect all arguments for the java command, following the shell quoting and substitution rules +eval set -- $DEFAULT_JVM_OPTS $JAVA_OPTS $GRADLE_OPTS "\"-Dorg.gradle.appname=$APP_BASE_NAME\"" -classpath "\"$CLASSPATH\"" org.gradle.wrapper.GradleWrapperMain "$APP_ARGS" + +exec "$JAVACMD" "$@" diff --git a/gradlew.bat b/gradlew.bat new file mode 100644 index 0000000..24467a1 --- /dev/null +++ b/gradlew.bat @@ -0,0 +1,100 @@ +@rem +@rem Copyright 2015 the original author or authors. +@rem +@rem Licensed under the Apache License, Version 2.0 (the "License"); +@rem you may not use this file except in compliance with the License. +@rem You may obtain a copy of the License at +@rem +@rem https://www.apache.org/licenses/LICENSE-2.0 +@rem +@rem Unless required by applicable law or agreed to in writing, software +@rem distributed under the License is distributed on an "AS IS" BASIS, +@rem WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +@rem See the License for the specific language governing permissions and +@rem limitations under the License. +@rem + +@if "%DEBUG%" == "" @echo off +@rem ########################################################################## +@rem +@rem Gradle startup script for Windows +@rem +@rem ########################################################################## + +@rem Set local scope for the variables with windows NT shell +if "%OS%"=="Windows_NT" setlocal + +set DIRNAME=%~dp0 +if "%DIRNAME%" == "" set DIRNAME=. +set APP_BASE_NAME=%~n0 +set APP_HOME=%DIRNAME% + +@rem Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script. +set DEFAULT_JVM_OPTS="-Xmx64m" "-Xms64m" + +@rem Find java.exe +if defined JAVA_HOME goto findJavaFromJavaHome + +set JAVA_EXE=java.exe +%JAVA_EXE% -version >NUL 2>&1 +if "%ERRORLEVEL%" == "0" goto init + +echo. +echo ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH. +echo. +echo Please set the JAVA_HOME variable in your environment to match the +echo location of your Java installation. + +goto fail + +:findJavaFromJavaHome +set JAVA_HOME=%JAVA_HOME:"=% +set JAVA_EXE=%JAVA_HOME%/bin/java.exe + +if exist "%JAVA_EXE%" goto init + +echo. +echo ERROR: JAVA_HOME is set to an invalid directory: %JAVA_HOME% +echo. +echo Please set the JAVA_HOME variable in your environment to match the +echo location of your Java installation. + +goto fail + +:init +@rem Get command-line arguments, handling Windows variants + +if not "%OS%" == "Windows_NT" goto win9xME_args + +:win9xME_args +@rem Slurp the command line arguments. +set CMD_LINE_ARGS= +set _SKIP=2 + +:win9xME_args_slurp +if "x%~1" == "x" goto execute + +set CMD_LINE_ARGS=%* + +:execute +@rem Setup the command line + +set CLASSPATH=%APP_HOME%\gradle\wrapper\gradle-wrapper.jar + +@rem Execute Gradle +"%JAVA_EXE%" %DEFAULT_JVM_OPTS% %JAVA_OPTS% %GRADLE_OPTS% "-Dorg.gradle.appname=%APP_BASE_NAME%" -classpath "%CLASSPATH%" org.gradle.wrapper.GradleWrapperMain %CMD_LINE_ARGS% + +:end +@rem End local scope for the variables with windows NT shell +if "%ERRORLEVEL%"=="0" goto mainEnd + +:fail +rem Set variable GRADLE_EXIT_CONSOLE if you need the _script_ return code instead of +rem the _cmd.exe /c_ return code! +if not "" == "%GRADLE_EXIT_CONSOLE%" exit 1 +exit /b 1 + +:mainEnd +if "%OS%"=="Windows_NT" endlocal + +:omega diff --git a/settings.gradle.kts b/settings.gradle.kts new file mode 100644 index 0000000..c8ff21c --- /dev/null +++ b/settings.gradle.kts @@ -0,0 +1,2 @@ +rootProject.name = "messagequeue-glue" + diff --git a/src/main/kotlin/Application.kt b/src/main/kotlin/Application.kt new file mode 100644 index 0000000..c3bcca3 --- /dev/null +++ b/src/main/kotlin/Application.kt @@ -0,0 +1,6 @@ +import worker.Glue + +fun main(args: Array) { + val glue = Glue() + glue.recieve() +} \ No newline at end of file diff --git a/src/main/kotlin/model/OutputModel.kt b/src/main/kotlin/model/OutputModel.kt new file mode 100644 index 0000000..646f252 --- /dev/null +++ b/src/main/kotlin/model/OutputModel.kt @@ -0,0 +1,8 @@ +package model + +import com.google.gson.annotations.SerializedName + +data class OutputModel ( + @SerializedName("tag") val tag: String, + @SerializedName("probability") val probability: Double +) \ No newline at end of file diff --git a/src/main/kotlin/worker/Glue.kt b/src/main/kotlin/worker/Glue.kt new file mode 100644 index 0000000..69149e2 --- /dev/null +++ b/src/main/kotlin/worker/Glue.kt @@ -0,0 +1,54 @@ +package worker + +import com.google.gson.Gson +import com.rabbitmq.client.BuiltinExchangeType +import com.rabbitmq.client.ConnectionFactory +import model.OutputModel +import com.viartemev.thewhiterabbit.channel.channel +import com.viartemev.thewhiterabbit.channel.consume +import kotlinx.coroutines.Dispatchers +import kotlinx.coroutines.GlobalScope +import kotlinx.coroutines.launch +import kotlin.random.Random + +class Glue { + var mqHost: String = System.getenv("MQ_HOST") ?: "localhost" + var mqUserName: String = System.getenv("MQ_USERNAME") ?: "rabbitmq" + var mqPassWord: String = System.getenv("MQ_PASSWORD") ?: "rabbitmq" + var mqInputExchange: String = System.getenv("MQ_IN_EXCHANGE") ?: "input" + var mqOutputExchange: String = System.getenv("MQ_OUT_EXCHANGE") ?: "output" + val gson = Gson() + fun recieve(){ + val factory = ConnectionFactory() + factory.host = mqHost + factory.username = mqUserName + factory.password = mqPassWord + + val inputConnection = factory.newConnection() + val inputChannel = inputConnection.createChannel() + + inputChannel.exchangeDeclare(mqInputExchange, BuiltinExchangeType.FANOUT) + val inputQueueName = inputChannel.queueDeclare().queue + inputChannel.queueBind(inputQueueName, mqInputExchange, "") + + val outputConnection = factory.newConnection() + val outputChannel = outputConnection.createChannel() + + outputChannel.exchangeDeclare(mqOutputExchange, BuiltinExchangeType.FANOUT) + val outputQueueName = outputChannel.queueDeclare().queue + outputChannel.queueBind(outputQueueName, mqOutputExchange, "") + + GlobalScope.launch(Dispatchers.Default) { + inputConnection.channel { + consume(inputQueueName) { + consumeMessageWithConfirm({ + val rawJson = String(it.body) + val probability = Random.nextDouble(0.0,1.0) + val outputObject = OutputModel(tag=rawJson, probability=probability) + outputChannel.basicPublish("",outputQueueName, null, gson.toJson(outputObject).toByteArray()) + }) + } + } + } + } +} \ No newline at end of file diff --git a/test.py b/test.py new file mode 100644 index 0000000..b3f0a78 --- /dev/null +++ b/test.py @@ -0,0 +1,18 @@ +#!/usr/bin/env python3 +import requests +import os.path +import json + +URL = "http://127.0.0.1:8080/sample" + +FILE = 'CommonStarling_102761_45.wav' + +files = { + "file": (os.path.basename(FILE), open(FILE,'rb').read(), 'audio/wave', {'Content-length' : os.path.getsize(FILE)}), + "description" : (None, json.dumps({"device_id" : "123"}), "application/json") +} + +r = requests.post(URL,files=files) +print("Content: ", r.content) +print("Headers:", r.headers) +r.raise_for_status()