Spring Boot - beetlsql

拼搏现实的明天。 2023-10-12 20:08 55阅读 0赞

## 数据源配置 ##

package com.lfsun.springbootormbeetlsql.config;
 
 import com.zaxxer.hikari.HikariDataSource;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.core.env.Environment;
 
 import javax.sql.DataSource;
 
 /**
 * 
 * Beetl数据源配置
 * 
 */
 @Configuration
 public class BeetlConfig {
 
 
 /**
 * Beetl需要显式的配置数据源，方可启动项目
 */
 @Bean(name = "datasource")
 public DataSource getDataSource(Environment env) {
 
 HikariDataSource dataSource = new HikariDataSource();
 dataSource.setDriverClassName(env.getProperty("spring.datasource.driver-class-name"));
 dataSource.setJdbcUrl(env.getProperty("spring.datasource.url"));
 dataSource.setUsername(env.getProperty("spring.datasource.username"));
 dataSource.setPassword(env.getProperty("spring.datasource.password"));
 return dataSource;
 }
 }

## dao ##

package com.lfsun.springbootormbeetlsql.dao;
 
 import com.lfsun.springbootormbeetlsql.entity.User;
 import org.beetl.sql.core.mapper.BaseMapper;
 import org.springframework.stereotype.Component;
 
 /**
 * 
 * UserDao
 * 
 */
 @Component
 public interface UserDao extends BaseMapper<User> {
 
 
 }

## 实现类 ##

package com.lfsun.springbootormbeetlsql.service.impl;
 
 import cn.hutool.core.util.ObjectUtil;
 
 import com.lfsun.springbootormbeetlsql.dao.UserDao;
 import com.lfsun.springbootormbeetlsql.entity.User;
 import com.lfsun.springbootormbeetlsql.service.UserService;
 import lombok.extern.slf4j.Slf4j;
 import org.beetl.sql.core.engine.PageQuery;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
 import java.util.List;
 
 /**
 * 
 * User Service
 * 
 
 */
 @Service
 @Slf4j
 public class UserServiceImpl implements UserService {
 
 
 private final UserDao userDao;
 
 @Autowired
 public UserServiceImpl(UserDao userDao) {
 
 this.userDao = userDao;
 }
 
 /**
 * 新增用户
 *
 * @param user 用户
 */
 @Override
 public User saveUser(User user) {
 
 userDao.insert(user, true);
 return user;
 }
 
 /**
 * 批量插入用户
 *
 * @param users 用户列表
 */
 @Override
 public void saveUserList(List<User> users) {
 
 userDao.insertBatch(users);
 }
 
 /**
 * 根据主键删除用户
 *
 * @param id 主键
 */
 @Override
 public void deleteUser(Long id) {
 
 userDao.deleteById(id);
 }
 
 /**
 * 更新用户
 *
 * @param user 用户
 * @return 更新后的用户
 */
 @Override
 public User updateUser(User user) {
 
 if (ObjectUtil.isNull(user)) {
 
 throw new RuntimeException("用户id不能为null");
 }
 userDao.updateTemplateById(user);
 return userDao.single(user.getId());
 }
 
 /**
 * 查询单个用户
 *
 * @param id 主键id
 * @return 用户信息
 */
 @Override
 public User getUser(Long id) {
 
 return userDao.single(id);
 }
 
 /**
 * 查询用户列表
 *
 * @return 用户列表
 */
 @Override
 public List<User> getUserList() {
 
 return userDao.all();
 }
 
 /**
 * 分页查询
 *
 * @param currentPage 当前页
 * @param pageSize 每页条数
 * @return 分页用户列表
 */
 @Override
 public PageQuery<User> getUserByPage(Integer currentPage, Integer pageSize) {
 
 return userDao.createLambdaQuery().page(currentPage, pageSize);
 }
 }

## 测试 ##

package com.lfsun.springbootormbeetlsql.service;
 
 import cn.hutool.core.collection.CollUtil;
 import cn.hutool.core.date.DateTime;
 import cn.hutool.core.util.IdUtil;
 import cn.hutool.core.util.ObjectUtil;
 import cn.hutool.crypto.SecureUtil;
 import cn.hutool.json.JSONUtil;
 import com.lfsun.springbootormbeetlsql.SpringBootOrmBeetlsqlApplicationTests;
 import com.lfsun.springbootormbeetlsql.entity.User;
 import lombok.extern.slf4j.Slf4j;
 import org.assertj.core.util.Lists;
 import org.beetl.sql.core.engine.PageQuery;
 import org.junit.Assert;
 import org.junit.Test;
 import org.springframework.beans.factory.annotation.Autowired;
 
 import java.util.List;
 
 /**
 * 
 * User Service测试
 * 
 */
 @Slf4j
 public class UserServiceTest extends SpringBootOrmBeetlsqlApplicationTests {
 
 @Autowired
 private UserService userService;
 
 @Test
 public void saveUser() {
 
 String salt = IdUtil.fastSimpleUUID();
 User user = User.builder().name("saveTest3").password(SecureUtil.md5("123456" + salt)).salt(salt).email("saveTest3@lfsun.com").phoneNumber("17300000003").status(1).lastLoginTime(new DateTime()).createTime(new DateTime()).lastUpdateTime(new DateTime()).build();
 
 user = userService.saveUser(user);
 Assert.assertTrue(ObjectUtil.isNotNull(user.getId()));
 log.debug("【user】= {}", user);
 }
 
 @Test
 public void saveUserList() {
 
 List<User> users = Lists.newArrayList();
 for (int i = 5; i < 15; i++) {
 
 String salt = IdUtil.fastSimpleUUID();
 User user = User.builder().name("saveTest" + i).password(SecureUtil.md5("123456" + salt)).salt(salt).email("saveTest" + i + "@lfsun.com").phoneNumber("1730000000" + i).status(1).lastLoginTime(new DateTime()).createTime(new DateTime()).lastUpdateTime(new DateTime()).build();
 users.add(user);
 }
 userService.saveUserList(users);
 Assert.assertTrue(userService.getUserList().size() > 2);
 }
 
 @Test
 public void deleteUser() {
 
 userService.deleteUser(1L);
 User user = userService.getUser(1L);
 Assert.assertTrue(ObjectUtil.isNull(user));
 }
 
 @Test
 public void updateUser() {
 
 User user = userService.getUser(2L);
 user.setName("beetlSql 修改后的名字");
 User update = userService.updateUser(user);
 Assert.assertEquals("beetlSql 修改后的名字", update.getName());
 log.debug("【update】= {}", update);
 }
 
 @Test
 public void getUser() {
 
 User user = userService.getUser(1L);
 Assert.assertNotNull(user);
 log.debug("【user】= {}", user);
 }
 
 @Test
 public void getUserList() {
 
 List<User> userList = userService.getUserList();
 Assert.assertTrue(CollUtil.isNotEmpty(userList));
 log.debug("【userList】= {}", userList);
 }
 
 @Test
 public void getUserByPage() {
 
 List<User> userList = userService.getUserList();
 PageQuery<User> userByPage = userService.getUserByPage(1, 5);
 Assert.assertEquals(5, userByPage.getList().size());
 Assert.assertEquals(userList.size(), userByPage.getTotalRow());
 log.debug("【userByPage】= {}", JSONUtil.toJsonStr(userByPage));
 }
 }

## 完整代码案例地址 ##

[Spring Boot 集成 beetlsql][Spring Boot _ beetlsql]

[Spring Boot _ beetlsql]: https://gitee.com/lfsun666/springboot-study-2023/tree/master/spring-boot-orm-beetlsql