Coverage Summary for Class: PagedClause (com.kotlinorm.orm.pagination)

 /**
  * Copyright 2022-2025 kronos-orm
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
  *     http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
 
 package com.kotlinorm.orm.pagination
 
 import com.kotlinorm.beans.dsl.Field
 import com.kotlinorm.beans.dsl.KSelectable
 import com.kotlinorm.beans.task.KronosQueryTask
 import com.kotlinorm.enums.KColumnType.CUSTOM_CRITERIA_SQL
 import com.kotlinorm.enums.QueryType.QueryList
 import com.kotlinorm.interfaces.KPojo
 import com.kotlinorm.interfaces.KronosDataSourceWrapper
 import com.kotlinorm.utils.DataSourceUtil.orDefault
 import com.kotlinorm.utils.logAndReturn
 
 class PagedClause<K : KPojo, T : KSelectable<K>>(
     private val selectClause: T
 ) {
     fun query(wrapper: KronosDataSourceWrapper? = null): Pair<Int, List<Map<String, Any>>> {
         val tasks = this.build(wrapper)
         val total = tasks.first.queryOne<Int>()
         val records = tasks.second.query()
         return total to records
     }
 
     inline fun <reified E : KPojo> queryList(wrapper: KronosDataSourceWrapper? = null): Pair<Int, List<E>> {
         val tasks = this.build(wrapper)
         val total = tasks.first.queryOne<Int>()
         val records = tasks.second.queryList<E>()
         return total to records
     }
 
     @JvmName("queryForList")
     @Suppress("UNCHECKED_CAST")
     fun queryList(wrapper: KronosDataSourceWrapper? = null): Pair<Int, List<K>> {
         val tasks = this.build()
         val total = tasks.first.queryOne<Int>()
         with(tasks.second) {
             beforeQuery?.invoke(this)
             val records =
                 atomicTask.logAndReturn(
                     wrapper.orDefault().forList(atomicTask, selectClause.pojo::class, true, listOf()), QueryList
                 )
 
             afterQuery?.invoke(records, QueryList, wrapper.orDefault())
             return total to records as List<K>
         }
     }
 
     fun build(wrapper: KronosDataSourceWrapper? = null): Pair<KronosQueryTask, KronosQueryTask> {
         val recordsTask = selectClause.build(wrapper)
         selectClause.pageEnabled = false
         selectClause.limitCapacity = 0
         if(selectClause.selectAll || selectClause.selectFields.none { it.type == CUSTOM_CRITERIA_SQL }) {
             // 不能直接将 select字段变成1，因为查询的字段可能在where条件中使用
             // 例如：select (select count(1) from table1) as count from table2 where count > 0
             // 只有查询的字段为空或者没有自定义sql时才能将select字段变成1
             selectClause.selectFields = linkedSetOf(Field("1", type = CUSTOM_CRITERIA_SQL))
             selectClause.selectAll = false
         }
         val cntTask = selectClause.build(wrapper)
         cntTask.atomicTask.sql = "SELECT COUNT(1) FROM (${cntTask.atomicTask.sql}) AS total_count"
         cntTask.beforeQuery = null
         cntTask.afterQuery = null
         return cntTask to recordsTask
     }
 }