查看动态代理文件-笔记

创建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
  1. 保存生成的代理类文件(可选)
    如果你想查看动态代理生成的字节码文件,可以通过以下方式保存代理类:

JDK 动态代理(未测试)

代码中添加 JVM 参数:

System.getProperties().put("jdk.proxy.ProxyGenerator.saveGeneratedFiles", "true");

CGLIB 动态代理(可用)

代码中 添加系统属性:

System.setProperty(DebuggingClassWriter.DEBUG_LOCATION_PROPERTY, "./cglib-classes");

在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值