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

add JdbcTypeMapping.registerDateTypes() method

Browse Source
This commit is contained in:
开源海哥 2023-03-29 12:21:27 +08:00
parent c2ee7cf042
commit 7568c93a35
1 changed files with 34 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/dialect/JdbcTypeMapping.java
View File

@ -1,3 +1,18 @@
/**
* Copyright (c) 2022-2023, Mybatis-Flex (fuhai999@gmail.com).
* <p>
* 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
* <p>
* http://www.apache.org/licenses/LICENSE-2.0
* <p>
* 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.
*/
package com.mybatisflex.codegen.dialect; package com.mybatisflex.codegen.dialect;
import com.mybatisflex.core.util.StringUtil; import com.mybatisflex.core.util.StringUtil;
@ -10,17 +25,32 @@ public class JdbcTypeMapping {
private static Map<String, String> mapping = new HashMap<>(); private static Map<String, String> mapping = new HashMap<>();
static { static {
mapping.put("[B", "byte[]"); registerMapping("[B", "byte[]");
} }
public void registerMapping(Class<?> from, Class<?> to) { public static void registerMapping(Class<?> from, Class<?> to) {
mapping.put(from.getName(), to.getName()); registerMapping(from.getName(), to.getName());
} }
public void registerMapping(String from, String to) { public static void registerMapping(String from, String to) {
mapping.put(from, to); mapping.put(from, to);
} }
public static Map<String, String> getMapping() {
return mapping;
}
/**
* 当只使用 date 类型来映射数据库的所有 "时间" 类型时，调用此方法
*/
public static void registerDateTypes() {
registerMapping("java.sql.Time", "java.util.Date");
registerMapping("java.sql.Timestamp", "java.util.Date");
registerMapping("java.time.LocalDateTime", "java.util.Date");
registerMapping("java.time.LocalDate", "java.util.Date");
}
static String getType(String jdbcType) { static String getType(String jdbcType) {
String registered = mapping.get(jdbcType); String registered = mapping.get(jdbcType);
return StringUtil.isNotBlank(registered) ? registered : jdbcType; return StringUtil.isNotBlank(registered) ? registered : jdbcType;