diff --git a/src/main/java/com/alibaba/druid/support/opds/udf/ExportConditions.java b/src/main/java/com/alibaba/druid/support/opds/udf/ExportConditions.java new file mode 100644 index 0000000000..5c26375c5c --- /dev/null +++ b/src/main/java/com/alibaba/druid/support/opds/udf/ExportConditions.java @@ -0,0 +1,72 @@ +/* + * Copyright 1999-2101 Alibaba Group Holding Ltd. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package com.alibaba.druid.support.opds.udf; + +import java.util.ArrayList; +import java.util.List; + +import com.alibaba.druid.sql.SQLUtils; +import com.alibaba.druid.sql.ast.SQLStatement; +import com.alibaba.druid.sql.visitor.SchemaStatVisitor; +import com.alibaba.druid.stat.TableStat; +import com.alibaba.druid.stat.TableStat.Condition; +import com.alibaba.druid.support.json.JSONUtils; +import com.aliyun.odps.udf.UDF; + +public class ExportConditions extends UDF { + + public String evaluate(String sql) { + return evaluate(sql, null); + } + + public String evaluate(String sql, String dbType) { + try { + List statementList = SQLUtils.parseStatements(sql, dbType); + SchemaStatVisitor visitor = SQLUtils.createSchemaStatVisitor(dbType); + + for (SQLStatement stmt : statementList) { + stmt.accept(visitor); + } + + List> rows = new ArrayList>(); + List conditions = visitor.getConditions(); + for (int i = 0; i < conditions.size(); ++i) { + TableStat.Condition condition = conditions.get(i); + String column = condition.getColumn().toString(); + String operator = condition.getOperator(); + List values = condition.getValues(); + + List row = new ArrayList(); + row.add(column); + row.add(operator); + if (values.size() == 0) { + row.add(null); + } else if (values.size() == 1) { + row.add(values.get(0)); + } else { + row.add(values); + } + rows.add(row); + } + + return JSONUtils.toJSONString(rows); + } catch (Throwable ex) { + System.err.println("error sql : " + sql); + ex.printStackTrace(); + return null; + } + } +} diff --git a/src/test/java/com/alibaba/druid/bvt/support/odps/udf/ExportConditionsTest.java b/src/test/java/com/alibaba/druid/bvt/support/odps/udf/ExportConditionsTest.java new file mode 100644 index 0000000000..d896c90194 --- /dev/null +++ b/src/test/java/com/alibaba/druid/bvt/support/odps/udf/ExportConditionsTest.java @@ -0,0 +1,16 @@ +package com.alibaba.druid.bvt.support.odps.udf; + +import org.junit.Assert; + +import com.alibaba.druid.support.opds.udf.ExportConditions; + +import junit.framework.TestCase; + +public class ExportConditionsTest extends TestCase { + ExportConditions udf = new ExportConditions(); + + public void test_export_conditions() throws Exception { + String result = udf.evaluate("select * from t where id = 3 and name = 'chensheng'"); + Assert.assertEquals("[[\"t.id\",\"=\",3],[\"t.name\",\"=\",\"chensheng\"]]", result); + } +}