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

最新推荐文章于 2024-08-21 17:08:10 发布

A_ChunUnique

最新推荐文章于 2024-08-21 17:08:10 发布

阅读量2.4k

点赞数 1

分类专栏： Java

本文链接：https://blog.csdn.net/gavin_chun/article/details/80561744

版权

Java 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

工具类

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));

    }
}