View Javadoc
1   /*
2    *    Copyright 2016-2025 the original author or authors.
3    *
4    *    Licensed under the Apache License, Version 2.0 (the "License");
5    *    you may not use this file except in compliance with the License.
6    *    You may obtain a copy of the License at
7    *
8    *       https://www.apache.org/licenses/LICENSE-2.0
9    *
10   *    Unless required by applicable law or agreed to in writing, software
11   *    distributed under the License is distributed on an "AS IS" BASIS,
12   *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13   *    See the License for the specific language governing permissions and
14   *    limitations under the License.
15   */
16  package org.mybatis.dynamic.sql.select.render;
17  
18  import java.util.Objects;
19  
20  import org.mybatis.dynamic.sql.render.RenderedParameterInfo;
21  import org.mybatis.dynamic.sql.render.RenderingContext;
22  import org.mybatis.dynamic.sql.select.PagingModel;
23  import org.mybatis.dynamic.sql.util.FragmentAndParameters;
24  
25  public class LimitAndOffsetPagingModelRenderer {
26      private final RenderingContext renderingContext;
27      private final Long limit;
28      private final PagingModel pagingModel;
29  
30      public LimitAndOffsetPagingModelRenderer(RenderingContext renderingContext,
31                                               Long limit, PagingModel pagingModel) {
32          this.renderingContext = renderingContext;
33          this.limit = Objects.requireNonNull(limit);
34          this.pagingModel = pagingModel;
35      }
36  
37      public FragmentAndParameters render() {
38          return pagingModel.offset().map(this::renderLimitAndOffset)
39                  .orElseGet(this::renderLimitOnly);
40      }
41  
42      private FragmentAndParameters renderLimitOnly() {
43          RenderedParameterInfo limitParameterInfo = renderingContext.calculateLimitParameterInfo();
44          return FragmentAndParameters.withFragment("limit " + limitParameterInfo.renderedPlaceHolder()) //$NON-NLS-1$
45                  .withParameter(limitParameterInfo.parameterMapKey(), limit)
46                  .build();
47      }
48  
49      private FragmentAndParameters renderLimitAndOffset(Long offset) {
50          RenderedParameterInfo limitParameterInfo = renderingContext.calculateLimitParameterInfo();
51          RenderedParameterInfo offsetParameterInfo = renderingContext.calculateOffsetParameterInfo();
52          return FragmentAndParameters.withFragment("limit " + limitParameterInfo.renderedPlaceHolder() //$NON-NLS-1$
53                      + " offset " + offsetParameterInfo.renderedPlaceHolder()) //$NON-NLS-1$
54                  .withParameter(limitParameterInfo.parameterMapKey(), limit)
55                  .withParameter(offsetParameterInfo.parameterMapKey(), offset)
56                  .build();
57      }
58  }