diff --git a/ThreadPool/build.gradle b/ThreadPool/build.gradle new file mode 100644 index 0000000..8300af3 --- /dev/null +++ b/ThreadPool/build.gradle @@ -0,0 +1,15 @@ +group 'ru.ifmo' +version '1.0-SNAPSHOT' + +apply plugin: 'java' + +sourceCompatibility = 1.8 + +repositories { + mavenCentral() +} + +dependencies { + testCompile group: 'junit', name: 'junit', version: '4.12' + testCompile group: 'org.assertj', name: 'assertj-core', version: '3.9.0' +} diff --git a/ThreadPool/gradle/wrapper/gradle-wrapper.properties b/ThreadPool/gradle/wrapper/gradle-wrapper.properties new file mode 100644 index 0000000..b9d8cbd --- /dev/null +++ b/ThreadPool/gradle/wrapper/gradle-wrapper.properties @@ -0,0 +1,6 @@ +#Sat Nov 10 18:12:31 MSK 2018 +distributionBase=GRADLE_USER_HOME +distributionPath=wrapper/dists +zipStoreBase=GRADLE_USER_HOME +zipStorePath=wrapper/dists +distributionUrl=https\://services.gradle.org/distributions/gradle-4.0-bin.zip diff --git a/ThreadPool/gradlew b/ThreadPool/gradlew new file mode 100755 index 0000000..cccdd3d --- /dev/null +++ b/ThreadPool/gradlew @@ -0,0 +1,172 @@ +#!/usr/bin/env sh + +############################################################################## +## +## 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="" + +# 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, switch paths to Windows format before running java +if $cygwin ; 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=$((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" + +# by default we should be in the correct project dir, but when run from Finder on Mac, the cwd is wrong +if [ "$(uname)" = "Darwin" ] && [ "$HOME" = "$PWD" ]; then + cd "$(dirname "$0")" +fi + +exec "$JAVACMD" "$@" diff --git a/ThreadPool/gradlew.bat b/ThreadPool/gradlew.bat new file mode 100644 index 0000000..e95643d --- /dev/null +++ b/ThreadPool/gradlew.bat @@ -0,0 +1,84 @@ +@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= + +@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/ThreadPool/settings.gradle b/ThreadPool/settings.gradle new file mode 100644 index 0000000..64ea02f --- /dev/null +++ b/ThreadPool/settings.gradle @@ -0,0 +1,2 @@ +rootProject.name = 'threadpool' + diff --git a/ThreadPool/src/main/java/ru/ifmo/threadpool/LightExecutionException.java b/ThreadPool/src/main/java/ru/ifmo/threadpool/LightExecutionException.java new file mode 100644 index 0000000..7975036 --- /dev/null +++ b/ThreadPool/src/main/java/ru/ifmo/threadpool/LightExecutionException.java @@ -0,0 +1,9 @@ +package ru.ifmo.threadpool; + +public class LightExecutionException extends Exception { + + LightExecutionException(Throwable throwable) { + super(throwable); + } + +} diff --git a/ThreadPool/src/main/java/ru/ifmo/threadpool/LightFuture.java b/ThreadPool/src/main/java/ru/ifmo/threadpool/LightFuture.java new file mode 100644 index 0000000..56c5a48 --- /dev/null +++ b/ThreadPool/src/main/java/ru/ifmo/threadpool/LightFuture.java @@ -0,0 +1,12 @@ +package ru.ifmo.threadpool; + +import java.util.function.Function; + +public interface LightFuture { + + boolean isReady(); + + R get() throws LightExecutionException; + + LightFuture thenApply(Function function); +} diff --git a/ThreadPool/src/main/java/ru/ifmo/threadpool/ThreadPool.java b/ThreadPool/src/main/java/ru/ifmo/threadpool/ThreadPool.java new file mode 100644 index 0000000..a67bd41 --- /dev/null +++ b/ThreadPool/src/main/java/ru/ifmo/threadpool/ThreadPool.java @@ -0,0 +1,11 @@ +package ru.ifmo.threadpool; + +import java.util.function.Supplier; + +public interface ThreadPool { + + LightFuture submit(Supplier supplier); + + void shutdown(); + +} diff --git a/ThreadPool/src/main/java/ru/ifmo/threadpool/ThreadPoolImpl.java b/ThreadPool/src/main/java/ru/ifmo/threadpool/ThreadPoolImpl.java new file mode 100644 index 0000000..14bed99 --- /dev/null +++ b/ThreadPool/src/main/java/ru/ifmo/threadpool/ThreadPoolImpl.java @@ -0,0 +1,144 @@ +package ru.ifmo.threadpool; + +import java.util.ArrayList; +import java.util.LinkedList; +import java.util.List; +import java.util.Queue; +import java.util.function.Function; +import java.util.function.Supplier; + +public class ThreadPoolImpl implements ThreadPool { + + private final List threads = new ArrayList<>(); + + private volatile boolean isWorking; + + private final Queue taskQueue = new LinkedList<>(); + + public ThreadPoolImpl(int threadsNumber) { + for (int i = 0; i < threadsNumber; i++) { + Thread thread = new Thread(new ThreadPoolWorker()); + thread.start(); + threads.add(thread); + } + isWorking = true; + } + + @Override + public LightFuture submit(Supplier supplier) { + if (!isWorking) { + return null; + } + LightFutureImpl task = new LightFutureImpl<>(supplier); + synchronized (taskQueue) { + pushTaskToQueue(task); + } + return task; + } + + @Override + public void shutdown() { + threads.forEach(Thread::interrupt); + for (Thread thread : threads) { + try { + thread.join(); + } catch (InterruptedException ignored) { + } + } + isWorking = false; + } + + private void pushTaskToQueue(LightFutureImpl task) { + synchronized (taskQueue) { + taskQueue.add(task); + taskQueue.notify(); + } + } + + private class LightFutureImpl implements LightFuture, Runnable { + + private final Supplier task; + private volatile boolean isReady = false; + private volatile R result = null; + private volatile LightExecutionException exception = null; + private List waitingTasks = new ArrayList<>(); + + private LightFutureImpl(Supplier task) { + this.task = task; + } + + @Override + public boolean isReady() { + return isReady; + } + + @Override + public R get() throws LightExecutionException { + while (!isReady()) { + Thread.yield(); + } + if (exception != null) { + throw exception; + } + return result; + } + + @Override + public LightFuture thenApply(Function function) { + Supplier supplier = () -> { + try { + return function.apply(this.get()); + } catch (LightExecutionException e) { + throw new RuntimeException(e.getMessage(), e.getCause()); + } + }; + LightFutureImpl nextTask = new LightFutureImpl<>(supplier); + synchronized (this) { + if (!isReady) { + waitingTasks.add(nextTask); + } else { + ThreadPoolImpl.this.pushTaskToQueue(nextTask); + } + } + return nextTask; + } + + private synchronized void submitWaitingTasks() { + waitingTasks.forEach(ThreadPoolImpl.this::pushTaskToQueue); + waitingTasks.clear(); + } + + @Override + public void run() { + try { + result = task.get(); + } catch (Exception e) { + exception = new LightExecutionException(e); + } + isReady = true; + submitWaitingTasks(); + } + } + + private class ThreadPoolWorker implements Runnable { + + @Override + public void run() { + try { + while (!Thread.currentThread().isInterrupted()) { + LightFutureImpl task; + synchronized (taskQueue) { + if (taskQueue.isEmpty()) { + taskQueue.wait(); + } + task = taskQueue.poll(); + } + if (task != null) { + task.run(); + } + } + } catch (InterruptedException ignored) { + } + } + } +} diff --git a/ThreadPool/src/test/java/ru/ifmo/threadpool/ThreadPoolImplTest.java b/ThreadPool/src/test/java/ru/ifmo/threadpool/ThreadPoolImplTest.java new file mode 100644 index 0000000..245b71b --- /dev/null +++ b/ThreadPool/src/test/java/ru/ifmo/threadpool/ThreadPoolImplTest.java @@ -0,0 +1,79 @@ +package ru.ifmo.threadpool; + +import org.junit.Test; + +import java.util.ArrayList; +import java.util.List; + +import static org.assertj.core.api.AssertionsForClassTypes.assertThat; + + +public class ThreadPoolImplTest { + + private final int threadNumber = 4; + + @Test + public void TestCorrectThreadsNumber() { + int otherThreads = Thread.activeCount(); + ThreadPool threadPool = new ThreadPoolImpl(threadNumber); + assertThat(Thread.activeCount() - otherThreads).isEqualTo(threadNumber); + } + + @Test + public void testManyIndependentTasks() throws LightExecutionException { + ThreadPool threadPool = new ThreadPoolImpl(threadNumber); + List> tasks = new ArrayList<>(); + for (int i = 0; i < 1000; i++) { + final int j = i; + tasks.add(threadPool.submit(() -> j * j)); + } + for (int i = 0; i < 1000; i++) { + assertThat(tasks.get(i).get()).isEqualTo(i * i); + } + } + + @Test + public void testDependentTasksCorrectOrder() throws LightExecutionException { + ThreadPool threadPool = new ThreadPoolImpl(threadNumber); + List> tasks = new ArrayList<>(); + tasks.add(threadPool.submit(() -> 1)); + for (int i = 1; i < 10; i++) { + tasks.add(tasks.get(i - 1).thenApply(k -> k * 2)); + } + for (int i = 0; i < 10; i++) { + assertThat(tasks.get(i).get()).isEqualTo((int) Math.pow(2, i)); + } + } + + @Test + public void testDependentTasksReverseOrder() throws LightExecutionException { + ThreadPool threadPool = new ThreadPoolImpl(threadNumber); + List> tasks = new ArrayList<>(); + tasks.add(threadPool.submit(() -> 1)); + for (int i = 1; i < 10; i++) { + tasks.add(tasks.get(i - 1).thenApply(k -> k * 2)); + } + for (int i = 9; i >= 0; i--) { + assertThat(tasks.get(i).get()).isEqualTo((int) Math.pow(2, i)); + } + } + + @Test(expected = LightExecutionException.class) + public void testLightExecutionException() throws LightExecutionException { + ThreadPool threadPool = new ThreadPoolImpl(threadNumber); + LightFuture task = threadPool.submit(() -> { + throw new RuntimeException("kek!"); + }); + task.get(); + } + + @Test + public void TestShutdown() { + int otherThreads = Thread.activeCount(); + ThreadPool threadPool = new ThreadPoolImpl(threadNumber); + assertThat(Thread.activeCount() - otherThreads).isEqualTo(threadNumber); + threadPool.shutdown(); + assertThat(Thread.activeCount()).isEqualTo(otherThreads); + } + +}