From 6f9cb7740238fd8c34e0f13d035a0d6537ae3db6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=AD=99=E5=B0=8F=E4=BA=91?= Date: Wed, 18 Dec 2024 17:09:09 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E8=AE=B0=E5=BD=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .gitattributes | 2 + .gitignore | 33 +++ .mvn/wrapper/maven-wrapper.properties | 19 ++ mvnw | 259 ++++++++++++++++++ mvnw.cmd | 149 ++++++++++ pid.txt | 0 pom.xml | 61 +++++ .../com/tuoheng/steam/StreamApplication.java | 14 + .../controller/StreamRecordController.java | 53 ++++ .../steam/controller/dto/Response.java | 23 ++ .../com/tuoheng/steam/dos/ProcessType.java | 5 + .../com/tuoheng/steam/dos/StreamProcess.java | 78 ++++++ .../com/tuoheng/steam/dos/StreamTask.java | 126 +++++++++ .../tuoheng/steam/service/ITaskService.java | 14 + .../tuoheng/steam/service/TaskService.java | 238 ++++++++++++++++ .../service/innerService/ProcessService.java | 119 ++++++++ .../tuoheng/steam/util/ProcessManager.java | 105 +++++++ .../com/tuoheng/steam/util/TimeUtils.java | 15 + src/main/resources/application.properties | 4 + .../tuoheng/steam/StreamApplicationTests.java | 34 +++ 20 files changed, 1351 insertions(+) create mode 100644 .gitattributes create mode 100644 .gitignore create mode 100644 .mvn/wrapper/maven-wrapper.properties create mode 100755 mvnw create mode 100644 mvnw.cmd create mode 100644 pid.txt create mode 100644 pom.xml create mode 100644 src/main/java/com/tuoheng/steam/StreamApplication.java create mode 100644 src/main/java/com/tuoheng/steam/controller/StreamRecordController.java create mode 100644 src/main/java/com/tuoheng/steam/controller/dto/Response.java create mode 100644 src/main/java/com/tuoheng/steam/dos/ProcessType.java create mode 100644 src/main/java/com/tuoheng/steam/dos/StreamProcess.java create mode 100644 src/main/java/com/tuoheng/steam/dos/StreamTask.java create mode 100644 src/main/java/com/tuoheng/steam/service/ITaskService.java create mode 100644 src/main/java/com/tuoheng/steam/service/TaskService.java create mode 100644 src/main/java/com/tuoheng/steam/service/innerService/ProcessService.java create mode 100644 src/main/java/com/tuoheng/steam/util/ProcessManager.java create mode 100644 src/main/java/com/tuoheng/steam/util/TimeUtils.java create mode 100644 src/main/resources/application.properties create mode 100644 src/test/java/com/tuoheng/steam/StreamApplicationTests.java diff --git a/.gitattributes b/.gitattributes new file mode 100644 index 0000000..3b41682 --- /dev/null +++ b/.gitattributes @@ -0,0 +1,2 @@ +/mvnw text eol=lf +*.cmd text eol=crlf diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..549e00a --- /dev/null +++ b/.gitignore @@ -0,0 +1,33 @@ +HELP.md +target/ +!.mvn/wrapper/maven-wrapper.jar +!**/src/main/**/target/ +!**/src/test/**/target/ + +### STS ### +.apt_generated +.classpath +.factorypath +.project +.settings +.springBeans +.sts4-cache + +### IntelliJ IDEA ### +.idea +*.iws +*.iml +*.ipr + +### NetBeans ### +/nbproject/private/ +/nbbuild/ +/dist/ +/nbdist/ +/.nb-gradle/ +build/ +!**/src/main/**/build/ +!**/src/test/**/build/ + +### VS Code ### +.vscode/ diff --git a/.mvn/wrapper/maven-wrapper.properties b/.mvn/wrapper/maven-wrapper.properties new file mode 100644 index 0000000..d58dfb7 --- /dev/null +++ b/.mvn/wrapper/maven-wrapper.properties @@ -0,0 +1,19 @@ +# Licensed to the Apache Software Foundation (ASF) under one +# or more contributor license agreements. See the NOTICE file +# distributed with this work for additional information +# regarding copyright ownership. The ASF licenses this file +# to you 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 +# +# http://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. +wrapperVersion=3.3.2 +distributionType=only-script +distributionUrl=https://repo.maven.apache.org/maven2/org/apache/maven/apache-maven/3.9.9/apache-maven-3.9.9-bin.zip diff --git a/mvnw b/mvnw new file mode 100755 index 0000000..19529dd --- /dev/null +++ b/mvnw @@ -0,0 +1,259 @@ +#!/bin/sh +# ---------------------------------------------------------------------------- +# Licensed to the Apache Software Foundation (ASF) under one +# or more contributor license agreements. See the NOTICE file +# distributed with this work for additional information +# regarding copyright ownership. The ASF licenses this file +# to you 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 +# +# http://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. +# ---------------------------------------------------------------------------- + +# ---------------------------------------------------------------------------- +# Apache Maven Wrapper startup batch script, version 3.3.2 +# +# Optional ENV vars +# ----------------- +# JAVA_HOME - location of a JDK home dir, required when download maven via java source +# MVNW_REPOURL - repo url base for downloading maven distribution +# MVNW_USERNAME/MVNW_PASSWORD - user and password for downloading maven +# MVNW_VERBOSE - true: enable verbose log; debug: trace the mvnw script; others: silence the output +# ---------------------------------------------------------------------------- + +set -euf +[ "${MVNW_VERBOSE-}" != debug ] || set -x + +# OS specific support. +native_path() { printf %s\\n "$1"; } +case "$(uname)" in +CYGWIN* | MINGW*) + [ -z "${JAVA_HOME-}" ] || JAVA_HOME="$(cygpath --unix "$JAVA_HOME")" + native_path() { cygpath --path --windows "$1"; } + ;; +esac + +# set JAVACMD and JAVACCMD +set_java_home() { + # For Cygwin and MinGW, ensure paths are in Unix format before anything is touched + 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" + JAVACCMD="$JAVA_HOME/jre/sh/javac" + else + JAVACMD="$JAVA_HOME/bin/java" + JAVACCMD="$JAVA_HOME/bin/javac" + + if [ ! -x "$JAVACMD" ] || [ ! -x "$JAVACCMD" ]; then + echo "The JAVA_HOME environment variable is not defined correctly, so mvnw cannot run." >&2 + echo "JAVA_HOME is set to \"$JAVA_HOME\", but \"\$JAVA_HOME/bin/java\" or \"\$JAVA_HOME/bin/javac\" does not exist." >&2 + return 1 + fi + fi + else + JAVACMD="$( + 'set' +e + 'unset' -f command 2>/dev/null + 'command' -v java + )" || : + JAVACCMD="$( + 'set' +e + 'unset' -f command 2>/dev/null + 'command' -v javac + )" || : + + if [ ! -x "${JAVACMD-}" ] || [ ! -x "${JAVACCMD-}" ]; then + echo "The java/javac command does not exist in PATH nor is JAVA_HOME set, so mvnw cannot run." >&2 + return 1 + fi + fi +} + +# hash string like Java String::hashCode +hash_string() { + str="${1:-}" h=0 + while [ -n "$str" ]; do + char="${str%"${str#?}"}" + h=$(((h * 31 + $(LC_CTYPE=C printf %d "'$char")) % 4294967296)) + str="${str#?}" + done + printf %x\\n $h +} + +verbose() { :; } +[ "${MVNW_VERBOSE-}" != true ] || verbose() { printf %s\\n "${1-}"; } + +die() { + printf %s\\n "$1" >&2 + exit 1 +} + +trim() { + # MWRAPPER-139: + # Trims trailing and leading whitespace, carriage returns, tabs, and linefeeds. + # Needed for removing poorly interpreted newline sequences when running in more + # exotic environments such as mingw bash on Windows. + printf "%s" "${1}" | tr -d '[:space:]' +} + +# parse distributionUrl and optional distributionSha256Sum, requires .mvn/wrapper/maven-wrapper.properties +while IFS="=" read -r key value; do + case "${key-}" in + distributionUrl) distributionUrl=$(trim "${value-}") ;; + distributionSha256Sum) distributionSha256Sum=$(trim "${value-}") ;; + esac +done <"${0%/*}/.mvn/wrapper/maven-wrapper.properties" +[ -n "${distributionUrl-}" ] || die "cannot read distributionUrl property in ${0%/*}/.mvn/wrapper/maven-wrapper.properties" + +case "${distributionUrl##*/}" in +maven-mvnd-*bin.*) + MVN_CMD=mvnd.sh _MVNW_REPO_PATTERN=/maven/mvnd/ + case "${PROCESSOR_ARCHITECTURE-}${PROCESSOR_ARCHITEW6432-}:$(uname -a)" in + *AMD64:CYGWIN* | *AMD64:MINGW*) distributionPlatform=windows-amd64 ;; + :Darwin*x86_64) distributionPlatform=darwin-amd64 ;; + :Darwin*arm64) distributionPlatform=darwin-aarch64 ;; + :Linux*x86_64*) distributionPlatform=linux-amd64 ;; + *) + echo "Cannot detect native platform for mvnd on $(uname)-$(uname -m), use pure java version" >&2 + distributionPlatform=linux-amd64 + ;; + esac + distributionUrl="${distributionUrl%-bin.*}-$distributionPlatform.zip" + ;; +maven-mvnd-*) MVN_CMD=mvnd.sh _MVNW_REPO_PATTERN=/maven/mvnd/ ;; +*) MVN_CMD="mvn${0##*/mvnw}" _MVNW_REPO_PATTERN=/org/apache/maven/ ;; +esac + +# apply MVNW_REPOURL and calculate MAVEN_HOME +# maven home pattern: ~/.m2/wrapper/dists/{apache-maven-,maven-mvnd--}/ +[ -z "${MVNW_REPOURL-}" ] || distributionUrl="$MVNW_REPOURL$_MVNW_REPO_PATTERN${distributionUrl#*"$_MVNW_REPO_PATTERN"}" +distributionUrlName="${distributionUrl##*/}" +distributionUrlNameMain="${distributionUrlName%.*}" +distributionUrlNameMain="${distributionUrlNameMain%-bin}" +MAVEN_USER_HOME="${MAVEN_USER_HOME:-${HOME}/.m2}" +MAVEN_HOME="${MAVEN_USER_HOME}/wrapper/dists/${distributionUrlNameMain-}/$(hash_string "$distributionUrl")" + +exec_maven() { + unset MVNW_VERBOSE MVNW_USERNAME MVNW_PASSWORD MVNW_REPOURL || : + exec "$MAVEN_HOME/bin/$MVN_CMD" "$@" || die "cannot exec $MAVEN_HOME/bin/$MVN_CMD" +} + +if [ -d "$MAVEN_HOME" ]; then + verbose "found existing MAVEN_HOME at $MAVEN_HOME" + exec_maven "$@" +fi + +case "${distributionUrl-}" in +*?-bin.zip | *?maven-mvnd-?*-?*.zip) ;; +*) die "distributionUrl is not valid, must match *-bin.zip or maven-mvnd-*.zip, but found '${distributionUrl-}'" ;; +esac + +# prepare tmp dir +if TMP_DOWNLOAD_DIR="$(mktemp -d)" && [ -d "$TMP_DOWNLOAD_DIR" ]; then + clean() { rm -rf -- "$TMP_DOWNLOAD_DIR"; } + trap clean HUP INT TERM EXIT +else + die "cannot create temp dir" +fi + +mkdir -p -- "${MAVEN_HOME%/*}" + +# Download and Install Apache Maven +verbose "Couldn't find MAVEN_HOME, downloading and installing it ..." +verbose "Downloading from: $distributionUrl" +verbose "Downloading to: $TMP_DOWNLOAD_DIR/$distributionUrlName" + +# select .zip or .tar.gz +if ! command -v unzip >/dev/null; then + distributionUrl="${distributionUrl%.zip}.tar.gz" + distributionUrlName="${distributionUrl##*/}" +fi + +# verbose opt +__MVNW_QUIET_WGET=--quiet __MVNW_QUIET_CURL=--silent __MVNW_QUIET_UNZIP=-q __MVNW_QUIET_TAR='' +[ "${MVNW_VERBOSE-}" != true ] || __MVNW_QUIET_WGET='' __MVNW_QUIET_CURL='' __MVNW_QUIET_UNZIP='' __MVNW_QUIET_TAR=v + +# normalize http auth +case "${MVNW_PASSWORD:+has-password}" in +'') MVNW_USERNAME='' MVNW_PASSWORD='' ;; +has-password) [ -n "${MVNW_USERNAME-}" ] || MVNW_USERNAME='' MVNW_PASSWORD='' ;; +esac + +if [ -z "${MVNW_USERNAME-}" ] && command -v wget >/dev/null; then + verbose "Found wget ... using wget" + wget ${__MVNW_QUIET_WGET:+"$__MVNW_QUIET_WGET"} "$distributionUrl" -O "$TMP_DOWNLOAD_DIR/$distributionUrlName" || die "wget: Failed to fetch $distributionUrl" +elif [ -z "${MVNW_USERNAME-}" ] && command -v curl >/dev/null; then + verbose "Found curl ... using curl" + curl ${__MVNW_QUIET_CURL:+"$__MVNW_QUIET_CURL"} -f -L -o "$TMP_DOWNLOAD_DIR/$distributionUrlName" "$distributionUrl" || die "curl: Failed to fetch $distributionUrl" +elif set_java_home; then + verbose "Falling back to use Java to download" + javaSource="$TMP_DOWNLOAD_DIR/Downloader.java" + targetZip="$TMP_DOWNLOAD_DIR/$distributionUrlName" + cat >"$javaSource" <<-END + public class Downloader extends java.net.Authenticator + { + protected java.net.PasswordAuthentication getPasswordAuthentication() + { + return new java.net.PasswordAuthentication( System.getenv( "MVNW_USERNAME" ), System.getenv( "MVNW_PASSWORD" ).toCharArray() ); + } + public static void main( String[] args ) throws Exception + { + setDefault( new Downloader() ); + java.nio.file.Files.copy( java.net.URI.create( args[0] ).toURL().openStream(), java.nio.file.Paths.get( args[1] ).toAbsolutePath().normalize() ); + } + } + END + # For Cygwin/MinGW, switch paths to Windows format before running javac and java + verbose " - Compiling Downloader.java ..." + "$(native_path "$JAVACCMD")" "$(native_path "$javaSource")" || die "Failed to compile Downloader.java" + verbose " - Running Downloader.java ..." + "$(native_path "$JAVACMD")" -cp "$(native_path "$TMP_DOWNLOAD_DIR")" Downloader "$distributionUrl" "$(native_path "$targetZip")" +fi + +# If specified, validate the SHA-256 sum of the Maven distribution zip file +if [ -n "${distributionSha256Sum-}" ]; then + distributionSha256Result=false + if [ "$MVN_CMD" = mvnd.sh ]; then + echo "Checksum validation is not supported for maven-mvnd." >&2 + echo "Please disable validation by removing 'distributionSha256Sum' from your maven-wrapper.properties." >&2 + exit 1 + elif command -v sha256sum >/dev/null; then + if echo "$distributionSha256Sum $TMP_DOWNLOAD_DIR/$distributionUrlName" | sha256sum -c >/dev/null 2>&1; then + distributionSha256Result=true + fi + elif command -v shasum >/dev/null; then + if echo "$distributionSha256Sum $TMP_DOWNLOAD_DIR/$distributionUrlName" | shasum -a 256 -c >/dev/null 2>&1; then + distributionSha256Result=true + fi + else + echo "Checksum validation was requested but neither 'sha256sum' or 'shasum' are available." >&2 + echo "Please install either command, or disable validation by removing 'distributionSha256Sum' from your maven-wrapper.properties." >&2 + exit 1 + fi + if [ $distributionSha256Result = false ]; then + echo "Error: Failed to validate Maven distribution SHA-256, your Maven distribution might be compromised." >&2 + echo "If you updated your Maven version, you need to update the specified distributionSha256Sum property." >&2 + exit 1 + fi +fi + +# unzip and move +if command -v unzip >/dev/null; then + unzip ${__MVNW_QUIET_UNZIP:+"$__MVNW_QUIET_UNZIP"} "$TMP_DOWNLOAD_DIR/$distributionUrlName" -d "$TMP_DOWNLOAD_DIR" || die "failed to unzip" +else + tar xzf${__MVNW_QUIET_TAR:+"$__MVNW_QUIET_TAR"} "$TMP_DOWNLOAD_DIR/$distributionUrlName" -C "$TMP_DOWNLOAD_DIR" || die "failed to untar" +fi +printf %s\\n "$distributionUrl" >"$TMP_DOWNLOAD_DIR/$distributionUrlNameMain/mvnw.url" +mv -- "$TMP_DOWNLOAD_DIR/$distributionUrlNameMain" "$MAVEN_HOME" || [ -d "$MAVEN_HOME" ] || die "fail to move MAVEN_HOME" + +clean || : +exec_maven "$@" diff --git a/mvnw.cmd b/mvnw.cmd new file mode 100644 index 0000000..249bdf3 --- /dev/null +++ b/mvnw.cmd @@ -0,0 +1,149 @@ +<# : batch portion +@REM ---------------------------------------------------------------------------- +@REM Licensed to the Apache Software Foundation (ASF) under one +@REM or more contributor license agreements. See the NOTICE file +@REM distributed with this work for additional information +@REM regarding copyright ownership. The ASF licenses this file +@REM to you under the Apache License, Version 2.0 (the +@REM "License"); you may not use this file except in compliance +@REM with the License. You may obtain a copy of the License at +@REM +@REM http://www.apache.org/licenses/LICENSE-2.0 +@REM +@REM Unless required by applicable law or agreed to in writing, +@REM software distributed under the License is distributed on an +@REM "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY +@REM KIND, either express or implied. See the License for the +@REM specific language governing permissions and limitations +@REM under the License. +@REM ---------------------------------------------------------------------------- + +@REM ---------------------------------------------------------------------------- +@REM Apache Maven Wrapper startup batch script, version 3.3.2 +@REM +@REM Optional ENV vars +@REM MVNW_REPOURL - repo url base for downloading maven distribution +@REM MVNW_USERNAME/MVNW_PASSWORD - user and password for downloading maven +@REM MVNW_VERBOSE - true: enable verbose log; others: silence the output +@REM ---------------------------------------------------------------------------- + +@IF "%__MVNW_ARG0_NAME__%"=="" (SET __MVNW_ARG0_NAME__=%~nx0) +@SET __MVNW_CMD__= +@SET __MVNW_ERROR__= +@SET __MVNW_PSMODULEP_SAVE=%PSModulePath% +@SET PSModulePath= +@FOR /F "usebackq tokens=1* delims==" %%A IN (`powershell -noprofile "& {$scriptDir='%~dp0'; $script='%__MVNW_ARG0_NAME__%'; icm -ScriptBlock ([Scriptblock]::Create((Get-Content -Raw '%~f0'))) -NoNewScope}"`) DO @( + IF "%%A"=="MVN_CMD" (set __MVNW_CMD__=%%B) ELSE IF "%%B"=="" (echo %%A) ELSE (echo %%A=%%B) +) +@SET PSModulePath=%__MVNW_PSMODULEP_SAVE% +@SET __MVNW_PSMODULEP_SAVE= +@SET __MVNW_ARG0_NAME__= +@SET MVNW_USERNAME= +@SET MVNW_PASSWORD= +@IF NOT "%__MVNW_CMD__%"=="" (%__MVNW_CMD__% %*) +@echo Cannot start maven from wrapper >&2 && exit /b 1 +@GOTO :EOF +: end batch / begin powershell #> + +$ErrorActionPreference = "Stop" +if ($env:MVNW_VERBOSE -eq "true") { + $VerbosePreference = "Continue" +} + +# calculate distributionUrl, requires .mvn/wrapper/maven-wrapper.properties +$distributionUrl = (Get-Content -Raw "$scriptDir/.mvn/wrapper/maven-wrapper.properties" | ConvertFrom-StringData).distributionUrl +if (!$distributionUrl) { + Write-Error "cannot read distributionUrl property in $scriptDir/.mvn/wrapper/maven-wrapper.properties" +} + +switch -wildcard -casesensitive ( $($distributionUrl -replace '^.*/','') ) { + "maven-mvnd-*" { + $USE_MVND = $true + $distributionUrl = $distributionUrl -replace '-bin\.[^.]*$',"-windows-amd64.zip" + $MVN_CMD = "mvnd.cmd" + break + } + default { + $USE_MVND = $false + $MVN_CMD = $script -replace '^mvnw','mvn' + break + } +} + +# apply MVNW_REPOURL and calculate MAVEN_HOME +# maven home pattern: ~/.m2/wrapper/dists/{apache-maven-,maven-mvnd--}/ +if ($env:MVNW_REPOURL) { + $MVNW_REPO_PATTERN = if ($USE_MVND) { "/org/apache/maven/" } else { "/maven/mvnd/" } + $distributionUrl = "$env:MVNW_REPOURL$MVNW_REPO_PATTERN$($distributionUrl -replace '^.*'+$MVNW_REPO_PATTERN,'')" +} +$distributionUrlName = $distributionUrl -replace '^.*/','' +$distributionUrlNameMain = $distributionUrlName -replace '\.[^.]*$','' -replace '-bin$','' +$MAVEN_HOME_PARENT = "$HOME/.m2/wrapper/dists/$distributionUrlNameMain" +if ($env:MAVEN_USER_HOME) { + $MAVEN_HOME_PARENT = "$env:MAVEN_USER_HOME/wrapper/dists/$distributionUrlNameMain" +} +$MAVEN_HOME_NAME = ([System.Security.Cryptography.MD5]::Create().ComputeHash([byte[]][char[]]$distributionUrl) | ForEach-Object {$_.ToString("x2")}) -join '' +$MAVEN_HOME = "$MAVEN_HOME_PARENT/$MAVEN_HOME_NAME" + +if (Test-Path -Path "$MAVEN_HOME" -PathType Container) { + Write-Verbose "found existing MAVEN_HOME at $MAVEN_HOME" + Write-Output "MVN_CMD=$MAVEN_HOME/bin/$MVN_CMD" + exit $? +} + +if (! $distributionUrlNameMain -or ($distributionUrlName -eq $distributionUrlNameMain)) { + Write-Error "distributionUrl is not valid, must end with *-bin.zip, but found $distributionUrl" +} + +# prepare tmp dir +$TMP_DOWNLOAD_DIR_HOLDER = New-TemporaryFile +$TMP_DOWNLOAD_DIR = New-Item -Itemtype Directory -Path "$TMP_DOWNLOAD_DIR_HOLDER.dir" +$TMP_DOWNLOAD_DIR_HOLDER.Delete() | Out-Null +trap { + if ($TMP_DOWNLOAD_DIR.Exists) { + try { Remove-Item $TMP_DOWNLOAD_DIR -Recurse -Force | Out-Null } + catch { Write-Warning "Cannot remove $TMP_DOWNLOAD_DIR" } + } +} + +New-Item -Itemtype Directory -Path "$MAVEN_HOME_PARENT" -Force | Out-Null + +# Download and Install Apache Maven +Write-Verbose "Couldn't find MAVEN_HOME, downloading and installing it ..." +Write-Verbose "Downloading from: $distributionUrl" +Write-Verbose "Downloading to: $TMP_DOWNLOAD_DIR/$distributionUrlName" + +$webclient = New-Object System.Net.WebClient +if ($env:MVNW_USERNAME -and $env:MVNW_PASSWORD) { + $webclient.Credentials = New-Object System.Net.NetworkCredential($env:MVNW_USERNAME, $env:MVNW_PASSWORD) +} +[Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12 +$webclient.DownloadFile($distributionUrl, "$TMP_DOWNLOAD_DIR/$distributionUrlName") | Out-Null + +# If specified, validate the SHA-256 sum of the Maven distribution zip file +$distributionSha256Sum = (Get-Content -Raw "$scriptDir/.mvn/wrapper/maven-wrapper.properties" | ConvertFrom-StringData).distributionSha256Sum +if ($distributionSha256Sum) { + if ($USE_MVND) { + Write-Error "Checksum validation is not supported for maven-mvnd. `nPlease disable validation by removing 'distributionSha256Sum' from your maven-wrapper.properties." + } + Import-Module $PSHOME\Modules\Microsoft.PowerShell.Utility -Function Get-FileHash + if ((Get-FileHash "$TMP_DOWNLOAD_DIR/$distributionUrlName" -Algorithm SHA256).Hash.ToLower() -ne $distributionSha256Sum) { + Write-Error "Error: Failed to validate Maven distribution SHA-256, your Maven distribution might be compromised. If you updated your Maven version, you need to update the specified distributionSha256Sum property." + } +} + +# unzip and move +Expand-Archive "$TMP_DOWNLOAD_DIR/$distributionUrlName" -DestinationPath "$TMP_DOWNLOAD_DIR" | Out-Null +Rename-Item -Path "$TMP_DOWNLOAD_DIR/$distributionUrlNameMain" -NewName $MAVEN_HOME_NAME | Out-Null +try { + Move-Item -Path "$TMP_DOWNLOAD_DIR/$MAVEN_HOME_NAME" -Destination $MAVEN_HOME_PARENT | Out-Null +} catch { + if (! (Test-Path -Path "$MAVEN_HOME" -PathType Container)) { + Write-Error "fail to move MAVEN_HOME" + } +} finally { + try { Remove-Item $TMP_DOWNLOAD_DIR -Recurse -Force | Out-Null } + catch { Write-Warning "Cannot remove $TMP_DOWNLOAD_DIR" } +} + +Write-Output "MVN_CMD=$MAVEN_HOME/bin/$MVN_CMD" diff --git a/pid.txt b/pid.txt new file mode 100644 index 0000000..e69de29 diff --git a/pom.xml b/pom.xml new file mode 100644 index 0000000..dd50354 --- /dev/null +++ b/pom.xml @@ -0,0 +1,61 @@ + + + 4.0.0 + + org.springframework.boot + spring-boot-starter-parent + 2.7.18 + + + com.example + stream_server + 0.0.1-SNAPSHOT + demo + demo + + + + + + + + + + + + + + + 11 + + + + org.springframework.boot + spring-boot-starter-web + + + + org.springframework.boot + spring-boot-starter-test + test + + + + io.minio + minio + 3.0.10 + + + + + + + + org.springframework.boot + spring-boot-maven-plugin + + + + + diff --git a/src/main/java/com/tuoheng/steam/StreamApplication.java b/src/main/java/com/tuoheng/steam/StreamApplication.java new file mode 100644 index 0000000..441364e --- /dev/null +++ b/src/main/java/com/tuoheng/steam/StreamApplication.java @@ -0,0 +1,14 @@ +package com.tuoheng.steam; + +import org.springframework.boot.SpringApplication; +import org.springframework.boot.autoconfigure.SpringBootApplication; + +@SpringBootApplication +public class StreamApplication { + + + public static void main(String[] args) { + SpringApplication.run(StreamApplication.class, args); + } + +} diff --git a/src/main/java/com/tuoheng/steam/controller/StreamRecordController.java b/src/main/java/com/tuoheng/steam/controller/StreamRecordController.java new file mode 100644 index 0000000..a141721 --- /dev/null +++ b/src/main/java/com/tuoheng/steam/controller/StreamRecordController.java @@ -0,0 +1,53 @@ +package com.tuoheng.steam.controller; + + +import com.tuoheng.steam.controller.dto.Response; +import com.tuoheng.steam.dos.StreamTask; +import com.tuoheng.steam.service.ITaskService; +import com.tuoheng.steam.service.TaskService; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.web.bind.annotation.*; + +@RestController() +@RequestMapping("/record") +public class StreamRecordController { + + private static final Logger logger = LoggerFactory.getLogger(StreamRecordController.class); + + @Autowired + ITaskService taskService; + + @GetMapping("start") + public Response startRecording(@RequestParam String streamUrl) { + return Response.success(taskService.startTask(streamUrl)); + } + + @GetMapping("stop") + public Response stopRecording(@RequestParam String streamUrl){ + return Response.success(taskService.stopTask(streamUrl)); + } + + @GetMapping("info") + public Response getLastTask(@RequestParam String streamUrl){ + return Response.success(taskService.getLastTask(streamUrl)); + } + + /** + * 方便测试时候使用 + * @return + */ + @GetMapping("exit") + public String exit(){ + Thread thread = new Thread(() -> { + try { + taskService.stopAllTask(); + Thread.sleep(10000L); + }catch (Exception e){ + } + System.exit(0); + }); + return "OK"; + } +} diff --git a/src/main/java/com/tuoheng/steam/controller/dto/Response.java b/src/main/java/com/tuoheng/steam/controller/dto/Response.java new file mode 100644 index 0000000..6ebb510 --- /dev/null +++ b/src/main/java/com/tuoheng/steam/controller/dto/Response.java @@ -0,0 +1,23 @@ +package com.tuoheng.steam.controller.dto; + +import java.io.Serializable; + +public class Response implements Serializable { + Integer error; + T data; + + public T getData() { + return data; + } + + public void setData(T data) { + this.data = data; + } + + public static Response success(T v){ + Response r = new Response(); + r.setData(v); + r.error = 0; + return r; + } +} diff --git a/src/main/java/com/tuoheng/steam/dos/ProcessType.java b/src/main/java/com/tuoheng/steam/dos/ProcessType.java new file mode 100644 index 0000000..7b006a9 --- /dev/null +++ b/src/main/java/com/tuoheng/steam/dos/ProcessType.java @@ -0,0 +1,5 @@ +package com.tuoheng.steam.dos; + +public enum ProcessType { + RECORD,MERGE +} diff --git a/src/main/java/com/tuoheng/steam/dos/StreamProcess.java b/src/main/java/com/tuoheng/steam/dos/StreamProcess.java new file mode 100644 index 0000000..6e2efcf --- /dev/null +++ b/src/main/java/com/tuoheng/steam/dos/StreamProcess.java @@ -0,0 +1,78 @@ +package com.tuoheng.steam.dos; + +import com.tuoheng.steam.util.TimeUtils; + +import java.io.Serializable; +import java.util.Date; +import java.util.concurrent.CompletableFuture; + + +public class StreamProcess implements Serializable { + + Process process; + Date createTime; + String stopTime; + ProcessType processType; + String fileName; + + + + public String startTime(){ + return TimeUtils.formatDateToString(createTime); + } + + public String getStopTime() { + return stopTime; + } + + public void setStopTime(String stopTime) { + this.stopTime = stopTime; + } + + public CompletableFuture onExit(){ + return process.onExit(); + } + + public void destroy(){ + process.destroy(); + } + + public ProcessType getProcessType() { + return processType; + } + + public void setProcessType(ProcessType processType) { + this.processType = processType; + } + + + + public Long getInnerProcessId() { + return process.pid(); + } + + public Date getCreateTime() { + return createTime; + } + + public void setCreateTime(Date createTime) { + this.createTime = createTime; + } + + public StreamProcess(Process process, String fileName, ProcessType processType) { + this.process = process; + this.fileName = fileName; + this.createTime = new Date(); + this.processType = processType; + } + + + + public String getFileName() { + return fileName; + } + + public void setFileName(String fileName) { + this.fileName = fileName; + } +} diff --git a/src/main/java/com/tuoheng/steam/dos/StreamTask.java b/src/main/java/com/tuoheng/steam/dos/StreamTask.java new file mode 100644 index 0000000..b3410da --- /dev/null +++ b/src/main/java/com/tuoheng/steam/dos/StreamTask.java @@ -0,0 +1,126 @@ +package com.tuoheng.steam.dos; + +import com.tuoheng.steam.util.TimeUtils; + +import java.io.Serializable; +import java.util.ArrayList; +import java.util.Date; +import java.util.List; + +public class StreamTask implements Serializable { + + Long taskId; + Date startTime; + String strStartTime; + /** + * 接收到结束命令的时间 + */ + String stopCommandTime; + /** + * 执行生成TS文件的命令 + */ + List streamProcesses = new ArrayList<>(); + /** + * 合并进程的命令 + */ + StreamProcess mergeProcess; + + private static Long pid = 0L; + + private synchronized static Long generatePid(){ + if(pid==Long.MAX_VALUE){ + pid = 0L; + }else { + pid++; + } + return pid; + } + + + public String getStrStartTime() { + return strStartTime; + } + + public void setStrStartTime(String strStartTime) { + this.strStartTime = strStartTime; + } + + public void setStartTime(String startTime) { + this.strStartTime = startTime; + } + + + public String getStopCommandTime() { + return stopCommandTime; + } + + public void setStopCommandTime(String stopCommandTime) { + this.stopCommandTime = stopCommandTime; + } + + + + public List getStreamProcesses() { + return streamProcesses; + } + + public void setStreamProcesses(List streamProcesses) { + this.streamProcesses = streamProcesses; + } + + public StreamProcess getMergeProcess() { + return mergeProcess; + } + + public void setMergeProcess(StreamProcess mergeProcess) { + this.mergeProcess = mergeProcess; + } + + public String getStreamUrl() { + return streamUrl; + } + + public void setStreamUrl(String streamUrl) { + this.streamUrl = streamUrl; + } + + + String streamUrl; + + + public Long getTaskId() { + return taskId; + } + + public void setTaskId(Long taskId) { + this.taskId = taskId; + } + + public Date getStartTime() { + return startTime; + } + + public void setStartTime(Date startTime) { + this.startTime = startTime; + } + + public StreamTask(String streamUrl) { + this.taskId = generatePid(); + this.streamUrl = streamUrl; + this.startTime = new Date(); + this.strStartTime = TimeUtils.formatDateToString(this.startTime); + this.stopCommandTime = null; + } + + + + public String getOutFileName() { + return outFileName; + } + + public void setOutFileName(String outFileName) { + this.outFileName = outFileName; + } + + String outFileName; +} diff --git a/src/main/java/com/tuoheng/steam/service/ITaskService.java b/src/main/java/com/tuoheng/steam/service/ITaskService.java new file mode 100644 index 0000000..711aad8 --- /dev/null +++ b/src/main/java/com/tuoheng/steam/service/ITaskService.java @@ -0,0 +1,14 @@ +package com.tuoheng.steam.service; + +import com.tuoheng.steam.dos.StreamTask; + +public interface ITaskService { + + public StreamTask startTask(String streamUrl); + + public StreamTask stopTask(String streamUrl); + + public StreamTask getLastTask(String streamUrl); + + public void stopAllTask(); +} diff --git a/src/main/java/com/tuoheng/steam/service/TaskService.java b/src/main/java/com/tuoheng/steam/service/TaskService.java new file mode 100644 index 0000000..95db085 --- /dev/null +++ b/src/main/java/com/tuoheng/steam/service/TaskService.java @@ -0,0 +1,238 @@ +package com.tuoheng.steam.service; + +import com.tuoheng.steam.dos.StreamProcess; +import com.tuoheng.steam.dos.StreamTask; +import com.tuoheng.steam.service.innerService.ProcessService; +import com.tuoheng.steam.util.ProcessManager; +import com.tuoheng.steam.util.TimeUtils; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; +import org.springframework.util.CollectionUtils; + +import javax.annotation.PostConstruct; +import java.io.File; +import java.io.IOException; +import java.util.*; +import java.util.concurrent.*; + +@Service +public class TaskService implements ITaskService{ + + private static final Logger logger = LoggerFactory.getLogger(TaskService.class); + + @Autowired + ProcessService processService; + + /** + * 执行中的任务 + */ + ConcurrentHashMap runningTasks = new ConcurrentHashMap<>(); + + /** + * 历史任务 + */ + ConcurrentHashMap> historyTasks = new ConcurrentHashMap<>(); + + /** + * 任务池 + */ + ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1); + + @PostConstruct + public void init() { + + /** + * + */ + scheduler.scheduleWithFixedDelay(new Runnable() { + @Override + public void run() { + runningTasks.forEach((key, value) -> { + Calendar calendar = Calendar.getInstance(); + calendar.add(Calendar.HOUR, -2); + Date twoHoursAgo = calendar.getTime(); + if (value.getStartTime().before(twoHoursAgo)) { + logger.error("taskId {} 执行超时,手动关闭", key); + stopTask(key); + } + }); + + for (Map.Entry> entry : historyTasks.entrySet()) { + + for (StreamTask task : entry.getValue()) { + List streamProcesses = task.getStreamProcesses(); + for (StreamProcess process : streamProcesses) { + process.destroy(); + File file = new File(process.getFileName()); + if(file.exists()){ + file.delete(); + } + } + } + + while (entry.getValue().size()>10){ + entry.getValue().pollFirst(); + } + } + } + }, 60, 60, TimeUnit.SECONDS); + } + + public StreamTask startTask(String streamUrl) { + + /* + 当前有任务,则返回当前的任务 + */ + if (runningTasks.containsKey(streamUrl)) { + return runningTasks.get(streamUrl); + } + + /* + 当前无任务,则开启新任务 + */ + StreamTask taskInstance = new StreamTask(streamUrl); + if (runningTasks.putIfAbsent(streamUrl, taskInstance) == null) { + logger.info("streamUrl {} startTask {} ", streamUrl, taskInstance.getTaskId()); + startTask(streamUrl, taskInstance); + runningTasks.put(streamUrl, taskInstance); + return taskInstance; + } else { + return runningTasks.get(streamUrl); + } + + } + + public StreamTask stopTask(String streamUrl) { + StreamTask currentStreamTask = runningTasks.remove(streamUrl); + + if (currentStreamTask!= null) { + currentStreamTask.setStopCommandTime(TimeUtils.formatDateToString(new Date())); + + if (!CollectionUtils.isEmpty(currentStreamTask.getStreamProcesses())) { + for (StreamProcess streamProcess : currentStreamTask.getStreamProcesses()) { + logger.info("streamUrl {} taskId {} destroy Process {}", streamUrl,currentStreamTask.getTaskId() ,streamProcess.getInnerProcessId()); + streamProcess.destroy(); + } + } + + try { + String outFileName = UUID.randomUUID().toString() + ".mp4"; + StreamProcess mergeProcess = processService.mergeStream(outFileName, currentStreamTask.getStreamProcesses()); + /** + * 注册进程 + */ + ProcessManager.registerProcess(mergeProcess.getInnerProcessId()); + currentStreamTask.setMergeProcess(mergeProcess); + + + logger.info("streamUrl {} taskId {} startMergeProcess {} ", streamUrl, currentStreamTask.getTaskId(), + mergeProcess.getInnerProcessId()); + + CompletableFuture future = mergeProcess.onExit(); + future.thenRun(() -> { + logger.info("streamUrl {} taskId {} MergeProcess {} Over", streamUrl,currentStreamTask.getTaskId(), mergeProcess.getInnerProcessId()); + mergeProcess.setStopTime(TimeUtils.formatDateToString(new Date())); + ProcessManager.unRegisterProcess(mergeProcess.getInnerProcessId()); + for(StreamProcess streamProcess :currentStreamTask.getStreamProcesses()){ + File file = new File(streamProcess.getFileName()); + if(file.exists()){ + file.delete(); + } + } + }).exceptionally(ex -> { + logger.error("streamUrl {} taskId {} MergeProcess {} exceptionally", streamUrl,currentStreamTask.getTaskId(), mergeProcess.getInnerProcessId(), ex); + mergeProcess.setStopTime(TimeUtils.formatDateToString(new Date())); + ProcessManager.unRegisterProcess(mergeProcess.getInnerProcessId()); + for(StreamProcess streamProcess :currentStreamTask.getStreamProcesses()){ + File file = new File(streamProcess.getFileName()); + if(file.exists()){ + file.delete(); + } + } + return null; + }); + currentStreamTask.setOutFileName(outFileName); + + if( historyTasks.containsKey(streamUrl)){ + historyTasks.get(streamUrl).add(currentStreamTask); + }else { + historyTasks.put(streamUrl, new LinkedList<>()); + historyTasks.get(streamUrl).offerLast(currentStreamTask); + } + + return currentStreamTask; + } catch (Exception e) { + logger.info("taskId {} Stop exceptionally", streamUrl); + return null; + } + + }else { + if(historyTasks.containsKey(streamUrl)){ + return historyTasks.get(streamUrl).peekLast(); + }else { + return null; + } + } + } + + public StreamTask getLastTask(String streamUrl) { + StreamTask current = runningTasks.get(streamUrl); + if(Objects.nonNull(current)){ + return current; + }else { + if(historyTasks.containsKey(streamUrl)){ + return historyTasks.get(streamUrl).peekLast(); + } + return null; + } + } + + private void startTask(String streamUrl, StreamTask streamTask) { + + try { + /* + 任务不存在了,直接退出 + */ + if (streamTask.getStopCommandTime() != null) { + logger.info("streamUrl {} Task {} Stopped", streamUrl, streamTask.getTaskId()); + return; + } + + StreamProcess streamProcess = processService.recordStream(streamUrl); + /** + * 注册进程 + */ + ProcessManager.registerProcess(streamProcess.getInnerProcessId()); + streamTask.getStreamProcesses().add(streamProcess); + + logger.info("streamUrl {} taskId {} startProcess {} ", streamUrl, streamTask.getTaskId(),streamProcess.getInnerProcessId()); + + CompletableFuture future = streamProcess.onExit(); + future.thenRun(() -> { + logger.info("streamUrl {} taskId {} Process {} Over", streamUrl, streamTask.getTaskId(),streamProcess.getInnerProcessId()); + streamProcess.setStopTime(TimeUtils.formatDateToString(new Date())); + startTask(streamUrl, streamTask); + ProcessManager.unRegisterProcess(streamProcess.getInnerProcessId()); + }).exceptionally(ex -> { + logger.info("streamUrl {} taskId {} Process {} Exceptionally", streamUrl, streamTask.getTaskId(),streamProcess.getInnerProcessId(),ex); + streamProcess.setStopTime(TimeUtils.formatDateToString(new Date())); + startTask(streamUrl, streamTask); + ProcessManager.unRegisterProcess(streamProcess.getInnerProcessId()); + return null; + }); + + } catch (IOException e) { + logger.info("streamUrl {} taskId {} startProcess Exception", streamUrl,streamTask.getTaskId(),e); + } + + } + + + public void stopAllTask(){ + runningTasks.forEach((key, value) -> { + stopTask(key); + }); + } +} diff --git a/src/main/java/com/tuoheng/steam/service/innerService/ProcessService.java b/src/main/java/com/tuoheng/steam/service/innerService/ProcessService.java new file mode 100644 index 0000000..6057b92 --- /dev/null +++ b/src/main/java/com/tuoheng/steam/service/innerService/ProcessService.java @@ -0,0 +1,119 @@ +package com.tuoheng.steam.service.innerService; + +import com.tuoheng.steam.dos.StreamProcess; +import com.tuoheng.steam.dos.ProcessType; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.beans.factory.annotation.Value; +import org.springframework.stereotype.Service; + +import java.io.BufferedReader; +import java.io.File; +import java.io.IOException; +import java.io.InputStreamReader; +import java.util.ArrayList; +import java.util.List; +import java.util.UUID; +import java.util.concurrent.ExecutorService; +import java.util.concurrent.Executors; + +@Service +public class ProcessService { + + @Value("${srs.splitPath}") + private String splitPath; + + @Value("${srs.targetPath}") + private String targetPath; + + private static final Logger logger = LoggerFactory.getLogger(ProcessService.class); + + /** + * 任务池 + */ + ExecutorService loggingService = Executors.newCachedThreadPool(); + + + public StreamProcess recordStream(String streamUrl) throws IOException { + + String recordFileName = splitPath+'/'+UUID.randomUUID().toString() +".ts"; + String command = String.format( + "/data/ffmpeg/bin/ffmpeg -i %s -c copy -f mpegts -reconnect 1 -reconnect_streamed 1 -reconnect_delay_max 2 %s", + streamUrl, recordFileName); + logger.info("recordStream {}", command); + ProcessBuilder pb = new ProcessBuilder(command.split(" ")); + pb.redirectErrorStream(true); + Process process = pb.start(); + + loggingService.execute(new Runnable() { + @Override + public void run() { + try (BufferedReader reader = new BufferedReader( + new InputStreamReader(process.getInputStream()))) { + String line; + while ((line = reader.readLine()) != null) { +// logger.info("recordStream-------- {}",line); + } + } catch (IOException e) { + logger.info("recordStream-------- Over"); + } + } + }); + + return new StreamProcess(process,recordFileName, ProcessType.RECORD); + + }; + + public StreamProcess mergeStream(String outFileName, List streamProcesses) throws IOException { + + List fileList = new ArrayList<>(); + for(StreamProcess streamProcess : streamProcesses){ + File newFile = new File(streamProcess.getFileName()); + if(newFile.exists()){ + fileList.add(streamProcess.getFileName()); + } + } + + + if(fileList.isEmpty()){ + return null; + } + String command = ""; + outFileName = targetPath + "/" +outFileName; + if(fileList.size() == 1){ + command = String.format( + "/data/ffmpeg/bin/ffmpeg -i %s -c copy %s", + fileList.get(0), outFileName); + }else { + String filePaths = String.join("|", fileList); + command = String.format( + "/data/ffmpeg/bin/ffmpeg -i \"concat:%s\" -c copy %s", + filePaths, outFileName); + } + logger.info("mergeStream {}", command); + ProcessBuilder pb = new ProcessBuilder(command.split(" ")); + pb.redirectErrorStream(true); + Process process = pb.start(); + + loggingService.execute(new Runnable() { + @Override + public void run() { + try (BufferedReader reader = new BufferedReader( + new InputStreamReader(process.getInputStream()))) { + String line; + while ((line = reader.readLine()) != null) { + logger.info("mergeStream-------- {}",line); + } + } catch (IOException e) { + logger.info("mergeStream-------- Over"); + } + } + }); + + + return new StreamProcess(process,outFileName,ProcessType.MERGE); + + }; + + +} diff --git a/src/main/java/com/tuoheng/steam/util/ProcessManager.java b/src/main/java/com/tuoheng/steam/util/ProcessManager.java new file mode 100644 index 0000000..e8740c6 --- /dev/null +++ b/src/main/java/com/tuoheng/steam/util/ProcessManager.java @@ -0,0 +1,105 @@ +package com.tuoheng.steam.util; + +import org.springframework.stereotype.Service; + +import java.io.*; +import java.util.*; + +@Service +public class ProcessManager { + + private static final String DELIMITER = "="; + + static Map runningProcessIds = new HashMap<>(); + + static String pidPath ="pid.txt"; + + static { + runningProcessIds = loadFromFile(pidPath); + for(Map.Entry entry : runningProcessIds.entrySet()){ + try { + killProcessByPID(entry.getKey()); + }catch (Exception e){ + e.printStackTrace(); + } + } + runningProcessIds.clear(); + writeToFile(runningProcessIds); + } + + public static void registerProcess(Long process) { + runningProcessIds.put(process,TimeUtils.formatDateToString(new Date())); + writeToFile(runningProcessIds); + } + + public static void unRegisterProcess(Long process) { + runningProcessIds.remove(process); + writeToFile(runningProcessIds); + } + + public static void writeToFile(Map map){ + try (BufferedWriter writer = new BufferedWriter(new FileWriter(pidPath))) { + for (Map.Entry entry : map.entrySet()) { + writer.write(entry.getKey() + DELIMITER + entry.getValue()); + writer.newLine(); + } + }catch (IOException ignored){} + } + + public static void appendToFile(Long pid,String strDate) { + + try (BufferedWriter writer = new BufferedWriter(new FileWriter(pidPath,true))) { + writer.write(pid+ DELIMITER + strDate); + writer.newLine(); + }catch (IOException ignored) {} + } + + public static Map loadFromFile(String filePath){ + Map map = new LinkedHashMap<>(); + File file = new File(filePath); + try { + if(!file.exists()){ + file.createNewFile(); + } + }catch (Exception e){ + e.printStackTrace(); + } + try (BufferedReader reader = new BufferedReader(new FileReader(filePath))) { + String line; + while ((line = reader.readLine()) != null) { + String[] parts = line.split(ProcessManager.DELIMITER, 2); // Split into key and value + if (parts.length == 2) { + Long key = Long.parseLong(parts[0].trim()); + String value = parts[1].trim(); + map.put(key, value); + } + } + } catch (IOException e) { + throw new RuntimeException(e); + } + return map; + } + + public static void killProcessByPID(long pid) throws IOException, InterruptedException { + String os = System.getProperty("os.name").toLowerCase(); + ProcessBuilder processBuilder; + + if (os.contains("win")) { + // Windows: Use taskkill command + processBuilder = new ProcessBuilder("cmd", "/c", "taskkill /PID " + pid + " /F"); + } else { + // Linux/Unix/macOS: Use kill command + processBuilder = new ProcessBuilder("bash", "-c", "kill -9 " + pid); + } + + Process process = processBuilder.start(); + int exitCode = process.waitFor(); + + if (exitCode == 0) { + System.out.println("Process with PID " + pid + " terminated successfully."); + } else { + System.err.println("Failed to terminate process with PID " + pid + ". Exit code: " + exitCode); + } + } + +} diff --git a/src/main/java/com/tuoheng/steam/util/TimeUtils.java b/src/main/java/com/tuoheng/steam/util/TimeUtils.java new file mode 100644 index 0000000..23f7d47 --- /dev/null +++ b/src/main/java/com/tuoheng/steam/util/TimeUtils.java @@ -0,0 +1,15 @@ +package com.tuoheng.steam.util; + +import java.text.ParseException; +import java.text.SimpleDateFormat; +import java.util.Date; + +public class TimeUtils { + + private static final String DATE_FORMAT = "yyyy-MM-dd HH:mm:ss"; + + public static String formatDateToString(Date date) { + SimpleDateFormat sdf = new SimpleDateFormat(DATE_FORMAT); + return sdf.format(date); + } +} diff --git a/src/main/resources/application.properties b/src/main/resources/application.properties new file mode 100644 index 0000000..6af6220 --- /dev/null +++ b/src/main/resources/application.properties @@ -0,0 +1,4 @@ +spring.application.name=demo +server.port = 9011 +srs.splitPath=/data/java/srs/stream_server/temp +srs.targetPath=/data/java/srs/srs/trunk/objs/nginx/html \ No newline at end of file diff --git a/src/test/java/com/tuoheng/steam/StreamApplicationTests.java b/src/test/java/com/tuoheng/steam/StreamApplicationTests.java new file mode 100644 index 0000000..06ed5b3 --- /dev/null +++ b/src/test/java/com/tuoheng/steam/StreamApplicationTests.java @@ -0,0 +1,34 @@ +package com.tuoheng.steam; + +import com.tuoheng.steam.service.TaskService; +import org.junit.jupiter.api.Test; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.boot.test.context.SpringBootTest; + +@SpringBootTest +class StreamApplicationTests { + + @Autowired + TaskService taskService; + + @Test + void contextLoads() { + taskService.startTask("rtmp://stream.t-aaron.com/live/123"); + + boolean stop = false; + + while (!stop) { + + try { + Thread.sleep(60000L); + }catch (Exception e){ + + } + + } + + + taskService.stopAllTask(); + } + +}