|
1 | 1 | package com.easy.query.test;
|
2 | 2 |
|
3 | 3 | import com.easy.query.api.proxy.base.MapProxy;
|
| 4 | +import com.easy.query.core.api.pagination.EasyPageResult; |
4 | 5 | import com.easy.query.core.basic.extension.listener.JdbcExecuteAfterArg;
|
5 | 6 | import com.easy.query.core.metadata.EntityMetadata;
|
6 | 7 | import com.easy.query.core.util.EasySQLUtil;
|
| 8 | +import com.easy.query.test.common.PageResult; |
| 9 | +import com.easy.query.test.dto.autodto.SchoolClassAOProp14; |
7 | 10 | import com.easy.query.test.entity.BlogEntity;
|
8 | 11 | import com.easy.query.test.entity.Topic;
|
9 | 12 | import com.easy.query.test.entity.proxy.TopicProxy;
|
| 13 | +import com.easy.query.test.entity.school.SchoolClass; |
10 | 14 | import com.easy.query.test.listener.ListenerContext;
|
| 15 | +import com.easy.query.test.page.mp.SearchCountPager; |
11 | 16 | import org.junit.Assert;
|
12 | 17 | import org.junit.Test;
|
13 | 18 |
|
@@ -94,4 +99,87 @@ public void caseWhenTest2() {
|
94 | 99 | Assert.assertEquals("123(String),1(String),2(String),123(String),1(Integer),456(String),1(Integer),3(Integer),4(Integer),someTitle(String)", EasySQLUtil.sqlParameterToString(jdbcExecuteAfterArg.getBeforeArg().getSqlParameters().get(0)));
|
95 | 100 |
|
96 | 101 | }
|
| 102 | + |
| 103 | + @Test |
| 104 | + public void searchCountPageTest(){ |
| 105 | + System.out.println("查询count"); |
| 106 | + |
| 107 | + { |
| 108 | + |
| 109 | + |
| 110 | + ListenerContext listenerContext = new ListenerContext(true); |
| 111 | + listenerContextManager.startListen(listenerContext); |
| 112 | + |
| 113 | + PageResult<Topic> pageResult1 = easyEntityQuery.queryable(Topic.class) |
| 114 | + .where(t_topic -> { |
| 115 | + t_topic.title().like("123"); |
| 116 | + }).toPageResult(SearchCountPager.of(1, 10, true)); |
| 117 | + |
| 118 | + Assert.assertNotNull(listenerContext.getJdbcExecuteAfterArgs()); |
| 119 | + |
| 120 | + { |
| 121 | + |
| 122 | + JdbcExecuteAfterArg jdbcExecuteAfterArg = listenerContext.getJdbcExecuteAfterArgs().get(0); |
| 123 | + Assert.assertEquals("SELECT COUNT(*) FROM `t_topic` WHERE `title` LIKE ?", jdbcExecuteAfterArg.getBeforeArg().getSql()); |
| 124 | + Assert.assertEquals("%123%(String)", EasySQLUtil.sqlParameterToString(jdbcExecuteAfterArg.getBeforeArg().getSqlParameters().get(0))); |
| 125 | + } |
| 126 | + { |
| 127 | + JdbcExecuteAfterArg jdbcExecuteAfterArg = listenerContext.getJdbcExecuteAfterArgs().get(1); |
| 128 | + Assert.assertEquals("SELECT `id`,`stars`,`title`,`create_time` FROM `t_topic` WHERE `title` LIKE ? LIMIT 10", jdbcExecuteAfterArg.getBeforeArg().getSql()); |
| 129 | + Assert.assertEquals("%123%(String)", EasySQLUtil.sqlParameterToString(jdbcExecuteAfterArg.getBeforeArg().getSqlParameters().get(0))); |
| 130 | + } |
| 131 | + listenerContextManager.clear(); |
| 132 | + } |
| 133 | + System.out.println("不要查询count"); |
| 134 | + |
| 135 | + |
| 136 | + { |
| 137 | + |
| 138 | + |
| 139 | + ListenerContext listenerContext = new ListenerContext(true); |
| 140 | + listenerContextManager.startListen(listenerContext); |
| 141 | + |
| 142 | + PageResult<Topic> pageResult2 = easyEntityQuery.queryable(Topic.class) |
| 143 | + .where(t_topic -> { |
| 144 | + t_topic.title().like("123"); |
| 145 | + }).toPageResult(SearchCountPager.of(1, 10, false)); |
| 146 | + |
| 147 | + Assert.assertNotNull(listenerContext.getJdbcExecuteAfterArgs()); |
| 148 | + |
| 149 | + { |
| 150 | + JdbcExecuteAfterArg jdbcExecuteAfterArg = listenerContext.getJdbcExecuteAfterArgs().get(0); |
| 151 | + Assert.assertEquals("SELECT `id`,`stars`,`title`,`create_time` FROM `t_topic` WHERE `title` LIKE ? LIMIT 10", jdbcExecuteAfterArg.getBeforeArg().getSql()); |
| 152 | + Assert.assertEquals("%123%(String)", EasySQLUtil.sqlParameterToString(jdbcExecuteAfterArg.getBeforeArg().getSqlParameters().get(0))); |
| 153 | + } |
| 154 | + listenerContextManager.clear(); |
| 155 | + } |
| 156 | + { |
| 157 | + |
| 158 | + |
| 159 | + ListenerContext listenerContext = new ListenerContext(true); |
| 160 | + listenerContextManager.startListen(listenerContext); |
| 161 | + |
| 162 | + EasyPageResult<Topic> pageResult2 = easyEntityQuery.queryable(Topic.class) |
| 163 | + .where(t_topic -> { |
| 164 | + t_topic.id().isNotNull(); |
| 165 | + }).orderBy(t_topic -> t_topic.createTime().asc()).toPageResult(7, 10); |
| 166 | + |
| 167 | + Assert.assertNotNull(listenerContext.getJdbcExecuteAfterArgs()); |
| 168 | + |
| 169 | + { |
| 170 | + { |
| 171 | + |
| 172 | + JdbcExecuteAfterArg jdbcExecuteAfterArg = listenerContext.getJdbcExecuteAfterArgs().get(0); |
| 173 | + Assert.assertEquals("SELECT COUNT(*) FROM `t_topic` WHERE `id` IS NOT NULL", jdbcExecuteAfterArg.getBeforeArg().getSql()); |
| 174 | +// Assert.assertEquals("%123%(String)", EasySQLUtil.sqlParameterToString(jdbcExecuteAfterArg.getBeforeArg().getSqlParameters().get(0))); |
| 175 | + } |
| 176 | + { |
| 177 | + JdbcExecuteAfterArg jdbcExecuteAfterArg = listenerContext.getJdbcExecuteAfterArgs().get(1); |
| 178 | + Assert.assertEquals("SELECT `id`,`stars`,`title`,`create_time` FROM `t_topic` WHERE `id` IS NOT NULL ORDER BY `create_time` DESC LIMIT 10 OFFSET 31", jdbcExecuteAfterArg.getBeforeArg().getSql()); |
| 179 | +// Assert.assertEquals("%123%(String)", EasySQLUtil.sqlParameterToString(jdbcExecuteAfterArg.getBeforeArg().getSqlParameters().get(0))); |
| 180 | + } |
| 181 | + } |
| 182 | + listenerContextManager.clear(); |
| 183 | + } |
| 184 | + } |
97 | 185 | }
|
0 commit comments