├── .gitignore
├── LICENSE.md
├── README.md
├── build.gradle
├── gradle
└── wrapper
│ ├── gradle-wrapper.jar
│ └── gradle-wrapper.properties
├── gradlew
├── gradlew.bat
└── src
├── main
├── java
│ └── nextstep
│ │ └── subway
│ │ └── SubwayApplication.java
└── resources
│ ├── application.properties
│ └── logback-access.xml
└── test
└── java
└── nextstep
├── study
└── JgraphtTest.java
└── subway
├── acceptance
└── AcceptanceTest.java
├── unit
├── LineServiceMockTest.java
├── LineServiceTest.java
└── LineTest.java
└── utils
└── DatabaseCleanup.java
/.gitignore:
--------------------------------------------------------------------------------
1 | .gradle
2 | build/
3 | !gradle/wrapper/gradle-wrapper.jar
4 | !**/src/main/**
5 | !**/src/test/**
6 |
7 | ### STS ###
8 | .apt_generated
9 | .classpath
10 | .factorypath
11 | .project
12 | .settings
13 | .springBeans
14 | .sts4-cache
15 |
16 | ### IntelliJ IDEA ###
17 | .idea
18 | *.iws
19 | *.iml
20 | *.ipr
21 | out/
22 |
--------------------------------------------------------------------------------
/LICENSE.md:
--------------------------------------------------------------------------------
1 | MIT License
2 |
3 | Copyright (c) 2020 next-step
4 |
5 | Permission is hereby granted, free of charge, to any person obtaining a copy
6 | of this software and associated documentation files (the "Software"), to deal
7 | in the Software without restriction, including without limitation the rights
8 | to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9 | copies of the Software, and to permit persons to whom the Software is
10 | furnished to do so, subject to the following conditions:
11 |
12 | The above copyright notice and this permission notice shall be included in all
13 | copies or substantial portions of the Software.
14 |
15 | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 | IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 | FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18 | AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 | LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20 | OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21 | SOFTWARE.
22 |
--------------------------------------------------------------------------------
/README.md:
--------------------------------------------------------------------------------
1 | # 지하철 노선도 미션
2 | [ATDD 강의](https://edu.nextstep.camp/c/R89PYi5H) 실습을 위한 지하철 노선도 애플리케이션
--------------------------------------------------------------------------------
/build.gradle:
--------------------------------------------------------------------------------
1 | plugins {
2 | id 'org.springframework.boot' version '2.7.1'
3 | id 'io.spring.dependency-management' version '1.0.11.RELEASE'
4 | id 'java'
5 | }
6 |
7 | group = 'nextstep'
8 | version = '0.0.1-SNAPSHOT'
9 | sourceCompatibility = '11'
10 |
11 | repositories {
12 | mavenCentral()
13 | }
14 |
15 | dependencies {
16 | // spring
17 | implementation 'org.springframework.boot:spring-boot-starter-web'
18 | implementation 'org.springframework.boot:spring-boot-starter-data-jpa'
19 |
20 | // log
21 | implementation 'net.rakugakibox.spring.boot:logback-access-spring-boot-starter:2.7.1'
22 |
23 | // jgraph
24 | implementation 'org.jgrapht:jgrapht-core:1.0.1'
25 |
26 | testImplementation 'org.springframework.boot:spring-boot-starter-test'
27 | testImplementation 'io.rest-assured:rest-assured:4.5.1'
28 |
29 | runtimeOnly 'com.h2database:h2'
30 | }
31 |
32 | test {
33 | useJUnitPlatform()
34 | }
35 |
--------------------------------------------------------------------------------
/gradle/wrapper/gradle-wrapper.jar:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/next-step/atdd-subway-path/6af4278d8fa41fdd68c0a53e10092d0792abd796/gradle/wrapper/gradle-wrapper.jar
--------------------------------------------------------------------------------
/gradle/wrapper/gradle-wrapper.properties:
--------------------------------------------------------------------------------
1 | distributionBase=GRADLE_USER_HOME
2 | distributionPath=wrapper/dists
3 | distributionUrl=https\://services.gradle.org/distributions/gradle-7.5.1-bin.zip
4 | zipStoreBase=GRADLE_USER_HOME
5 | zipStorePath=wrapper/dists
6 |
--------------------------------------------------------------------------------
/gradlew:
--------------------------------------------------------------------------------
1 | #!/usr/bin/env sh
2 |
3 | #
4 | # Copyright 2015 the original author or authors.
5 | #
6 | # Licensed under the Apache License, Version 2.0 (the "License");
7 | # you may not use this file except in compliance with the License.
8 | # You may obtain a copy of the License at
9 | #
10 | # https://www.apache.org/licenses/LICENSE-2.0
11 | #
12 | # Unless required by applicable law or agreed to in writing, software
13 | # distributed under the License is distributed on an "AS IS" BASIS,
14 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 | # See the License for the specific language governing permissions and
16 | # limitations under the License.
17 | #
18 |
19 | ##############################################################################
20 | ##
21 | ## Gradle start up script for UN*X
22 | ##
23 | ##############################################################################
24 |
25 | # Attempt to set APP_HOME
26 | # Resolve links: $0 may be a link
27 | PRG="$0"
28 | # Need this for relative symlinks.
29 | while [ -h "$PRG" ] ; do
30 | ls=`ls -ld "$PRG"`
31 | link=`expr "$ls" : '.*-> \(.*\)$'`
32 | if expr "$link" : '/.*' > /dev/null; then
33 | PRG="$link"
34 | else
35 | PRG=`dirname "$PRG"`"/$link"
36 | fi
37 | done
38 | SAVED="`pwd`"
39 | cd "`dirname \"$PRG\"`/" >/dev/null
40 | APP_HOME="`pwd -P`"
41 | cd "$SAVED" >/dev/null
42 |
43 | APP_NAME="Gradle"
44 | APP_BASE_NAME=`basename "$0"`
45 |
46 | # Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
47 | DEFAULT_JVM_OPTS='"-Xmx64m" "-Xms64m"'
48 |
49 | # Use the maximum available, or set MAX_FD != -1 to use that value.
50 | MAX_FD="maximum"
51 |
52 | warn () {
53 | echo "$*"
54 | }
55 |
56 | die () {
57 | echo
58 | echo "$*"
59 | echo
60 | exit 1
61 | }
62 |
63 | # OS specific support (must be 'true' or 'false').
64 | cygwin=false
65 | msys=false
66 | darwin=false
67 | nonstop=false
68 | case "`uname`" in
69 | CYGWIN* )
70 | cygwin=true
71 | ;;
72 | Darwin* )
73 | darwin=true
74 | ;;
75 | MINGW* )
76 | msys=true
77 | ;;
78 | NONSTOP* )
79 | nonstop=true
80 | ;;
81 | esac
82 |
83 | CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar
84 |
85 |
86 | # Determine the Java command to use to start the JVM.
87 | if [ -n "$JAVA_HOME" ] ; then
88 | if [ -x "$JAVA_HOME/jre/sh/java" ] ; then
89 | # IBM's JDK on AIX uses strange locations for the executables
90 | JAVACMD="$JAVA_HOME/jre/sh/java"
91 | else
92 | JAVACMD="$JAVA_HOME/bin/java"
93 | fi
94 | if [ ! -x "$JAVACMD" ] ; then
95 | die "ERROR: JAVA_HOME is set to an invalid directory: $JAVA_HOME
96 |
97 | Please set the JAVA_HOME variable in your environment to match the
98 | location of your Java installation."
99 | fi
100 | else
101 | JAVACMD="java"
102 | which java >/dev/null 2>&1 || die "ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
103 |
104 | Please set the JAVA_HOME variable in your environment to match the
105 | location of your Java installation."
106 | fi
107 |
108 | # Increase the maximum file descriptors if we can.
109 | if [ "$cygwin" = "false" -a "$darwin" = "false" -a "$nonstop" = "false" ] ; then
110 | MAX_FD_LIMIT=`ulimit -H -n`
111 | if [ $? -eq 0 ] ; then
112 | if [ "$MAX_FD" = "maximum" -o "$MAX_FD" = "max" ] ; then
113 | MAX_FD="$MAX_FD_LIMIT"
114 | fi
115 | ulimit -n $MAX_FD
116 | if [ $? -ne 0 ] ; then
117 | warn "Could not set maximum file descriptor limit: $MAX_FD"
118 | fi
119 | else
120 | warn "Could not query maximum file descriptor limit: $MAX_FD_LIMIT"
121 | fi
122 | fi
123 |
124 | # For Darwin, add options to specify how the application appears in the dock
125 | if $darwin; then
126 | GRADLE_OPTS="$GRADLE_OPTS \"-Xdock:name=$APP_NAME\" \"-Xdock:icon=$APP_HOME/media/gradle.icns\""
127 | fi
128 |
129 | # For Cygwin or MSYS, switch paths to Windows format before running java
130 | if [ "$cygwin" = "true" -o "$msys" = "true" ] ; then
131 | APP_HOME=`cygpath --path --mixed "$APP_HOME"`
132 | CLASSPATH=`cygpath --path --mixed "$CLASSPATH"`
133 |
134 | JAVACMD=`cygpath --unix "$JAVACMD"`
135 |
136 | # We build the pattern for arguments to be converted via cygpath
137 | ROOTDIRSRAW=`find -L / -maxdepth 1 -mindepth 1 -type d 2>/dev/null`
138 | SEP=""
139 | for dir in $ROOTDIRSRAW ; do
140 | ROOTDIRS="$ROOTDIRS$SEP$dir"
141 | SEP="|"
142 | done
143 | OURCYGPATTERN="(^($ROOTDIRS))"
144 | # Add a user-defined pattern to the cygpath arguments
145 | if [ "$GRADLE_CYGPATTERN" != "" ] ; then
146 | OURCYGPATTERN="$OURCYGPATTERN|($GRADLE_CYGPATTERN)"
147 | fi
148 | # Now convert the arguments - kludge to limit ourselves to /bin/sh
149 | i=0
150 | for arg in "$@" ; do
151 | CHECK=`echo "$arg"|egrep -c "$OURCYGPATTERN" -`
152 | CHECK2=`echo "$arg"|egrep -c "^-"` ### Determine if an option
153 |
154 | if [ $CHECK -ne 0 ] && [ $CHECK2 -eq 0 ] ; then ### Added a condition
155 | eval `echo args$i`=`cygpath --path --ignore --mixed "$arg"`
156 | else
157 | eval `echo args$i`="\"$arg\""
158 | fi
159 | i=`expr $i + 1`
160 | done
161 | case $i in
162 | 0) set -- ;;
163 | 1) set -- "$args0" ;;
164 | 2) set -- "$args0" "$args1" ;;
165 | 3) set -- "$args0" "$args1" "$args2" ;;
166 | 4) set -- "$args0" "$args1" "$args2" "$args3" ;;
167 | 5) set -- "$args0" "$args1" "$args2" "$args3" "$args4" ;;
168 | 6) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" ;;
169 | 7) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" ;;
170 | 8) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" ;;
171 | 9) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" "$args8" ;;
172 | esac
173 | fi
174 |
175 | # Escape application args
176 | save () {
177 | for i do printf %s\\n "$i" | sed "s/'/'\\\\''/g;1s/^/'/;\$s/\$/' \\\\/" ; done
178 | echo " "
179 | }
180 | APP_ARGS=`save "$@"`
181 |
182 | # Collect all arguments for the java command, following the shell quoting and substitution rules
183 | eval set -- $DEFAULT_JVM_OPTS $JAVA_OPTS $GRADLE_OPTS "\"-Dorg.gradle.appname=$APP_BASE_NAME\"" -classpath "\"$CLASSPATH\"" org.gradle.wrapper.GradleWrapperMain "$APP_ARGS"
184 |
185 | exec "$JAVACMD" "$@"
186 |
--------------------------------------------------------------------------------
/gradlew.bat:
--------------------------------------------------------------------------------
1 | @rem
2 | @rem Copyright 2015 the original author or authors.
3 | @rem
4 | @rem Licensed under the Apache License, Version 2.0 (the "License");
5 | @rem you may not use this file except in compliance with the License.
6 | @rem You may obtain a copy of the License at
7 | @rem
8 | @rem https://www.apache.org/licenses/LICENSE-2.0
9 | @rem
10 | @rem Unless required by applicable law or agreed to in writing, software
11 | @rem distributed under the License is distributed on an "AS IS" BASIS,
12 | @rem WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | @rem See the License for the specific language governing permissions and
14 | @rem limitations under the License.
15 | @rem
16 |
17 | @if "%DEBUG%" == "" @echo off
18 | @rem ##########################################################################
19 | @rem
20 | @rem Gradle startup script for Windows
21 | @rem
22 | @rem ##########################################################################
23 |
24 | @rem Set local scope for the variables with windows NT shell
25 | if "%OS%"=="Windows_NT" setlocal
26 |
27 | set DIRNAME=%~dp0
28 | if "%DIRNAME%" == "" set DIRNAME=.
29 | set APP_BASE_NAME=%~n0
30 | set APP_HOME=%DIRNAME%
31 |
32 | @rem Resolve any "." and ".." in APP_HOME to make it shorter.
33 | for %%i in ("%APP_HOME%") do set APP_HOME=%%~fi
34 |
35 | @rem Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
36 | set DEFAULT_JVM_OPTS="-Xmx64m" "-Xms64m"
37 |
38 | @rem Find java.exe
39 | if defined JAVA_HOME goto findJavaFromJavaHome
40 |
41 | set JAVA_EXE=java.exe
42 | %JAVA_EXE% -version >NUL 2>&1
43 | if "%ERRORLEVEL%" == "0" goto init
44 |
45 | echo.
46 | echo ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
47 | echo.
48 | echo Please set the JAVA_HOME variable in your environment to match the
49 | echo location of your Java installation.
50 |
51 | goto fail
52 |
53 | :findJavaFromJavaHome
54 | set JAVA_HOME=%JAVA_HOME:"=%
55 | set JAVA_EXE=%JAVA_HOME%/bin/java.exe
56 |
57 | if exist "%JAVA_EXE%" goto init
58 |
59 | echo.
60 | echo ERROR: JAVA_HOME is set to an invalid directory: %JAVA_HOME%
61 | echo.
62 | echo Please set the JAVA_HOME variable in your environment to match the
63 | echo location of your Java installation.
64 |
65 | goto fail
66 |
67 | :init
68 | @rem Get command-line arguments, handling Windows variants
69 |
70 | if not "%OS%" == "Windows_NT" goto win9xME_args
71 |
72 | :win9xME_args
73 | @rem Slurp the command line arguments.
74 | set CMD_LINE_ARGS=
75 | set _SKIP=2
76 |
77 | :win9xME_args_slurp
78 | if "x%~1" == "x" goto execute
79 |
80 | set CMD_LINE_ARGS=%*
81 |
82 | :execute
83 | @rem Setup the command line
84 |
85 | set CLASSPATH=%APP_HOME%\gradle\wrapper\gradle-wrapper.jar
86 |
87 |
88 | @rem Execute Gradle
89 | "%JAVA_EXE%" %DEFAULT_JVM_OPTS% %JAVA_OPTS% %GRADLE_OPTS% "-Dorg.gradle.appname=%APP_BASE_NAME%" -classpath "%CLASSPATH%" org.gradle.wrapper.GradleWrapperMain %CMD_LINE_ARGS%
90 |
91 | :end
92 | @rem End local scope for the variables with windows NT shell
93 | if "%ERRORLEVEL%"=="0" goto mainEnd
94 |
95 | :fail
96 | rem Set variable GRADLE_EXIT_CONSOLE if you need the _script_ return code instead of
97 | rem the _cmd.exe /c_ return code!
98 | if not "" == "%GRADLE_EXIT_CONSOLE%" exit 1
99 | exit /b 1
100 |
101 | :mainEnd
102 | if "%OS%"=="Windows_NT" endlocal
103 |
104 | :omega
105 |
--------------------------------------------------------------------------------
/src/main/java/nextstep/subway/SubwayApplication.java:
--------------------------------------------------------------------------------
1 | package nextstep.subway;
2 |
3 | import org.springframework.boot.SpringApplication;
4 | import org.springframework.boot.autoconfigure.SpringBootApplication;
5 |
6 | @SpringBootApplication
7 | public class SubwayApplication {
8 |
9 | public static void main(String[] args) {
10 | SpringApplication.run(SubwayApplication.class, args);
11 | }
12 |
13 | }
14 |
--------------------------------------------------------------------------------
/src/main/resources/application.properties:
--------------------------------------------------------------------------------
1 | spring.jpa.properties.hibernate.show_sql=true
2 | spring.jpa.properties.hibernate.format_sql=true
--------------------------------------------------------------------------------
/src/main/resources/logback-access.xml:
--------------------------------------------------------------------------------
1 |
2 |
3 |
4 | %fullRequest%n%n%fullResponse
5 |
6 |
7 |
8 |
--------------------------------------------------------------------------------
/src/test/java/nextstep/study/JgraphtTest.java:
--------------------------------------------------------------------------------
1 | package nextstep.study;
2 |
3 | import org.jgrapht.GraphPath;
4 | import org.jgrapht.alg.shortestpath.DijkstraShortestPath;
5 | import org.jgrapht.alg.shortestpath.KShortestPaths;
6 | import org.jgrapht.graph.DefaultWeightedEdge;
7 | import org.jgrapht.graph.WeightedMultigraph;
8 | import org.junit.jupiter.api.Test;
9 |
10 | import java.util.List;
11 |
12 | import static org.assertj.core.api.Assertions.assertThat;
13 |
14 |
15 | class JgraphtTest {
16 | @Test
17 | void getDijkstraShortestPath() {
18 | String source = "v3";
19 | String target = "v1";
20 |
21 | WeightedMultigraph graph = new WeightedMultigraph(DefaultWeightedEdge.class);
22 | graph.addVertex("v1");
23 | graph.addVertex("v2");
24 | graph.addVertex("v3");
25 | graph.setEdgeWeight(graph.addEdge("v1", "v2"), 2);
26 | graph.setEdgeWeight(graph.addEdge("v2", "v3"), 2);
27 | graph.setEdgeWeight(graph.addEdge("v1", "v3"), 100);
28 |
29 | DijkstraShortestPath dijkstraShortestPath = new DijkstraShortestPath(graph);
30 | List shortestPath = dijkstraShortestPath.getPath(source, target).getVertexList();
31 |
32 | assertThat(shortestPath.size()).isEqualTo(3);
33 | }
34 |
35 | @Test
36 | void getKShortestPaths() {
37 | String source = "v3";
38 | String target = "v1";
39 |
40 | WeightedMultigraph graph = new WeightedMultigraph(DefaultWeightedEdge.class);
41 | graph.addVertex("v1");
42 | graph.addVertex("v2");
43 | graph.addVertex("v3");
44 | graph.setEdgeWeight(graph.addEdge("v1", "v2"), 2);
45 | graph.setEdgeWeight(graph.addEdge("v2", "v3"), 2);
46 | graph.setEdgeWeight(graph.addEdge("v1", "v3"), 100);
47 |
48 | List paths = new KShortestPaths(graph, 100).getPaths(source, target);
49 |
50 | assertThat(paths).hasSize(2);
51 | paths.stream()
52 | .forEach(it -> {
53 | assertThat(it.getVertexList()).startsWith(source);
54 | assertThat(it.getVertexList()).endsWith(target);
55 | });
56 | }
57 | }
--------------------------------------------------------------------------------
/src/test/java/nextstep/subway/acceptance/AcceptanceTest.java:
--------------------------------------------------------------------------------
1 | package nextstep.subway.acceptance;
2 |
3 | import nextstep.subway.utils.DatabaseCleanup;
4 | import org.junit.jupiter.api.BeforeEach;
5 | import org.springframework.beans.factory.annotation.Autowired;
6 | import org.springframework.boot.test.context.SpringBootTest;
7 | import org.springframework.test.context.ActiveProfiles;
8 |
9 | @ActiveProfiles("test")
10 | @SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
11 | public class AcceptanceTest {
12 | @Autowired
13 | private DatabaseCleanup databaseCleanup;
14 |
15 | @BeforeEach
16 | public void setUp() {
17 | databaseCleanup.execute();
18 | }
19 | }
20 |
--------------------------------------------------------------------------------
/src/test/java/nextstep/subway/unit/LineServiceMockTest.java:
--------------------------------------------------------------------------------
1 | package nextstep.subway.unit;
2 |
3 | import org.junit.jupiter.api.Test;
4 | import org.junit.jupiter.api.extension.ExtendWith;
5 | import org.mockito.Mock;
6 | import org.mockito.junit.jupiter.MockitoExtension;
7 |
8 | @ExtendWith(MockitoExtension.class)
9 | public class LineServiceMockTest {
10 | @Mock
11 | private LineRepository lineRepository;
12 | @Mock
13 | private StationService stationService;
14 |
15 | @Test
16 | void addSection() {
17 | // given
18 | // lineRepository, stationService stub 설정을 통해 초기값 셋팅
19 |
20 | // when
21 | // lineService.addSection 호출
22 |
23 | // then
24 | // lineService.findLineById 메서드를 통해 검증
25 | }
26 | }
27 |
--------------------------------------------------------------------------------
/src/test/java/nextstep/subway/unit/LineServiceTest.java:
--------------------------------------------------------------------------------
1 | package nextstep.subway.unit;
2 |
3 | import org.junit.jupiter.api.Test;
4 | import org.springframework.beans.factory.annotation.Autowired;
5 | import org.springframework.boot.test.context.SpringBootTest;
6 | import org.springframework.transaction.annotation.Transactional;
7 |
8 | @SpringBootTest
9 | @Transactional
10 | public class LineServiceTest {
11 | @Autowired
12 | private StationRepository stationRepository;
13 | @Autowired
14 | private LineRepository lineRepository;
15 |
16 | @Autowired
17 | private LineService lineService;
18 |
19 | @Test
20 | void addSection() {
21 | // given
22 | // stationRepository와 lineRepository를 활용하여 초기값 셋팅
23 |
24 | // when
25 | // lineService.addSection 호출
26 |
27 | // then
28 | // line.getSections 메서드를 통해 검증
29 | }
30 | }
31 |
--------------------------------------------------------------------------------
/src/test/java/nextstep/subway/unit/LineTest.java:
--------------------------------------------------------------------------------
1 | package nextstep.subway.unit;
2 |
3 | import org.junit.jupiter.api.Test;
4 |
5 | class LineTest {
6 | @Test
7 | void addSection() {
8 | }
9 |
10 | @Test
11 | void getStations() {
12 | }
13 |
14 | @Test
15 | void removeSection() {
16 | }
17 | }
18 |
--------------------------------------------------------------------------------
/src/test/java/nextstep/subway/utils/DatabaseCleanup.java:
--------------------------------------------------------------------------------
1 | package nextstep.subway.utils;
2 |
3 | import org.springframework.beans.factory.InitializingBean;
4 | import org.springframework.context.annotation.Profile;
5 | import org.springframework.stereotype.Service;
6 | import org.springframework.transaction.annotation.Transactional;
7 |
8 | import javax.persistence.Entity;
9 | import javax.persistence.EntityManager;
10 | import javax.persistence.PersistenceContext;
11 | import java.util.List;
12 | import java.util.stream.Collectors;
13 |
14 | @Profile("test")
15 | @Service
16 | public class DatabaseCleanup implements InitializingBean {
17 | @PersistenceContext
18 | private EntityManager entityManager;
19 |
20 | private List tableNames;
21 |
22 | @Override
23 | public void afterPropertiesSet() {
24 | tableNames = entityManager.getMetamodel().getEntities().stream()
25 | .filter(entity -> entity.getJavaType().getAnnotation(Entity.class) != null)
26 | .map(entity -> entity.getName())
27 | .collect(Collectors.toList());
28 | }
29 |
30 | @Transactional
31 | public void execute() {
32 | entityManager.flush();
33 | entityManager.createNativeQuery("SET REFERENTIAL_INTEGRITY FALSE").executeUpdate();
34 | for (String tableName : tableNames) {
35 | entityManager.createNativeQuery("TRUNCATE TABLE " + tableName).executeUpdate();
36 | entityManager.createNativeQuery("ALTER TABLE " + tableName + " ALTER COLUMN ID RESTART WITH 1").executeUpdate();
37 | }
38 | entityManager.createNativeQuery("SET REFERENTIAL_INTEGRITY TRUE").executeUpdate();
39 | }
40 | }
--------------------------------------------------------------------------------