Search in sources :

Example 1 with GroovyTemplateGenerator

use of com.github.mgramin.sqlboot.template.generator.impl.GroovyTemplateGenerator in project sql-boot by sql-boot.

the class FsResourceTypes method walk.

/**
 * @param path
 * @return
 */
private List<ResourceType> walk(final String path) {
    File[] files = new File(path).listFiles();
    if (files == null)
        return null;
    List<ResourceType> list = new ArrayList<>();
    for (File f : files) {
        if (f.isDirectory()) {
            File sqlFile = new File(f, "README.md");
            list.addAll(walk(f.getAbsolutePath()));
            final ResourceType jdbcResourceType;
            switch(f.getName()) {
                case "schema":
                    jdbcResourceType = new SchemaJdbcResourceType(dataSource);
                    break;
                case "table":
                    jdbcResourceType = new TableJdbcResourceType(dataSource);
                    break;
                case "child":
                    jdbcResourceType = new ChildTableJdbcResourceType(dataSource);
                    break;
                case "parent":
                    jdbcResourceType = new ParentTableJdbcResourceType(dataSource);
                    break;
                case "pk":
                    jdbcResourceType = new PkJdbcResourceType(dataSource);
                    break;
                case "index":
                    jdbcResourceType = new IndexJdbcResourceType(dataSource);
                    break;
                case "fk":
                    jdbcResourceType = new FkJdbcResourceType(dataSource);
                    break;
                case "view":
                    jdbcResourceType = new ViewJdbcResourceType(dataSource);
                    break;
                case "column":
                    jdbcResourceType = new ColumnJdbcResourceType(dataSource);
                    break;
                case "function":
                    jdbcResourceType = new FunctionJdbcResourceType(dataSource);
                    break;
                case "procedure":
                    jdbcResourceType = new ProcedureJdbcResourceType(dataSource);
                    break;
                default:
                    jdbcResourceType = null;
            }
            String sql = null;
            try {
                MarkdownFile markdownFile = new MarkdownFile(readFileToString(sqlFile, UTF_8));
                Map<String, String> parse = markdownFile.parse();
                Iterator<Map.Entry<String, String>> iterator = parse.entrySet().iterator();
                if (iterator.hasNext()) {
                    sql = iterator.next().getValue();
                }
            } catch (IOException e) {
            // TODO catch and process this exception
            }
            final ResourceType baseResourceType;
            if (sqlFile.exists() && sql != null) {
                baseResourceType = new SqlResourceType(new JdbcSqlQuery(dataSource, new GroovyTemplateGenerator(sql)), singletonList(f.getName()));
            } else if (jdbcResourceType != null) {
                baseResourceType = jdbcResourceType;
            } else {
                baseResourceType = null;
            }
            final ResourceType resourceType = new SelectWrapper(// new SqlBodyWrapper(
            new TemplateBodyWrapper(new PageWrapper(new LimitWrapper(// new WhereWrapper(
            baseResourceType)), new GroovyTemplateGenerator("EMPTY BODY ...")));
            if (baseResourceType != null) {
                list.add(resourceType);
            }
        }
    }
    return list;
}
Also used : SelectWrapper(com.github.mgramin.sqlboot.model.resource_type.wrappers.header.SelectWrapper) LimitWrapper(com.github.mgramin.sqlboot.model.resource_type.wrappers.list.LimitWrapper) SqlResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.sql.SqlResourceType) MarkdownFile(com.github.mgramin.sqlboot.model.resource_type.impl.composite.md.MarkdownFile) ArrayList(java.util.ArrayList) JdbcSqlQuery(com.github.mgramin.sqlboot.sql.impl.JdbcSqlQuery) FileUtils.readFileToString(org.apache.commons.io.FileUtils.readFileToString) ParentTableJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.relation.ParentTableJdbcResourceType) PkJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.pk.PkJdbcResourceType) ProcedureJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.procedure.ProcedureJdbcResourceType) PageWrapper(com.github.mgramin.sqlboot.model.resource_type.wrappers.list.PageWrapper) TemplateBodyWrapper(com.github.mgramin.sqlboot.model.resource_type.wrappers.body.TemplateBodyWrapper) FunctionJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.function.FunctionJdbcResourceType) PkJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.pk.PkJdbcResourceType) SqlResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.sql.SqlResourceType) FunctionJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.function.FunctionJdbcResourceType) TableJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.TableJdbcResourceType) ParentTableJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.relation.ParentTableJdbcResourceType) ResourceType(com.github.mgramin.sqlboot.model.resource_type.ResourceType) SchemaJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.SchemaJdbcResourceType) FkJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.fk.FkJdbcResourceType) ViewJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.view.ViewJdbcResourceType) IndexJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.index.IndexJdbcResourceType) ColumnJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.column.ColumnJdbcResourceType) ChildTableJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.relation.ChildTableJdbcResourceType) ProcedureJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.procedure.ProcedureJdbcResourceType) IOException(java.io.IOException) IndexJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.index.IndexJdbcResourceType) SchemaJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.SchemaJdbcResourceType) GroovyTemplateGenerator(com.github.mgramin.sqlboot.template.generator.impl.GroovyTemplateGenerator) FkJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.fk.FkJdbcResourceType) TableJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.TableJdbcResourceType) ParentTableJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.relation.ParentTableJdbcResourceType) ChildTableJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.relation.ChildTableJdbcResourceType) ViewJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.view.ViewJdbcResourceType) ChildTableJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.relation.ChildTableJdbcResourceType) MarkdownFile(com.github.mgramin.sqlboot.model.resource_type.impl.composite.md.MarkdownFile) File(java.io.File) ColumnJdbcResourceType(com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.column.ColumnJdbcResourceType)

