use of indi.mybatis.flying.statics.KeyGeneratorType in project mybatis.flying by limeng32.
the class CookOriginalSql method fetchFlyingFeature.
public static FlyingModel fetchFlyingFeature(String originalSql) {
if (flyingModelCache.get(originalSql) != null) {
return flyingModelCache.get(originalSql);
}
FlyingModel ret = new FlyingModel();
String extension = null;
if (null != originalSql && originalSql.startsWith(FLYING) && originalSql.indexOf(':') > -1) {
String dataSourceIdAndConnectionCatalog = null;
String s1 = null;
if ((originalSql.startsWith(FLYING_LEFTBRACKET) || originalSql.startsWith(FLYING_QUESTIONMARK_LEFTBRACKET)) && originalSql.indexOf(')') > 0) {
String s0 = originalSql.substring(0, originalSql.indexOf(')') + 1);
dataSourceIdAndConnectionCatalog = s0.substring(s0.indexOf('(') + 1, s0.lastIndexOf(')'));
s1 = originalSql.substring(0, originalSql.indexOf(':', originalSql.indexOf(')')));
} else {
s1 = originalSql.substring(0, originalSql.indexOf(':'));
}
if (FLYING.equals(s1) || FLYING_QUESTIONMARK.equals(s1) || s1.startsWith(FLYING_LEFTBRACKET) || originalSql.startsWith(FLYING_QUESTIONMARK_LEFTBRACKET)) {
String s2 = null;
if (s1.startsWith(FLYING_LEFTBRACKET) || originalSql.startsWith(FLYING_QUESTIONMARK_LEFTBRACKET)) {
s2 = originalSql.substring(originalSql.indexOf(":", originalSql.indexOf(')')) + 1, originalSql.length());
} else {
s2 = originalSql.substring(originalSql.indexOf(':') + 1, originalSql.length());
}
String actionTypeStr = null;
if (s2.indexOf(':') > -1) {
actionTypeStr = s2.substring(0, s2.indexOf(':'));
} else {
actionTypeStr = s2;
}
if (actionTypeStr.endsWith(")") && actionTypeStr.indexOf('(') != -1) {
extension = actionTypeStr.substring(actionTypeStr.lastIndexOf('(') + 1, actionTypeStr.length() - 1);
actionTypeStr = actionTypeStr.substring(0, actionTypeStr.lastIndexOf('('));
}
ActionType actionType = ActionType.valueOf(actionTypeStr);
if (actionType != null) {
ret.setHasFlyingFeature(true);
if (dataSourceIdAndConnectionCatalog != null && dataSourceIdAndConnectionCatalog.indexOf(':') != -1) {
ret.setDataSourceId(dataSourceIdAndConnectionCatalog.substring(0, dataSourceIdAndConnectionCatalog.indexOf(':')));
ret.setConnectionCatalog(dataSourceIdAndConnectionCatalog.substring(dataSourceIdAndConnectionCatalog.indexOf(':') + 1, dataSourceIdAndConnectionCatalog.length()));
}
ret.setActionType(actionType);
if (s2.indexOf(':') > -1) {
String s3 = s2.substring(s2.indexOf(':') + 1, s2.length());
String ignoreTag = null;
if (s3.indexOf(':') > -1) {
ignoreTag = s3.substring(0, s3.indexOf(':'));
} else {
ignoreTag = s3;
}
ret.setIgnoreTag(ignoreTag);
}
if (ActionType.insert.equals(actionType) && extension != null) {
KeyGeneratorType keyGeneratorType = null;
if (extension.indexOf(".") == -1) {
try {
keyGeneratorType = KeyGeneratorType.valueOf(extension);
} catch (IllegalArgumentException e) {
logger.error(new StringBuffer(AutoMapperExceptionEnum.wrongKeyGeneratorType.description()).append(originalSql).append(" because of ").append(e).toString());
}
ret.setKeyGeneratorType(keyGeneratorType);
if (keyGeneratorType != null) {
KeyHandler keyHandler;
switch(keyGeneratorType) {
case uuid:
keyHandler = UuidKeyHandler.getInstance();
break;
case uuid_no_line:
keyHandler = UuidWithoutLineKeyHandler.getInstance();
break;
case millisecond:
keyHandler = MilliSecondKeyHandler.getInstance();
break;
case snowflake:
keyHandler = SnowFlakeKeyHandler.getInstance();
break;
default:
keyHandler = null;
break;
}
ret.setKeyHandler(keyHandler);
}
} else {
try {
@SuppressWarnings("unchecked") Class<? extends KeyHandler> clazz = (Class<? extends KeyHandler>) Class.forName(extension);
ret.setKeyGeneratorType(KeyGeneratorType.custom);
ret.setKeyHandler(clazz.newInstance());
} catch (ClassNotFoundException | InstantiationException | IllegalAccessException e) {
logger.error(new StringBuffer(AutoMapperExceptionEnum.wrongCustomKeyGenerator.description()).append(originalSql).append(" because of ").append(e).toString());
}
}
}
flyingModelCache.put(originalSql, ret);
return ret;
}
}
}
ret.setHasFlyingFeature(false);
flyingModelCache.put(originalSql, ret);
return ret;
}
Aggregations