创建PromptTemplateServiceImpl 代理对象
package net.cnki.aiserver.core.service.impl;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import net.cnki.aiserver.common.exception.AussinessException;
import net.cnki.aiserver.common.exception.BussinessException;
import net.cnki.aiserver.common.pojo.param.AddPromptTemplateParam;
import net.cnki.aiserver.common.pojo.param.DelPromptTemplateParam;
import net.cnki.aiserver.common.pojo.param.ListPtomptTemplateParam;
import net.cnki.aiserver.common.pojo.param.UpdatePromptTemplateParam;
import net.cnki.aiserver.common.pojo.result.PromptTemplateEntityResult;
import net.cnki.aiserver.common.principle.CurrentUserAccessor;
import net.cnki.aiserver.common.utils.JsonUtils;
import net.cnki.aiserver.core.entity.PromptTemplateEntity;
import net.cnki.aiserver.core.mapper.PromptTemplateMapper;
import net.cnki.aiserver.core.mapper.pagetesst;
import net.cnki.aiserver.core.service.PromptTemplateService;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.BeanUtils;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Propagation;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.Assert;
import java.io.IOException;
import java.util.Date;
import java.util.List;
@Service
public class PromptTemplateServiceImpl extends ServiceImpl<PromptTemplateMapper, PromptTemplateEntity> implements PromptTemplateService {
@Override
@Transactional(rollbackFor = AussinessException.class)
public boolean add(AddPromptTemplateParam param) throws BussinessException, AussinessException {
PromptTemplateEntity entity = new PromptTemplateEntity();
entity.setName(param.getName());
entity.setIdentity(param.getIdentity().trim());
entity.setDescription(param.getDescription());
entity.setModel(param.getModel());
entity.setSystemPrompt(param.getSystemPrompt());
entity.setSystemed(param.getSystemed());
entity.setFeatures(JsonUtils.toJsonString(param.getFeatures()));
entity.setEnabled(param.getEnabled());
entity.setCreateTime(new Date());
entity.setClientUserId(CurrentUserAccessor.getUserId());
boolean b = baseMapper.insert(entity) == 1;
//if (true) throw new AussinessException("add方法 抛异常");
add2(param);
// 抛出异常
return true;
}
@Transactional(rollbackFor = BussinessException.class,propagation = Propagation.NEVER)
public boolean add2(AddPromptTemplateParam param) throws AussinessException {
PromptTemplateEntity entity = new PromptTemplateEntity();
entity.setName(param.getName() + "2");
entity.setIdentity(param.getIdentity().trim() + "2");
entity.setDescription(param.getDescription() + "2");
entity.setModel(param.getModel() + "2");
entity.setSystemPrompt(param.getSystemPrompt() + "2");
entity.setSystemed(param.getSystemed());
entity.setFeatures(JsonUtils.toJsonString(param.getFeatures()));
entity.setEnabled(param.getEnabled());
entity.setCreateTime(new Date());
entity.setClientUserId(CurrentUserAccessor.getUserId());
boolean b = baseMapper.insert(entity) == 1;
if (true) throw new AussinessException("add2方法 抛异常");
return b;
}
@Override
public PromptTemplateEntity getByIdentity(String identity) {
Assert.notNull(identity, "identity参数不能为空");
var wrapper = new LambdaQueryWrapper<PromptTemplateEntity>().eq(PromptTemplateEntity::getIdentity, identity);
return baseMapper.selectOne(wrapper);
}
@Override
public boolean del(DelPromptTemplateParam param) {
return baseMapper.delete(new LambdaQueryWrapper<PromptTemplateEntity>()
.eq(PromptTemplateEntity::getId, param.getId())
.eq(PromptTemplateEntity::getClientUserId, CurrentUserAccessor.getUserId())) == 1;
}
@Override
public boolean updatePrompt(UpdatePromptTemplateParam param) {
PromptTemplateEntity entity = new PromptTemplateEntity();
BeanUtils.copyProperties(param, entity);
entity.setFeatures(JsonUtils.toJsonString(param.getFeatures()));
return baseMapper.update(entity, new LambdaQueryWrapper<PromptTemplateEntity>()
.eq(PromptTemplateEntity::getId, param.getId())
.eq(PromptTemplateEntity::getClientUserId, CurrentUserAccessor.getUserId())) == 1;
}
@Override
public Page listPrompt(Page page, ListPtomptTemplateParam param) {
Page selectPage = baseMapper.selectPage(page, new LambdaQueryWrapper<PromptTemplateEntity>()
.eq(StringUtils.isNotBlank(CurrentUserAccessor.getUserId()), PromptTemplateEntity::getClientUserId, CurrentUserAccessor.getUserId())
.eq(StringUtils.isNotBlank(param.getId()), PromptTemplateEntity::getId, param.getId())
.eq(StringUtils.isNotBlank(param.getName()), PromptTemplateEntity::getName, param.getName())
.eq(StringUtils.isNotBlank(param.getIdentity()), PromptTemplateEntity::getIdentity, param.getIdentity())
.eq(StringUtils.isNotBlank(param.getDescription()), PromptTemplateEntity::getDescription, param.getDescription())
.eq(StringUtils.isNotBlank(param.getModel()), PromptTemplateEntity::getModel, param.getModel())
.eq(StringUtils.isNotBlank(param.getSystemPrompt()), PromptTemplateEntity::getSystemPrompt, param.getSystemPrompt())
.eq(param.getSystemed() != null, PromptTemplateEntity::getSystemed, param.getSystemed())
.eq(param.getEnabled() != null, PromptTemplateEntity::getEnabled, param.getEnabled())
);
return selectPage;
}
}
要为 PromptTemplateServiceImpl 创建一个动态代理对象,可以使用 Java 的 JDK 动态代理 或 CGLIB 动态代理。以下是两种实现方式的代码示例。
1. 使用 JDK 动态代理
由于 PromptTemplateServiceImpl 实现了 PromptTemplateService 接口,因此可以使用 JDK 动态代理来创建代理对象。
代码实现
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
public class JDKDynamicProxyExample {
public static void main(String[] args) {
// 创建目标对象
PromptTemplateService target = new PromptTemplateServiceImpl();
// 创建代理对象
PromptTemplateService proxy = (PromptTemplateService) Proxy.newProxyInstance(
target.getClass().getClassLoader(), // 目标类的类加载器
target.getClass().getInterfaces(), // 目标类实现的接口
new InvocationHandler() { // 调用处理器
@Override
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
// 增强逻辑(如添加日志)
System.out.println("Before method: " + method.getName());
// 调用目标方法
Object result = method.invoke(target, args);
// 增强逻辑
System.out.println("After method: " + method.getName());
return result;
}
}
);
// 测试代理对象
try {
proxy.add(null); // 调用 add 方法
} catch (Exception e) {
System.out.println("Caught exception: " + e.getMessage());
}
}
}
运行结果
假设 add 方法抛出异常,输出如下:
Before method: add
Caught exception: add2方法 抛异常
After method: add
2. 使用 CGLIB 动态代理
如果 PromptTemplateServiceImpl 没有实现接口,或者你想基于继承的方式创建代理对象,可以使用 CGLIB 库。
引入依赖
首先,需要在项目中添加 CGLIB 的依赖(通常通过 Spring 已经包含)。
Maven 依赖:
<dependency>
<groupId>cglib</groupId>
<artifactId>cglib</artifactId>
<version>3.3.0</version>
</dependency>
代码实现
import net.sf.cglib.proxy.Enhancer;
import net.sf.cglib.proxy.MethodInterceptor;
import net.sf.cglib.proxy.MethodProxy;
import java.lang.reflect.Method;
public class CGLibDynamicProxyExample {
public static void main(String[] args) {
System.setProperty(DebuggingClassWriter.DEBUG_LOCATION_PROPERTY, "./cglib-classes");
// 创建目标对象
PromptTemplateServiceImpl target = new PromptTemplateServiceImpl();
// 创建增强器
Enhancer enhancer = new Enhancer();
enhancer.setSuperclass(PromptTemplateServiceImpl.class); // 设置父类
enhancer.setCallback(new MethodInterceptor() { // 设置回调
@Override
public Object intercept(Object obj, Method method, Object[] args, MethodProxy proxy) throws Throwable {
// 增强逻辑(如添加日志)
System.out.println("Before method: " + method.getName());
// 调用目标方法
Object result = proxy.invokeSuper(obj, args);
// 增强逻辑
System.out.println("After method: " + method.getName());
return result;
}
});
// 创建代理对象
PromptTemplateServiceImpl proxy = (PromptTemplateServiceImpl) enhancer.create();
// 测试代理对象
try {
proxy.add(null); // 调用 add 方法
} catch (Exception e) {
System.out.println("Caught exception: " + e.getMessage());
}
}
}
运行结果
假设 add 方法抛出异常,输出如下:
Before method: add
Caught exception: add2方法 抛异常
After method: add
- 保存生成的代理类文件(可选)
如果你想查看动态代理生成的字节码文件,可以通过以下方式保存代理类:
JDK 动态代理(未测试)
代码中添加 JVM 参数:
System.getProperties().put("jdk.proxy.ProxyGenerator.saveGeneratedFiles", "true");
CGLIB 动态代理(可用)
代码中 添加系统属性:
System.setProperty(DebuggingClassWriter.DEBUG_LOCATION_PROPERTY, "./cglib-classes");