001/**
002 * Copyright (c) 2015-2022, Michael Yang 杨福海 (fuhai999@gmail.com).
003 * <p>
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 * <p>
008 * http://www.apache.org/licenses/LICENSE-2.0
009 * <p>
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016package io.jboot.db.dialect;
017
018import com.jfinal.plugin.activerecord.dialect.PostgreSqlDialect;
019import io.jboot.db.model.Column;
020import io.jboot.db.model.SqlBuilder;
021import io.jboot.db.model.Join;
022import io.jboot.exception.JbootException;
023
024import java.util.List;
025
026
027public class JbootPostgreSqlDialect extends PostgreSqlDialect implements JbootDialect {
028
029
030    @Override
031    public String forFindByColumns(String alias, List<Join> joins, String table, String loadColumns, List<Column> columns, String orderBy, Object limit) {
032        StringBuilder sqlBuilder = SqlBuilder.forFindByColumns(alias, joins, table, loadColumns, columns, orderBy, '"');
033
034        if (limit == null) {
035            return sqlBuilder.toString();
036        }
037
038        if (limit instanceof Number) {
039            sqlBuilder.append(" limit ").append(limit).append(" offset ").append(0);
040            return sqlBuilder.toString();
041        } else if (limit instanceof String && limit.toString().contains(",")) {
042            String[] startAndEnd = limit.toString().split(",");
043            String start = startAndEnd[0];
044            String end = startAndEnd[1];
045
046            sqlBuilder.append(" limit ").append(end).append(" offset ").append(start);
047            return sqlBuilder.toString();
048        } else {
049            throw new JbootException("sql limit is error!,limit must is Number of String like \"0,10\"");
050        }
051    }
052
053    @Override
054    public String forFindCountByColumns(String alias, List<Join> joins, String table, String loadColumns, List<Column> columns) {
055        return SqlBuilder.forFindCountByColumns(alias, joins, table, loadColumns, columns, '"');
056    }
057
058    @Override
059    public String forDeleteByColumns(String alias, List<Join> joins, String table, List<Column> columns) {
060        return SqlBuilder.forDeleteByColumns(alias, joins, table, columns, '"');
061    }
062
063
064    @Override
065    public String forPaginateSelect(String loadColumns) {
066        return "SELECT " + loadColumns;
067    }
068
069
070    @Override
071    public String forPaginateFrom(String alias, List<Join> joins, String table, List<Column> columns, String orderBy) {
072        return SqlBuilder.forPaginateFrom(alias, joins, table, columns, orderBy, '"');
073    }
074
075    @Override
076    public String forPaginateTotalRow(String select, String sqlExceptSelect, Object ext) {
077        String distinctSql = SqlBuilder.forPaginateDistinctTotalRow(select, sqlExceptSelect, ext);
078        return distinctSql != null ? distinctSql : super.forPaginateTotalRow(select, sqlExceptSelect, ext);
079    }
080
081}