SqlUtil.java 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. package org.example.join.util;
  2. import cn.hutool.core.util.StrUtil;
  3. import com.baomidou.mybatisplus.core.metadata.TableFieldInfo;
  4. import com.baomidou.mybatisplus.core.metadata.TableInfo;
  5. import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
  6. import com.baomidou.mybatisplus.core.toolkit.StringPool;
  7. import org.example.join.domain.Table;
  8. import java.util.Map;
  9. public class SqlUtil {
  10. public static String getTableAlias(Table table, Map<String, String> tableAliasMap) {
  11. String tableAlias = table.getAlias();
  12. String tableName = table.getName();
  13. if (StrUtil.isBlank(tableAlias)) {
  14. tableAlias = tableAliasMap.get(tableName);
  15. if (StrUtil.isBlank(tableAlias)) {
  16. throw new IllegalArgumentException("未知连表:" + tableName);
  17. }
  18. }
  19. return tableAlias;
  20. }
  21. public static String getTableDelFlag(Table table) {
  22. TableInfo tableInfo = TableInfoHelper.getTableInfo(table.getName());
  23. if (!tableInfo.isWithLogicDelete()) {
  24. return null;
  25. }
  26. TableFieldInfo logicDeleteFieldInfo = tableInfo.getLogicDeleteFieldInfo();
  27. return logicDeleteFieldInfo.getColumn();
  28. }
  29. public static String withBracket(String sql) {
  30. return StringPool.LEFT_BRACKET + sql + StringPool.RIGHT_BRACKET;
  31. }
  32. }