从输入的SQL参数中获取表名及字段名

最新推荐文章于 2023-12-19 18:18:42 发布

原创最新推荐文章于 2023-12-19 18:18:42 发布 · 2.5k 阅读

1 ·

CC 4.0 BY-SA版权

Java 专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一个简单的Java工具类，用于从SQL查询语句中提取表名和字段名。该工具通过字符串操作实现，能够处理基本的SELECT语句。

工具类

package com.isoft.mongo;/*
 * @Author: Create by Achun
 *@Time: 2018/6/3 23:20
 *
 */

import java.util.ArrayList;
import java.util.List;

public class SQLUtils {

    public List<String> getColumns(String querysql){
        List<String> column = new ArrayList<String>();
        String tmp = querysql.substring(querysql.indexOf("select") + 6,
                querysql.indexOf("from")).trim();
        if (tmp.indexOf("*") == -1){
            String cols[] = tmp.split(",");
            for (String c:cols){
                column.add(c);
            }
        }
        return column;
    }

    public String getTBname(String querysql){
        String tmp = querysql.substring(querysql.indexOf("from")+4).trim();
        int sx = tmp.indexOf(" ");
        if(sx == -1){
            return tmp;
        }else {
            return tmp.substring(0,sx);
        }
    }

}

测试：

package com.isoft.mongo;
/*
 * @Author: Create by Achun
 *@Time: 2018/6/3 23:32
 *
 */

import java.util.List;

public class SQLUtilsTest {
    public static void main(String[] args) {
        String querysql = "select id,name,age,exent_month from mg";
        SQLUtils utils = new SQLUtils();
        String tablename = utils.getTBname(querysql);
        List<String> columns = utils.getColumns(querysql);
        System.out.println("tablename:" + tablename);
        System.out.println("columns:" + columns);
        System.out.println("frist column:"+columns.get(0));

    }
}