Example 2 with GroovyTemplateGenerator

use of com.github.mgramin.sqlboot.template.generator.impl.GroovyTemplateGenerator in project sql-boot by sql-boot.

the class JdbcSqlQueryTest method medataData.

@Test
public void medataData() throws Exception {
    SqlQuery sqlQuery = new JdbcSqlQuery(dataSource, new GroovyTemplateGenerator("select name /* name */, email /* email */ from main_schema.users"));
    final Map<String, String> metadata = sqlQuery.metaData();
    System.out.println(metadata);
    assertEquals("email", metadata.get("email"));
    assertEquals("name", metadata.get("name"));
}
Also used : SqlQuery(com.github.mgramin.sqlboot.sql.SqlQuery) GroovyTemplateGenerator(com.github.mgramin.sqlboot.template.generator.impl.GroovyTemplateGenerator) Test(org.junit.Test)

Example 3 with GroovyTemplateGenerator

use of com.github.mgramin.sqlboot.template.generator.impl.GroovyTemplateGenerator in project sql-boot by sql-boot.

the class SqlResourceTypeTest method read2.

@Test
public void read2() {
    final String sql = "select * from (select table_schema as \"@table_schema\", table_name as \"@table_name\", column_name as \"@column_name\"" + "from information_schema.columns)";
    ResourceType type = new WhereWrapper(new SqlResourceType(new JdbcSqlQuery(dataSource, new GroovyTemplateGenerator(sql)), asList("column")));
    assertEquals(8, type.read(new DbUri("column/main_schema.users")).count());
}
Also used : DbUri(com.github.mgramin.sqlboot.model.uri.impl.DbUri) JdbcSqlQuery(com.github.mgramin.sqlboot.sql.impl.JdbcSqlQuery) ResourceType(com.github.mgramin.sqlboot.model.resource_type.ResourceType) GroovyTemplateGenerator(com.github.mgramin.sqlboot.template.generator.impl.GroovyTemplateGenerator) WhereWrapper(com.github.mgramin.sqlboot.model.resource_type.wrappers.list.WhereWrapper) Test(org.junit.Test)

Example 4 with GroovyTemplateGenerator

use of com.github.mgramin.sqlboot.template.generator.impl.GroovyTemplateGenerator in project sql-boot by sql-boot.

