init repository

2023-07-07 02:29:45 +08:00 · 2023-07-07 02:29:45 +08:00 · fba46c4f54
commit fba46c4f54
25 changed files with 972 additions and 0 deletions
--- a/.gitignore
+++ b/.gitignore
@ -0,0 +1,38 @@
+HELP.md
+.gradle
+build/
+!gradle/wrapper/gradle-wrapper.jar
+!**/src/main/**/build/
+!**/src/test/**/build/
+
+### STS ###
+.apt_generated
+.classpath
+.factorypath
+.project
+.settings
+.springBeans
+.sts4-cache
+bin/
+!**/src/main/**/bin/
+!**/src/test/**/bin/
+
+### IntelliJ IDEA ###
+.idea
+*.iws
+*.iml
+*.ipr
+out/
+!**/src/main/**/out/
+!**/src/test/**/out/
+
+### NetBeans ###
+/nbproject/private/
+/nbbuild/
+/dist/
+/nbdist/
+/.nb-gradle/
+
+### VS Code ###
+.vscode/
+logs/
--- a/build.gradle.kts
+++ b/build.gradle.kts
@ -0,0 +1,68 @@
+import org.jetbrains.kotlin.gradle.tasks.KotlinCompile
+import org.springframework.boot.gradle.tasks.bundling.BootBuildImage
+import org.springframework.boot.gradle.tasks.run.BootRun
+
+plugins {
+    id("org.springframework.boot") version "3.1.1"
+    id("io.spring.dependency-management") version "1.1.0"
+    id("org.graalvm.buildtools.native") version "0.9.23"
+    kotlin("jvm") version "1.8.22"
+    kotlin("plugin.spring") version "1.8.22"
+}
+
+group = "cn.edu.hfut.auto"
+version = "1.0.0"
+
+java {
+    sourceCompatibility = JavaVersion.VERSION_17
+}
+
+configurations {
+    compileOnly {
+        extendsFrom(configurations.annotationProcessor.get())
+    }
+    implementation {
+        exclude(group = "org.springframework.boot", module = "spring-boot-starter-logging")
+    }
+}
+
+repositories {
+    mavenCentral()
+}
+
+dependencies {
+    implementation("org.springframework.boot:spring-boot-starter-data-elasticsearch")
+    implementation("org.springframework.boot:spring-boot-starter-web")
+    implementation("com.fasterxml.jackson.module:jackson-module-kotlin")
+    // 日志
+    implementation("org.springframework.boot:spring-boot-starter-log4j2")
+    implementation("com.lmax:disruptor:3.4.4")
+    // kotlin 协程（spring 异步接口）
+    val kotlinxCoroutinesVersion = "1.7.1"
+    implementation("org.jetbrains.kotlinx:kotlinx-coroutines-jdk9:$kotlinxCoroutinesVersion")
+    implementation("org.jetbrains.kotlinx:kotlinx-coroutines-reactor:$kotlinxCoroutinesVersion")
+    implementation("org.jetbrains.kotlin:kotlin-reflect")
+//    implementation("org.springframework.kafka:spring-kafka")
+    annotationProcessor("org.springframework.boot:spring-boot-configuration-processor")
+    testImplementation("org.springframework.boot:spring-boot-starter-test")
+}
+
+tasks.withType<BootRun> {
+    @Suppress("SAFE_CALL_WILL_CHANGE_NULLABILITY", "UNNECESSARY_SAFE_CALL")
+    jvmArgs = jvmArgs?.toMutableList()?.plus("-Dlog4j.skipJansi=false")
+}
+
+tasks.withType<BootBuildImage> {
+    imageName.set("auto/aics_query")
+}
+
+tasks.withType<KotlinCompile> {
+    kotlinOptions {
+        freeCompilerArgs += "-Xjsr305=strict"
+        jvmTarget = "17"
+    }
+}
+
+tasks.withType<Test> {
+    useJUnitPlatform()
+}
--- a/buildDockerImage.ps1
+++ b/buildDockerImage.ps1
@ -0,0 +1,14 @@
+.\gradlew.bat bootJar
+
+java -Djarmode=layertools `
+-jar .\build\libs\AicsKnowledgeBase_query-1.0.0.jar `
+extract --destination build/extracted
+
+docker build -f .\Dockerfile -t auto/aics_query:latest .
+
+echo 'saving...'
+docker save -o ..\docker\aics_query.tar auto/aics_query:latest
+
+echo 'compressing...'
+Compress-Archive -Path ..\docker\aics_query.tar -DestinationPath ..\docker\aics_query.zip -Update
+pause
--- a/gradle/wrapper/gradle-wrapper.jar
+++ b/gradle/wrapper/gradle-wrapper.jar
--- a/gradle/wrapper/gradle-wrapper.properties
+++ b/gradle/wrapper/gradle-wrapper.properties
@ -0,0 +1,10 @@
+distributionBase=GRADLE_USER_HOME
+distributionPath=wrapper/dists
+distributionUrl=https\://services.gradle.org/distributions/gradle-8.1.1-bin.zip
+networkTimeout=10000
+zipStoreBase=GRADLE_USER_HOME
+zipStorePath=wrapper/dists
+systemProp.http.proxyHost=127.0.0.1
+systemProp.http.proxyPort=7890
+systemProp.https.proxyHost=127.0.0.1
+systemProp.https.proxyPort=7890
--- a/245
+++ b/245
@ -0,0 +1,245 @@
+#!/bin/sh
+
+#
+# Copyright © 2015-2021 the original 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 POSIX generated by Gradle.
+#
+#   Important for running:
+#
+#   (1) You need a POSIX-compliant shell to run this script. If your /bin/sh is
+#       noncompliant, but you have some other compliant shell such as ksh or
+#       bash, then to run this script, type that shell name before the whole
+#       command line, like:
+#
+#           ksh Gradle
+#
+#       Busybox and similar reduced shells will NOT work, because this script
+#       requires all of these POSIX shell features:
+#         * functions;
+#         * expansions «$var», «${var}», «${var:-default}», «${var+SET}»,
+#           «${var#prefix}», «${var%suffix}», and «$( cmd )»;
+#         * compound commands having a testable exit status, especially «case»;
+#         * various built-in commands including «command», «set», and «ulimit».
+#
+#   Important for patching:
+#
+#   (2) This script targets any POSIX shell, so it avoids extensions provided
+#       by Bash, Ksh, etc; in particular arrays are avoided.
+#
+#       The "traditional" practice of packing multiple parameters into a
+#       space-separated string is a well documented source of bugs and security
+#       problems, so this is (mostly) avoided, by progressively accumulating
+#       options in "$@", and eventually passing that to Java.
+#
+#       Where the inherited environment variables (DEFAULT_JVM_OPTS, JAVA_OPTS,
+#       and GRADLE_OPTS) rely on word-splitting, this is performed explicitly;
+#       see the in-line comments for details.
+#
+#       There are tweaks for specific operating systems such as AIX, CygWin,
+#       Darwin, MinGW, and NonStop.
+#
+#   (3) This script is generated from the Groovy template
+#       https://github.com/gradle/gradle/blob/HEAD/subprojects/plugins/src/main/resources/org/gradle/api/internal/plugins/unixStartScript.txt
+#       within the Gradle project.
+#
+#       You can find Gradle at https://github.com/gradle/gradle/.
+#
+##############################################################################
+
+# Attempt to set APP_HOME
+
+# Resolve links: $0 may be a link
+app_path=$0
+
+# Need this for daisy-chained symlinks.
+while
+    APP_HOME=${app_path%"${app_path##*/}"}  # leaves a trailing /; empty if no leading path
+    [ -h "$app_path" ]
+do
+    ls=$( ls -ld "$app_path" )
+    link=${ls#*' -> '}
+    case $link in             #(
+      /*)   app_path=$link ;; #(
+      *)    app_path=$APP_HOME$link ;;
+    esac
+done
+
+# This is normally unused
+# shellcheck disable=SC2034
+APP_BASE_NAME=${0##*/}
+APP_HOME=$( cd "${APP_HOME:-./}" && pwd -P ) || exit
+
+# Use the maximum available, or set MAX_FD != -1 to use that value.
+MAX_FD=maximum
+
+warn () {
+    echo "$*"
+} >&2
+
+die () {
+    echo
+    echo "$*"
+    echo
+    exit 1
+} >&2
+
+# 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  ;; #(
+  MSYS* | 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" && ! "$darwin" && ! "$nonstop" ; then
+    case $MAX_FD in #(
+      max*)
+        # In POSIX sh, ulimit -H is undefined. That's why the result is checked to see if it worked.
+        # shellcheck disable=SC3045
+        MAX_FD=$( ulimit -H -n ) ||
+            warn "Could not query maximum file descriptor limit"
+    esac
+    case $MAX_FD in  #(
+      '' | soft) :;; #(
+      *)
+        # In POSIX sh, ulimit -n is undefined. That's why the result is checked to see if it worked.
+        # shellcheck disable=SC3045
+        ulimit -n "$MAX_FD" ||
+            warn "Could not set maximum file descriptor limit to $MAX_FD"
+    esac
+fi
+
+# Collect all arguments for the java command, stacking in reverse order:
+#   * args from the command line
+#   * the main class name
+#   * -classpath
+#   * -D...appname settings
+#   * --module-path (only if needed)
+#   * DEFAULT_JVM_OPTS, JAVA_OPTS, and GRADLE_OPTS environment variables.
+
+# For Cygwin or MSYS, switch paths to Windows format before running java
+if "$cygwin" || "$msys" ; then
+    APP_HOME=$( cygpath --path --mixed "$APP_HOME" )
+    CLASSPATH=$( cygpath --path --mixed "$CLASSPATH" )
+
+    JAVACMD=$( cygpath --unix "$JAVACMD" )
+
+    # Now convert the arguments - kludge to limit ourselves to /bin/sh
+    for arg do
+        if
+            case $arg in                                #(
+              -*)   false ;;                            # don't mess with options #(
+              /?*)  t=${arg#/} t=/${t%%/*}              # looks like a POSIX filepath
+                    [ -e "$t" ] ;;                      #(
+              *)    false ;;
+            esac
+        then
+            arg=$( cygpath --path --ignore --mixed "$arg" )
+        fi
+        # Roll the args list around exactly as many times as the number of
+        # args, so each arg winds up back in the position where it started, but
+        # possibly modified.
+        #
+        # NB: a `for` loop captures its iteration list before it begins, so
+        # changing the positional parameters here affects neither the number of
+        # iterations, nor the values presented in `arg`.
+        shift                   # remove old arg
+        set -- "$@" "$arg"      # push replacement arg
+    done
+fi
+
+
+# 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"'
+
+# Collect all arguments for the java command;
+#   * $DEFAULT_JVM_OPTS, $JAVA_OPTS, and $GRADLE_OPTS can contain fragments of
+#     shell script including quotes and variable substitutions, so put them in
+#     double quotes to make sure that they get re-expanded; and
+#   * put everything else in single quotes, so that it's not re-expanded.
+
+set -- \
+        "-Dorg.gradle.appname=$APP_BASE_NAME" \
+        -classpath "$CLASSPATH" \
+        org.gradle.wrapper.GradleWrapperMain \
+        "$@"
+
+# Stop when "xargs" is not available.
+if ! command -v xargs >/dev/null 2>&1
+then
+    die "xargs is not available"
+fi
+
+# Use "xargs" to parse quoted args.
+#
+# With -n1 it outputs one arg per line, with the quotes and backslashes removed.
+#
+# In Bash we could simply go:
+#
+#   readarray ARGS < <( xargs -n1 <<<"$var" ) &&
+#   set -- "${ARGS[@]}" "$@"
+#
+# but POSIX shell has neither arrays nor command substitution, so instead we
+# post-process each arg (as a line of input to sed) to backslash-escape any
+# character that might be a shell metacharacter, then use eval to reverse
+# that process (while maintaining the separation between arguments), and wrap
+# the whole thing up as a single "set" statement.
+#
+# This will of course break if any of these variables contains a newline or
+# an unmatched quote.
+#
+
+eval "set -- $(
+        printf '%s\n' "$DEFAULT_JVM_OPTS $JAVA_OPTS $GRADLE_OPTS" |
+        xargs -n1 |
+        sed ' s~[^-[:alnum:]+,./:=@_]~\\&~g; ' |
+        tr '\n' ' '
+    )" '"$@"'
+
+exec "$JAVACMD" "$@"
--- a/gradlew.bat
+++ b/gradlew.bat
@ -0,0 +1,92 @@
+@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=.
+@rem This is normally unused
+set APP_BASE_NAME=%~n0
+set APP_HOME=%DIRNAME%
+
+@rem Resolve any "." and ".." in APP_HOME to make it shorter.
+for %%i in ("%APP_HOME%") do set APP_HOME=%%~fi
+
+@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% equ 0 goto execute
+
+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 execute
+
+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
+
+: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 %*
+
+:end
+@rem End local scope for the variables with windows NT shell
+if %ERRORLEVEL% equ 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!
+set EXIT_CODE=%ERRORLEVEL%
+if %EXIT_CODE% equ 0 set EXIT_CODE=1
+if not ""=="%GRADLE_EXIT_CONSOLE%" exit %EXIT_CODE%
+exit /b %EXIT_CODE%
+
+:mainEnd
+if "%OS%"=="Windows_NT" endlocal
+
+:omega
--- a/settings.gradle.kts
+++ b/settings.gradle.kts
@ -0,0 +1 @@
+rootProject.name = "AicsKnowledgeBase_query"
--- a/src/main/kotlin/cn/edu/hfut/auto/knowledge/AicsKnowledgeBaseQueryApplication.kt
+++ b/src/main/kotlin/cn/edu/hfut/auto/knowledge/AicsKnowledgeBaseQueryApplication.kt
@ -0,0 +1,11 @@
+package cn.edu.hfut.auto.knowledge
+
+import org.springframework.boot.autoconfigure.SpringBootApplication
+import org.springframework.boot.runApplication
+
+@SpringBootApplication
+class AicsKnowledgeBaseQueryApplication
+
+fun main(args: Array<String>) {
+    runApplication<AicsKnowledgeBaseQueryApplication>(*args)
+}
--- a/src/main/kotlin/cn/edu/hfut/auto/knowledge/config/SerializeConfig.kt
+++ b/src/main/kotlin/cn/edu/hfut/auto/knowledge/config/SerializeConfig.kt
@ -0,0 +1,34 @@
+package cn.edu.hfut.auto.knowledge.config
+
+import com.fasterxml.jackson.core.JsonGenerator
+import com.fasterxml.jackson.databind.JsonSerializer
+import com.fasterxml.jackson.databind.Module
+import com.fasterxml.jackson.databind.SerializerProvider
+import com.fasterxml.jackson.databind.module.SimpleModule
+import org.springframework.context.annotation.Bean
+import org.springframework.context.annotation.Configuration
+import org.springframework.data.elasticsearch.core.SearchHits
+
+@Configuration
+class SerializeConfig {
+    @Bean
+    fun elasticModule(): Module {
+        return SimpleModule().apply {
+            addSerializer(SearchHits::class.java, SearchHitsJsonSerializer())
+        }
+    }
+
+    class SearchHitsJsonSerializer : JsonSerializer<SearchHits<*>>() {
+        override fun serialize(value: SearchHits<*>?, gen: JsonGenerator, serializers: SerializerProvider) {
+            if (value == null) {
+                gen.writeNull()
+            } else {
+                gen.writeStartArray()
+                value.forEach {
+                    gen.writeObject(it)
+                }
+                gen.writeEndArray()
+            }
+        }
+    }
+}
--- a/src/main/kotlin/cn/edu/hfut/auto/knowledge/controller/SearchController.kt
+++ b/src/main/kotlin/cn/edu/hfut/auto/knowledge/controller/SearchController.kt
@ -0,0 +1,45 @@
+package cn.edu.hfut.auto.knowledge.controller
+
+import cn.edu.hfut.auto.knowledge.entity.KnowledgeFile
+import cn.edu.hfut.auto.knowledge.entity.Note
+import cn.edu.hfut.auto.knowledge.entity.vo.NoteUploadVO
+import cn.edu.hfut.auto.knowledge.service.KnowledgeFileService
+import cn.edu.hfut.auto.knowledge.service.NoteService
+import org.springframework.data.elasticsearch.core.SearchHits
+import org.springframework.web.bind.annotation.*
+import java.util.*
+
+@RestController
+@RequestMapping("/search")
+class SearchController(
+    private val knowledgeFileService: KnowledgeFileService,
+    private val noteService: NoteService
+) {
+    @GetMapping("/knowledge")
+    suspend fun searchInKnowledge(
+        text: String,
+        applyToBriefAndContent: Boolean,
+        @RequestParam tags: List<Long>
+    ): SearchHits<KnowledgeFile> {
+        return knowledgeFileService.search(text, applyToBriefAndContent, tags)
+    }
+
+    @GetMapping("/note")
+    suspend fun searchInNotes(
+        text: String,
+        applyToBriefAndContent: Boolean,
+        @RequestParam tags: List<Long>
+    ): SearchHits<Note> {
+        return noteService.search(text, applyToBriefAndContent, tags)
+    }
+
+    @GetMapping("/note/{noteId}/content")
+    suspend fun getNoteContent(@PathVariable noteId: UUID): String {
+        return noteService.findNoteByUUID(noteId)?.content ?: "null"
+    }
+
+    @PutMapping("/note/{noteId}")
+    suspend fun putNote(@PathVariable noteId: UUID, @RequestBody vo: NoteUploadVO) {
+        noteService.saveNote(noteId, vo.title, vo.content, vo.tags)
+    }
+}
--- a/src/main/kotlin/cn/edu/hfut/auto/knowledge/entity/KnowledgeFile.kt
+++ b/src/main/kotlin/cn/edu/hfut/auto/knowledge/entity/KnowledgeFile.kt
@ -0,0 +1,22 @@
+package cn.edu.hfut.auto.knowledge.entity
+
+import org.springframework.data.annotation.Id
+import org.springframework.data.elasticsearch.annotations.Document
+import org.springframework.data.elasticsearch.annotations.Field
+import org.springframework.data.elasticsearch.annotations.FieldType
+import java.util.*
+
+@Document(indexName = "knowledge_file")
+data class KnowledgeFile(
+    @Id
+    @Field(type = FieldType.Keyword)
+    val uuid: UUID,
+    @Field(type = FieldType.Text)
+    val name: String,
+    @Field(type = FieldType.Text)
+    val brief: String,
+    @Field(type = FieldType.Text)
+    val content: String?,
+    @Field(type = FieldType.Long)
+    val tags: List<Long>
+)
--- a/src/main/kotlin/cn/edu/hfut/auto/knowledge/entity/Note.kt
+++ b/src/main/kotlin/cn/edu/hfut/auto/knowledge/entity/Note.kt
@ -0,0 +1,20 @@
+package cn.edu.hfut.auto.knowledge.entity
+
+import org.springframework.data.annotation.Id
+import org.springframework.data.elasticsearch.annotations.Document
+import org.springframework.data.elasticsearch.annotations.Field
+import org.springframework.data.elasticsearch.annotations.FieldType
+import java.util.UUID
+
+@Document(indexName = "note")
+data class Note(
+    @Id
+    @Field(type = FieldType.Keyword)
+    val uuid: UUID,
+    @Field(type = FieldType.Text)
+    val title: String,
+    @Field(type = FieldType.Text)
+    val content: String,
+    @Field(type = FieldType.Long)
+    val tags: List<Long>
+)
--- a/src/main/kotlin/cn/edu/hfut/auto/knowledge/entity/vo/NoteVO.kt
+++ b/src/main/kotlin/cn/edu/hfut/auto/knowledge/entity/vo/NoteVO.kt
@ -0,0 +1,7 @@
+package cn.edu.hfut.auto.knowledge.entity.vo
+
+data class NoteUploadVO(
+    val title: String,
+    val content: String,
+    val tags: List<Long>
+)
--- a/src/main/kotlin/cn/edu/hfut/auto/knowledge/service/KnowledgeFileService.kt
+++ b/src/main/kotlin/cn/edu/hfut/auto/knowledge/service/KnowledgeFileService.kt
@ -0,0 +1,64 @@
+package cn.edu.hfut.auto.knowledge.service
+
+import cn.edu.hfut.auto.knowledge.entity.KnowledgeFile
+import cn.edu.hfut.auto.knowledge.util.*
+import org.springframework.data.elasticsearch.core.ElasticsearchOperations
+import org.springframework.data.elasticsearch.core.SearchHits
+import org.springframework.data.elasticsearch.core.query.FetchSourceFilterBuilder
+import org.springframework.data.elasticsearch.core.query.highlight.HighlightField
+import org.springframework.stereotype.Service
+
+@Service
+class KnowledgeFileService(private val operations: ElasticsearchOperations) {
+    companion object {
+        private val targetClass = KnowledgeFile::class.java
+    }
+
+    fun search(text: String, applyToBriefAndContent: Boolean, tags: List<Long>): SearchHits<KnowledgeFile> {
+        val query = nativeQuery {
+            query {
+                boolQuery {
+                    queryMust {
+                        if (applyToBriefAndContent) {
+                            combinedFieldsQuery {
+                                fields("name", "brief", "content").query(text)
+                            }
+                        } else {
+                            matchQuery {
+                                field("name").query(text)
+                            }
+                        }
+                    }
+                    tags.takeIf { it.isNotEmpty() }?.let { tagList ->
+                        queryFilter {
+                            tagList.forEach {
+                                termQuery {
+                                    field("tags").value(it)
+                                }
+                            }
+                        }
+                    }
+                }
+
+                withSourceFilter(FetchSourceFilterBuilder().withExcludes("content").build())
+            }
+
+            highlight<KnowledgeFile>(
+                if (applyToBriefAndContent) {
+                    listOf(
+                        HighlightField("name"), HighlightField("brief"), HighlightField("content")
+                    )
+                } else {
+                    listOf(HighlightField("name"))
+                }
+            ) {
+                withPreTags("<span style=\"background:yellow\">")
+                withPostTags("</span>")
+                withBoundaryMaxScan(50)
+                withBoundaryScanner("sentence")
+                withBoundaryScannerLocale("zh-CN")
+            }
+        }
+        return operations.search(query, targetClass)
+    }
+}
--- a/src/main/kotlin/cn/edu/hfut/auto/knowledge/service/NoteService.kt
+++ b/src/main/kotlin/cn/edu/hfut/auto/knowledge/service/NoteService.kt
@ -0,0 +1,81 @@
+package cn.edu.hfut.auto.knowledge.service
+
+import cn.edu.hfut.auto.knowledge.entity.KnowledgeFile
+import cn.edu.hfut.auto.knowledge.entity.Note
+import cn.edu.hfut.auto.knowledge.util.*
+import org.springframework.data.elasticsearch.core.ElasticsearchOperations
+import org.springframework.data.elasticsearch.core.SearchHits
+import org.springframework.data.elasticsearch.core.query.FetchSourceFilterBuilder
+import org.springframework.data.elasticsearch.core.query.highlight.HighlightField
+import org.springframework.stereotype.Service
+import java.util.UUID
+
+@Service
+class NoteService(private val operations: ElasticsearchOperations) {
+    companion object {
+        private val targetClass = Note::class.java
+    }
+
+    fun search(text: String, applyToBriefAndContent: Boolean, tags: List<Long>): SearchHits<Note> {
+        val query = nativeQuery {
+            query {
+                boolQuery {
+                    queryMust {
+                        if (applyToBriefAndContent) {
+                            combinedFieldsQuery {
+                                fields("title", "brief", "content").query(text)
+                            }
+                        } else {
+                            matchQuery {
+                                field("title").query(text)
+                            }
+                        }
+                    }
+                    tags.takeIf { it.isNotEmpty() }?.let { tagList ->
+                        queryFilter {
+                            tagList.forEach {
+                                termQuery {
+                                    field("tags").value(it)
+                                }
+                            }
+                        }
+                    }
+                }
+
+                withSourceFilter(FetchSourceFilterBuilder().withExcludes("content").build())
+            }
+
+            highlight<KnowledgeFile>(
+                if (applyToBriefAndContent) {
+                    listOf(
+                        HighlightField("title"), HighlightField("brief"), HighlightField("content")
+                    )
+                } else {
+                    listOf(HighlightField("title"))
+                }
+            ) {
+                withPreTags("<span style=\"background:yellow\">")
+                withPostTags("</span>")
+                withBoundaryMaxScan(50)
+                withBoundaryScanner("sentence")
+                withBoundaryScannerLocale("zh-CN")
+            }
+        }
+        return operations.search(query, targetClass)
+    }
+
+    fun findNoteByUUID(uuid: UUID): Note? {
+        val query = nativeQuery {
+            query {
+                matchQuery {
+                    field("uuid").query(uuid.toString())
+                }
+            }
+        }
+        return operations.search(query, targetClass).searchHits.firstOrNull()?.content
+    }
+
+    fun saveNote(id: UUID, title: String, content: String, tags: List<Long>) {
+        operations.save(Note(id, title, content, tags))
+    }
+}
--- a/src/main/kotlin/cn/edu/hfut/auto/knowledge/util/ElasticsearchKotlinExtensions.kt
+++ b/src/main/kotlin/cn/edu/hfut/auto/knowledge/util/ElasticsearchKotlinExtensions.kt
@ -0,0 +1,74 @@
+package cn.edu.hfut.auto.knowledge.util
+
+import co.elastic.clients.elasticsearch._types.query_dsl.*
+import org.springframework.data.elasticsearch.client.elc.NativeQuery
+import org.springframework.data.elasticsearch.client.elc.NativeQueryBuilder
+import org.springframework.data.elasticsearch.core.query.HighlightQuery
+import org.springframework.data.elasticsearch.core.query.highlight.Highlight
+import org.springframework.data.elasticsearch.core.query.highlight.HighlightField
+import org.springframework.data.elasticsearch.core.query.highlight.HighlightParameters.HighlightParametersBuilder
+
+@DslMarker
+annotation class ElasticNativeQueryDsl
+
+@DslMarker
+annotation class ElasticNativeTopQueryDsl
+
+@DslMarker
+annotation class ElasticNativeQueryLeafDsl
+
+@DslMarker
+annotation class ElasticNativeMiddleQueryDsl
+
+@ElasticNativeQueryDsl
+inline fun nativeQuery(builder: NativeQueryBuilder.() -> Unit): NativeQuery {
+    return NativeQuery(NativeQueryBuilder().apply(builder))
+}
+
+@ElasticNativeTopQueryDsl
+inline fun @receiver:ElasticNativeTopQueryDsl NativeQueryBuilder.query(builder: Query.Builder.() -> Unit) {
+    withQuery(Query.Builder().apply(builder).build())
+}
+
+@ElasticNativeQueryLeafDsl
+inline fun @receiver:ElasticNativeQueryLeafDsl Query.Builder.matchQuery(builder: MatchQuery.Builder.() -> Unit) {
+    match(MatchQuery.Builder().apply(builder).build())
+}
+
+@ElasticNativeQueryLeafDsl
+inline fun @receiver:ElasticNativeQueryLeafDsl Query.Builder.termQuery(builder: TermQuery.Builder.() -> Unit) {
+    term(TermQuery.Builder().apply(builder).build())
+}
+
+@ElasticNativeQueryLeafDsl
+inline fun @receiver:ElasticNativeQueryLeafDsl Query.Builder.combinedFieldsQuery(builder: CombinedFieldsQuery.Builder.() -> Unit) {
+    combinedFields(CombinedFieldsQuery.Builder().apply(builder).build())
+}
+
+@ElasticNativeTopQueryDsl
+inline fun <reified T> @receiver:ElasticNativeTopQueryDsl NativeQueryBuilder.highlight(
+    fields: List<HighlightField>,
+    builder: HighlightParametersBuilder.() -> Unit
+) {
+    withHighlightQuery(HighlightQuery(Highlight(HighlightParametersBuilder().apply(builder).build(), fields), T::class.java))
+}
+
+@ElasticNativeQueryLeafDsl
+inline fun @receiver:ElasticNativeQueryLeafDsl Query.Builder.boolQuery(builder: BoolQuery.Builder.() -> Unit) {
+    bool(BoolQuery.Builder().apply(builder).build())
+}
+
+@ElasticNativeMiddleQueryDsl
+inline fun BoolQuery.Builder.queryShould(builder: Query.Builder.() -> Unit) {
+    should(Query.Builder().apply(builder).build())
+}
+
+@ElasticNativeMiddleQueryDsl
+inline fun BoolQuery.Builder.queryMust(builder: Query.Builder.() -> Unit) {
+    must(Query.Builder().apply(builder).build())
+}
+
+@ElasticNativeMiddleQueryDsl
+inline fun BoolQuery.Builder.queryFilter(builder: Query.Builder.() -> Unit) {
+    filter(Query.Builder().apply(builder).build())
+}
--- a/src/main/resources/application-dev.yml
+++ b/src/main/resources/application-dev.yml
@ -0,0 +1,11 @@
+server:
+  http2:
+    enabled: true
+  port: 8082
+spring:
+  elasticsearch:
+    uris:
+      - localhost:9200
+
+logging:
+  config: classpath:log4j2-dev.xml
--- a/src/main/resources/application-prod.yml
+++ b/src/main/resources/application-prod.yml
@ -0,0 +1,11 @@
+server:
+  http2:
+    enabled: true
+  port: 8082
+spring:
+  elasticsearch:
+    uris:
+      - elasticsearch:9200
+
+logging:
+  config: classpath:log4j2-prod.xml
--- a/src/main/resources/application.yml
+++ b/src/main/resources/application.yml
@ -0,0 +1,3 @@
+spring:
+  profiles:
+    active: ${ENV:dev}
--- a/src/main/resources/log4j2-dev.xml
+++ b/src/main/resources/log4j2-dev.xml
@ -0,0 +1,26 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<Configuration status="WARN">
+
+    <!-- 变量配置，定义变量，方便在下面使用,主要配置一些参数 -->
+    <Properties>
+        <!-- 格式化输出：%date表示日期，%thread表示线程名，%-5level：级别从左显示5个字符宽度 %msg：日志消息，%n是换行符 -->
+        <!-- %logger{36} 表示 Logger 名字最长36个字符 -->
+        <property name="LOG_PATTERN"
+                  value="%date{MM-dd HH:mm:ss.SSS} %highlight{%-5level} [%thread]  %highlight{%logger{36}} : %msg%n" />
+        <!-- 定义日志存储的路径 -->
+        <property name="FILE_PATH" value="logs" />
+        <property name="FILE_NAME" value="maij" />
+    </Properties>
+
+    <Appenders>
+        <Console name="Console" target="SYSTEM_OUT">
+            <PatternLayout pattern="${LOG_PATTERN}"/>
+        </Console>
+    </Appenders>
+
+    <Loggers>
+        <Root level="info" includeLocation="false">
+            <AppenderRef ref="Console"/>
+        </Root>
+    </Loggers>
+</Configuration>
--- a/src/main/resources/log4j2-prod.xml
+++ b/src/main/resources/log4j2-prod.xml
@ -0,0 +1,46 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<Configuration status="WARN">
+
+    <Properties>
+        <property name="LOG_PATTERN"
+                  value="%date{MM-dd HH:mm:ss.SSS} %-5level [%thread]  %logger{36} : %msg%n"/>
+        <!-- 定义日志存储的路径 -->
+        <property name="FILE_PATH" value="./logs"/>
+        <property name="FILE_NAME" value="aics"/>
+    </Properties>
+
+    <Appenders>
+        <!-- warn -->
+        <RollingRandomAccessFile name="RollingFileWarn"
+                                 fileName="${FILE_PATH}/warn.log"
+                                 filePattern="${FILE_PATH}/${FILE_NAME}-WARN-%d{yyyy-MM-dd}_%i.log.gz">
+            <ThresholdFilter level="warn" onMatch="ACCEPT"
+                             onMismatch="DENY"/>
+            <PatternLayout pattern="${LOG_PATTERN}"/>
+            <Policies>
+                <TimeBasedTriggeringPolicy interval="1"/>
+                <SizeBasedTriggeringPolicy size="10MB"/>
+            </Policies>
+        </RollingRandomAccessFile>
+
+        <!-- error -->
+        <RollingRandomAccessFile name="RollingFileError"
+                                 fileName="${FILE_PATH}/error.log"
+                                 filePattern="${FILE_PATH}/${FILE_NAME}-ERROR-%d{yyyy-MM-dd}_%i.log.gz">
+            <ThresholdFilter level="error" onMatch="ACCEPT"
+                             onMismatch="DENY"/>
+            <PatternLayout pattern="${LOG_PATTERN}"/>
+            <Policies>
+                <TimeBasedTriggeringPolicy interval="1"/>
+                <SizeBasedTriggeringPolicy size="10MB"/>
+            </Policies>
+        </RollingRandomAccessFile>
+    </Appenders>
+
+    <Loggers>
+        <AsyncRoot level="info" includeLocation="false">
+            <AppenderRef ref="RollingFileWarn"/>
+            <AppenderRef ref="RollingFileError"/>
+        </AsyncRoot>
+    </Loggers>
+</Configuration>
--- a/src/test/kotlin/cn/edu/hfut/auto/knowledge/AicsKnowledgeBaseQueryApplicationTests.kt
+++ b/src/test/kotlin/cn/edu/hfut/auto/knowledge/AicsKnowledgeBaseQueryApplicationTests.kt
@ -0,0 +1,13 @@
+package cn.edu.hfut.auto.knowledge
+
+import org.junit.jupiter.api.Test
+import org.springframework.boot.test.context.SpringBootTest
+
+@SpringBootTest
+class AicsKnowledgeBaseQueryApplicationTests {
+
+    @Test
+    fun contextLoads() {
+    }
+
+}
--- a/src/test/kotlin/cn/edu/hfut/auto/knowledge/DataTests.kt
+++ b/src/test/kotlin/cn/edu/hfut/auto/knowledge/DataTests.kt
@ -0,0 +1,19 @@
+package cn.edu.hfut.auto.knowledge
+
+import cn.edu.hfut.auto.knowledge.service.KnowledgeFileService
+import org.junit.jupiter.api.Test
+import org.springframework.beans.factory.annotation.Autowired
+import org.springframework.boot.test.context.SpringBootTest
+
+@SpringBootTest
+class DataTests {
+
+    @Autowired
+    lateinit var knowledgeFileService: KnowledgeFileService
+    @Test
+    fun test() {
+        knowledgeFileService.search("java", true,).forEach {
+            println(it)
+        }
+    }
+}
--- a/src/test/kotlin/cn/edu/hfut/auto/knowledge/UUIDGenerator.kt
+++ b/src/test/kotlin/cn/edu/hfut/auto/knowledge/UUIDGenerator.kt
@ -0,0 +1,17 @@
+package cn.edu.hfut.auto.knowledge
+
+import org.junit.jupiter.api.Test
+import java.nio.ByteBuffer
+import java.util.*
+
+class UUIDGenerator {
+    @Test
+    fun test() {
+        val uuid = UUID.randomUUID()
+        val buffer = ByteBuffer.allocate(16)
+        buffer.putLong(uuid.mostSignificantBits)
+        buffer.putLong(uuid.leastSignificantBits)
+
+        println(Base64.getUrlEncoder().encodeToString(buffer.array()))
+    }
+}
				`@ -0,0 +1 @@`
				`rootProject.name = "AicsKnowledgeBase_query"`