-
Notifications
You must be signed in to change notification settings - Fork 2
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Fix when the column of db naming is not standard snake case,the sql query parse error.
- Loading branch information
huangxiaohu
committed
Jun 18, 2022
1 parent
af7a490
commit 6b86771
Showing
26 changed files
with
607 additions
and
103 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
237 changes: 237 additions & 0 deletions
237
src/main/java/com/sondertara/joya/cache/TableClassCache.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,237 @@ | ||
| package com.sondertara.joya.cache; | ||
|
|
||
| import com.google.common.collect.Maps; | ||
| import com.sondertara.common.exception.TaraException; | ||
| import com.sondertara.common.util.StringUtils; | ||
| import com.sondertara.joya.core.model.TableEntity; | ||
| import com.sondertara.joya.utils.cache.GuavaAbstractLoadingCache; | ||
| import com.sondertara.joya.utils.cache.ILocalCache; | ||
|
|
||
| import javax.persistence.Column; | ||
| import javax.persistence.Entity; | ||
| import javax.persistence.Id; | ||
| import javax.persistence.Table; | ||
| import javax.persistence.Transient; | ||
| import java.lang.reflect.Field; | ||
| import java.lang.reflect.Modifier; | ||
| import java.lang.reflect.ParameterizedType; | ||
| import java.lang.reflect.Type; | ||
| import java.util.HashSet; | ||
| import java.util.LinkedHashMap; | ||
| import java.util.Map; | ||
| import java.util.Optional; | ||
| import java.util.Set; | ||
| import java.util.concurrent.ExecutionException; | ||
|
|
||
| /** | ||
| * @author skydu | ||
| */ | ||
| public class TableClassCache extends GuavaAbstractLoadingCache<Class<?>, Map<String, Field>> implements ILocalCache<Class<?>, Map<String, Field>> { | ||
|
|
||
| private static volatile TableClassCache cache = null; | ||
|
|
||
| private TableClassCache() { | ||
| setMaximumSize(1000); | ||
| setExpireAfterWriteDuration(60 * 60); | ||
| } | ||
|
|
||
|
|
||
| public synchronized static TableClassCache getInstance() { | ||
| if (null == cache) { | ||
| synchronized (TableClassCache.class) { | ||
| if (null == cache) { | ||
| cache = new TableClassCache(); | ||
| } | ||
| } | ||
| } | ||
| return cache; | ||
| } | ||
|
|
||
| /** | ||
| * @return get current classloader | ||
| */ | ||
| public static ClassLoader getDefaultClassLoader() { | ||
| ClassLoader cl = null; | ||
| try { | ||
| cl = Thread.currentThread().getContextClassLoader(); | ||
| } catch (Throwable ex) { | ||
| // Cannot access thread context ClassLoader - falling back... | ||
| } | ||
| if (cl == null) { | ||
| // No thread context class loader -> use class loader of this class. | ||
| cl = TableClassCache.class.getClassLoader(); | ||
| if (cl == null) { | ||
| // getClassLoader() returning null indicates the bootstrap ClassLoader | ||
| try { | ||
| cl = ClassLoader.getSystemClassLoader(); | ||
| } catch (Throwable ex) { | ||
| // Cannot access system ClassLoader - oh well, maybe the caller can live with | ||
| // null... | ||
| } | ||
| } | ||
| } | ||
| return cl; | ||
| } | ||
|
|
||
| /** | ||
| * 获取类的所有字段(包括父类的) | ||
| * | ||
| * @param clazz class | ||
| * @return field | ||
| */ | ||
| public static Map<String, Field> getAllFields(Class<?> clazz) { | ||
| Map<String, Field> fields = Maps.newLinkedHashMap(); | ||
| Set<String> filedNames = new HashSet<>(); | ||
| for (Class<?> c = clazz; c != Object.class; c = c.getSuperclass()) { | ||
| try { | ||
| Field[] list = c.getDeclaredFields(); | ||
| for (Field field : list) { | ||
| String name = field.getName(); | ||
| if (filedNames.contains(name)) { | ||
| continue; | ||
| } | ||
| filedNames.add(field.getName()); | ||
| fields.put(field.getName(), field); | ||
| } | ||
| } catch (Exception e) { | ||
| throw new TaraException(e); | ||
| } | ||
| } | ||
| return fields; | ||
| } | ||
|
|
||
|
|
||
| /** | ||
| * 获取一个实体类对应数据库字段 | ||
| * | ||
| * @param bean java pojo | ||
| * @param <T> the class type of bean | ||
| * @return the table data | ||
| */ | ||
| public <T> TableEntity getTable(T bean, boolean readData) { | ||
| Map<String, Object> filedNames = new LinkedHashMap<>(); | ||
| Map<String, String> relation = Maps.newHashMap(); | ||
| Class<?> clazz = bean.getClass(); | ||
| Table table = clazz.getAnnotation(Table.class); | ||
| String tableName = null; | ||
| if (null != table) { | ||
| tableName = table.name(); | ||
| } else { | ||
| Entity entity = clazz.getAnnotation(Entity.class); | ||
| if (null != entity) { | ||
| tableName = entity.name(); | ||
| } | ||
| } | ||
| if (null == tableName) { | ||
| throw new TaraException("No [@Table] or [@Entity] annotation found for class->" + clazz); | ||
| } | ||
| TableEntity tableDTO = new TableEntity(); | ||
| tableDTO.setTableName(tableName); | ||
| Optional<Map<String, Field>> optional = get(clazz); | ||
| return optional.map(fields -> { | ||
| for (Field field : fields.values()) { | ||
| if (Modifier.isStatic(field.getModifiers())) { | ||
| continue; | ||
| } | ||
| if (field.isAnnotationPresent(Transient.class) || field.isAnnotationPresent(org.springframework.data.annotation.Transient.class)) { | ||
| continue; | ||
| } | ||
| String name = field.getName(); | ||
| name = StringUtils.toUnderlineCase(name); | ||
| Column column = field.getAnnotation(Column.class); | ||
| if (null != column) { | ||
| name = column.name(); | ||
| } | ||
| Id id = field.getAnnotation(Id.class); | ||
| if (null != id) { | ||
| tableDTO.setPrimaryKey(name); | ||
| tableDTO.setPrimaryKeyType(field.getType()); | ||
| } | ||
| if (filedNames.containsKey(name)) { | ||
| continue; | ||
| } | ||
| if (readData) { | ||
| try { | ||
| Object o; | ||
| if (field.isAccessible()) { | ||
| o = field.get(bean); | ||
| } else { | ||
| field.setAccessible(true); | ||
| o = field.get(bean); | ||
| field.setAccessible(false); | ||
| } | ||
| filedNames.put(name, o); | ||
| } catch (Exception e) { | ||
| e.printStackTrace(); | ||
| } | ||
| } | ||
| relation.put(name, field.getName()); | ||
|
|
||
| } | ||
| tableDTO.setData(filedNames); | ||
| tableDTO.setRelation(relation); | ||
| return tableDTO; | ||
| }).orElseThrow(() -> new TaraException("Get Class definition error")); | ||
|
|
||
| } | ||
|
|
||
| /** | ||
| * @param clazz | ||
| * @return the super class | ||
| */ | ||
| public static Class<?> getSuperClassGenricType(Class<?> clazz) { | ||
| Type genType = clazz.getGenericSuperclass(); | ||
| if (!(genType instanceof ParameterizedType)) { | ||
| if (genType instanceof Class) { | ||
| return getSuperClassGenricType((Class<?>) genType); | ||
| } | ||
| return Object.class; | ||
| } | ||
| Type[] params = ((ParameterizedType) genType).getActualTypeArguments(); | ||
| if (params.length == 0) { | ||
| return Object.class; | ||
| } | ||
| return (Class<?>) params[0]; | ||
| } | ||
|
|
||
|
|
||
| /** | ||
| * @param clazz the class | ||
| * @param fieldName field name | ||
| * @return the current Field | ||
| */ | ||
| public static Field getField(Class<?> clazz, String fieldName) { | ||
| Map<String, Field> map = getAllFields(clazz); | ||
| return map.get(fieldName); | ||
| } | ||
|
|
||
| public static Class<?> classForName(String name) throws ClassNotFoundException { | ||
| try { | ||
| ClassLoader classLoader = Thread.currentThread().getContextClassLoader(); | ||
| if (classLoader != null) { | ||
| return classLoader.loadClass(name); | ||
| } | ||
| } catch (Throwable ignore) { | ||
| } | ||
| return Class.forName(name); | ||
| } | ||
|
|
||
| @Override | ||
| protected Map<String, Field> fetchData(Class<?> key) { | ||
| return getAllFields(key); | ||
| } | ||
|
|
||
| @Override | ||
| public Optional<Map<String, Field>> get(Class<?> key) { | ||
| try { | ||
| return Optional.ofNullable(getValue(key)); | ||
| } catch (ExecutionException e) { | ||
| throw new TaraException(e); | ||
| } | ||
| } | ||
|
|
||
| @Override | ||
| public void remove(Class<?> key) { | ||
|
|
||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.