the class SqlResourceTypeTest method read.

@Test
public void read() throws Exception {
    final String sql = "select * from (select table_schema as \"@table_schema\", table_name as \"@table_name\" " + "from information_schema.tables)";
    final ResourceType type = new WhereWrapper(new SqlResourceType(new JdbcSqlQuery(dataSource, new GroovyTemplateGenerator(sql)), asList("table")));
    assertEquals(3, type.read(new DbUri("table/m.column")).count());
}
Also used : DbUri(com.github.mgramin.sqlboot.model.uri.impl.DbUri) JdbcSqlQuery(com.github.mgramin.sqlboot.sql.impl.JdbcSqlQuery) ResourceType(com.github.mgramin.sqlboot.model.resource_type.ResourceType) GroovyTemplateGenerator(com.github.mgramin.sqlboot.template.generator.impl.GroovyTemplateGenerator) WhereWrapper(com.github.mgramin.sqlboot.model.resource_type.wrappers.list.WhereWrapper) Test(org.junit.Test)

Example 5 with GroovyTemplateGenerator

use of com.github.mgramin.sqlboot.template.generator.impl.GroovyTemplateGenerator in project sql-boot by sql-boot.

the class SqlResourceTypeTest method path.

@Test
public void path() {
    final String sql = "select _schema, _table, _column from (select table_schema as \"_schema\", table_name as \"_table\", column_name as \"_column\"" + "from information_schema.columns)";
    final ResourceType type = new SqlResourceType(new JdbcSqlQuery(dataSource, new GroovyTemplateGenerator(sql)), asList("column"));
    final List<String> path = type.path();
    assertEquals("schema", path.get(0));
    assertEquals("table", path.get(1));
    assertEquals("column", path.get(2));
}
Also used : JdbcSqlQuery(com.github.mgramin.sqlboot.sql.impl.JdbcSqlQuery) ResourceType(com.github.mgramin.sqlboot.model.resource_type.ResourceType) GroovyTemplateGenerator(com.github.mgramin.sqlboot.template.generator.impl.GroovyTemplateGenerator) Test(org.junit.Test)

Aggregations

GroovyTemplateGenerator (com.github.mgramin.sqlboot.template.generator.impl.GroovyTemplateGenerator)5 ResourceType (com.github.mgramin.sqlboot.model.resource_type.ResourceType)4 JdbcSqlQuery (com.github.mgramin.sqlboot.sql.impl.JdbcSqlQuery)4 Test (org.junit.Test)4 WhereWrapper (com.github.mgramin.sqlboot.model.resource_type.wrappers.list.WhereWrapper)2 DbUri (com.github.mgramin.sqlboot.model.uri.impl.DbUri)2 MarkdownFile (com.github.mgramin.sqlboot.model.resource_type.impl.composite.md.MarkdownFile)1 SchemaJdbcResourceType (com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.SchemaJdbcResourceType)1 FunctionJdbcResourceType (com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.function.FunctionJdbcResourceType)1 ProcedureJdbcResourceType (com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.procedure.ProcedureJdbcResourceType)1 TableJdbcResourceType (com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.TableJdbcResourceType)1 ColumnJdbcResourceType (com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.column.ColumnJdbcResourceType)1 FkJdbcResourceType (com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.fk.FkJdbcResourceType)1 IndexJdbcResourceType (com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.index.IndexJdbcResourceType)1 PkJdbcResourceType (com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.pk.PkJdbcResourceType)1 ChildTableJdbcResourceType (com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.relation.ChildTableJdbcResourceType)1 ParentTableJdbcResourceType (com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.table.relation.ParentTableJdbcResourceType)1 ViewJdbcResourceType (com.github.mgramin.sqlboot.model.resource_type.impl.jdbc.schema.view.ViewJdbcResourceType)1 SqlResourceType (com.github.mgramin.sqlboot.model.resource_type.impl.sql.SqlResourceType)1 TemplateBodyWrapper (com.github.mgramin.sqlboot.model.resource_type.wrappers.body.TemplateBodyWrapper)1