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.SqlServerDialect;
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 JbootSqlServerDialect extends SqlServerDialect 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
033        StringBuilder sqlBuilder = SqlBuilder.forFindByColumns(alias, joins, table, loadColumns, columns, orderBy, ' ');
034
035        if (limit == null) {
036            return sqlBuilder.toString();
037        }
038
039        if (limit instanceof Number) {
040            StringBuilder ret = new StringBuilder();
041            ret.append("SELECT * FROM ( SELECT row_number() over (order by tempcolumn) temprownumber, * FROM ");
042            ret.append(" ( SELECT TOP ").append(limit).append(" tempcolumn=0,");
043            ret.append(sqlBuilder.toString().replaceFirst("(?i)select", ""));
044            ret.append(")vip)mvp ");
045            return ret.toString();
046
047
048        } else if (limit instanceof String && limit.toString().contains(",")) {
049            String[] startAndEnd = limit.toString().split(",");
050            String start = startAndEnd[0];
051            String end = startAndEnd[1];
052
053            StringBuilder ret = new StringBuilder();
054            ret.append("SELECT * FROM ( SELECT row_number() over (order by tempcolumn) temprownumber, * FROM ");
055            ret.append(" ( SELECT TOP ").append(end).append(" tempcolumn=0,");
056            ret.append(sqlBuilder.toString().replaceFirst("(?i)select", ""));
057            ret.append(")vip)mvp where temprownumber>").append(start);
058            return ret.toString();
059        } else {
060            throw new JbootException("sql limit is error!,limit must is Number of String like \"0,10\"");
061        }
062
063    }
064
065    @Override
066    public String forFindCountByColumns(String alias, List<Join> joins, String table, String loadColumns, List<Column> columns) {
067        return SqlBuilder.forFindCountByColumns(alias, joins, table, loadColumns, columns, ' ');
068    }
069
070    @Override
071    public String forDeleteByColumns(String alias, List<Join> joins, String table, List<Column> columns) {
072        return SqlBuilder.forDeleteByColumns(alias, joins, table, columns, ' ');
073    }
074
075
076    @Override
077    public String forPaginateSelect(String loadColumns) {
078        return "SELECT " + loadColumns;
079    }
080
081
082    @Override
083    public String forPaginateFrom(String alias, List<Join> joins, String table, List<Column> columns, String orderBy) {
084        return SqlBuilder.forPaginateFrom(alias, joins, table, columns, orderBy, ' ');
085    }
086
087    @Override
088    public String forPaginateTotalRow(String select, String sqlExceptSelect, Object ext) {
089        String distinctSql = SqlBuilder.forPaginateDistinctTotalRow(select, sqlExceptSelect, ext);
090        return distinctSql != null ? distinctSql : super.forPaginateTotalRow(select, sqlExceptSelect, ext);
091    }
092
093}