admin/mybatis-flex
1
0
Fork 0
mirror of https://gitee.com/mybatis-flex/mybatis-flex.git synced 2025-12-07 09:08:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

!489 支持trino数据库

Browse Source 
Merge pull request !489 from 简单风/support-trino
This commit is contained in:
Michael Yang 2024-08-29 06:32:16 +00:00 committed by Gitee
commit a0bc82c41c
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
3 changed files with 110 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mybatis-flex-core/src/main/java/com/mybatisflex/core/dialect/DbType.java
View File

@ -229,6 +229,11 @@ public enum DbType {
*/
DORIS("doris", "doris 数据库"),
/**
* Trino
*/
TRINO("trino", "trino 数据库"),
/**
* UNKNOWN DB
*/

2
mybatis-flex-core/src/main/java/com/mybatisflex/core/dialect/DialectFactory.java
View File

@ -150,6 +150,8 @@ public class DialectFactory {
return new CommonsDialectImpl(KeywordWrap.DOUBLE_QUOTATION, LimitOffsetProcessor.SINODB);
case SYBASE:
return new CommonsDialectImpl(KeywordWrap.DOUBLE_QUOTATION, LimitOffsetProcessor.SYBASE);
case TRINO:
return new CommonsDialectImpl(KeywordWrap.NONE, LimitOffsetProcessor.SQLSERVER);
default:
return new CommonsDialectImpl();
}

103
mybatis-flex-core/src/test/java/com/mybatisflex/coretest/TrinoDialectTester.java Normal file
View File

@ -0,0 +1,103 @@
package com.mybatisflex.coretest;
import com.mybatisflex.core.dialect.IDialect;
import com.mybatisflex.core.dialect.KeywordWrap;
import com.mybatisflex.core.dialect.LimitOffsetProcessor;
import com.mybatisflex.core.dialect.impl.CommonsDialectImpl;
import com.mybatisflex.core.query.QueryWrapper;
import org.junit.Assert;
import org.junit.Test;
import static com.mybatisflex.core.query.QueryMethods.select;
import static com.mybatisflex.coretest.table.AccountTableDef.ACCOUNT;
/**
* trino使用示例
*/
public class TrinoDialectTester {
private static final IDialect DIALECT = new CommonsDialectImpl(KeywordWrap.NONE, LimitOffsetProcessor.SQLSERVER);
/**
* 分页
*/
@Test
public void testPageSql() {
QueryWrapper query = new QueryWrapper()
.select()
.from(ACCOUNT)
.where(ACCOUNT.ID.in("100", "200"))
.and(ACCOUNT.SEX.eq(1))
.orderBy(ACCOUNT.ID.desc())
.limit(10, 10);
String sql = DIALECT.forSelectByQuery(query);
System.out.println(sql);
Assert.assertEquals("SELECT * FROM tb_account WHERE id IN (?, ?) AND sex = ? ORDER BY id DESC OFFSET 10 ROWS FETCH NEXT 10 ROWS ONLY", sql);
}
/**
* 聚合函数的过滤器
*/
@Test
public void testFilteringDuringAggregation() {
QueryWrapper query = new QueryWrapper()
.select(ACCOUNT.SEX)
.select("count(*) filter(where age > 10)")
.from(ACCOUNT)
.groupBy(ACCOUNT.SEX);
String sql = DIALECT.forSelectByQuery(query);
System.out.println(sql);
Assert.assertEquals("SELECT sex, count(*) filter(where age > 10) FROM tb_account GROUP BY sex", sql);
}
/**
* 利用array函数实现mysql group_concat功能
*/
@Test
public void testArrayFunction() {
QueryWrapper query = new QueryWrapper()
.select(ACCOUNT.SEX)
.select("array_join(array_agg(user_name), ',')")
.from(ACCOUNT)
.groupBy(ACCOUNT.SEX);
String sql = DIALECT.forSelectByQuery(query);
System.out.println(sql);
Assert.assertEquals("SELECT sex, array_join(array_agg(user_name), ',') FROM tb_account GROUP BY sex", sql);
}
/**
* 窗口函数
*/
@Test
public void testOverFunction() {
QueryWrapper query = new QueryWrapper()
.select()
.from(QueryWrapper.create().select(ACCOUNT.ID, ACCOUNT.USER_NAME)
.select("ROW_NUMBER() OVER(PARTITION BY sex ORDER BY birthday) AS row_num")
.from(ACCOUNT))
.where("row_num = 1");
String sql = DIALECT.forSelectByQuery(query);
System.out.println(sql);
Assert.assertEquals("SELECT * FROM (SELECT id, user_name, ROW_NUMBER() OVER(PARTITION BY sex ORDER BY birthday) AS row_num FROM tb_account) WHERE row_num = 1 ", sql);
}
/**
* with 子查询
*/
@Test
public void testWith() {
QueryWrapper query = new QueryWrapper()
.with("t").asSelect(
select().from(ACCOUNT).where(ACCOUNT.ID.ge(100))
)
.select()
.from("t")
.where("t.sex = 0");
String sql = DIALECT.forSelectByQuery(query);
System.out.println(sql);
Assert.assertEquals("WITH t AS (SELECT * FROM tb_account WHERE id >= ?) SELECT * FROM t WHERE t.sex = 0 ", sql);
}
}