admin/mybatis-flex
1
0
Fork 0
mirror of https://gitee.com/mybatis-flex/mybatis-flex.git synced 2025-12-07 00:58:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add support nested types for MappedStatementTypes.java

Browse Source
This commit is contained in:
Michael Yang 2024-08-23 16:04:18 +08:00
parent 417b04838b
commit 012381510d
2 changed files with 45 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
mybatis-flex-core/src/main/java/com/mybatisflex/core/mybatis/MappedStatementTypes.java
View File

@ -15,23 +15,32 @@
*/
package com.mybatisflex.core.mybatis;
import java.util.Stack;
public class MappedStatementTypes {
private MappedStatementTypes() {
}
private static final ThreadLocal<Class<?>> currentTypeTL = new ThreadLocal<>();
private static final ThreadLocal<Stack<Class<?>>> currentTypeTL = ThreadLocal.withInitial(Stack::new);
public static void setCurrentType(Class<?> type) {
currentTypeTL.set(type);
currentTypeTL.get().push(type);
}
public static Class<?> getCurrentType() {
return currentTypeTL.get();
Stack<Class<?>> stack = currentTypeTL.get();
return stack.isEmpty() ? null : stack.lastElement();
}
public static void clear() {
currentTypeTL.remove();
Stack<Class<?>> stack = currentTypeTL.get();
if (!stack.isEmpty()) {
stack.pop();
}
if (stack.isEmpty()) {
currentTypeTL.remove();
}
}
}

32
mybatis-flex-core/src/test/java/com/mybatisflex/core/util/MappedStatementTypesTest.java Normal file
View File

@ -0,0 +1,32 @@
package com.mybatisflex.core.util;
import com.mybatisflex.core.mybatis.MappedStatementTypes;
import org.junit.Assert;
import org.junit.Test;
public class MappedStatementTypesTest {
@Test
public void test() {
MappedStatementTypes.clear();
MappedStatementTypes.setCurrentType(String.class);
MappedStatementTypes.setCurrentType(MappedStatementTypesTest.class);
MappedStatementTypes.setCurrentType(StringUtilTest.class);
Assert.assertEquals(StringUtilTest.class, MappedStatementTypes.getCurrentType());
System.out.println(MappedStatementTypes.getCurrentType());
MappedStatementTypes.clear();
Assert.assertEquals(MappedStatementTypesTest.class, MappedStatementTypes.getCurrentType());
System.out.println(MappedStatementTypes.getCurrentType());
MappedStatementTypes.clear();
Assert.assertEquals(String.class, MappedStatementTypes.getCurrentType());
System.out.println(MappedStatementTypes.getCurrentType());
MappedStatementTypes.clear();
Assert.assertNull(MappedStatementTypes.getCurrentType());
System.out.println(MappedStatementTypes.getCurrentType());
}
}