package com.ard.tianyu.aspect;
|
|
import com.ard.common.exception.NoPermissionException;
|
import com.ard.common.utils.SecurityUtils;
|
import com.ard.tianyu.service.impl.UavControlManager;
|
import lombok.extern.slf4j.Slf4j;
|
import org.aspectj.lang.ProceedingJoinPoint;
|
import org.aspectj.lang.annotation.Around;
|
import org.aspectj.lang.annotation.Aspect;
|
import org.aspectj.lang.annotation.Pointcut;
|
import org.aspectj.lang.reflect.MethodSignature;
|
import org.springframework.stereotype.Component;
|
|
import java.lang.reflect.Field;
|
import java.lang.reflect.Method;
|
|
|
/**
|
* 控制权限切面
|
*
|
* @author liusuyi
|
*/
|
@Aspect
|
@Component
|
@Slf4j
|
public class ControlAuthorityAspect {
|
|
@Pointcut("@annotation(com.ard.tianyu.aspect.annotation.CheckControlAuthority)")
|
public void dsPointCut() {
|
|
}
|
|
@Around("dsPointCut()")
|
public Object around(ProceedingJoinPoint point) throws Throwable {
|
MethodSignature signature = (MethodSignature) point.getSignature();
|
Method method = signature.getMethod();
|
Boolean result = controlScopeFilter(point);
|
if (result) {
|
log.debug("已获取无人机控制权--{}", method.getName());
|
// 执行目标方法
|
return point.proceed();
|
}
|
log.debug("未获取无人机控制权--{}", method.getName());
|
throw new NoPermissionException("未获取无人机控制权");
|
}
|
|
/**
|
* 操控判断逻辑
|
*
|
* @param joinPoint 切点
|
*/
|
public Boolean controlScopeFilter(ProceedingJoinPoint joinPoint) {
|
// 获取方法参数
|
Object[] args = joinPoint.getArgs();
|
// 使用反射获取参数中的 boxSn
|
for (Object arg : args) {
|
String boxSn = getBoxSn(arg);
|
if (boxSn != null) {
|
//判断无人机控制者是否为当前登录用户
|
Long controller = UavControlManager.uavControlMap.get(boxSn);
|
if (controller != null && controller.equals(SecurityUtils.getUserId())) {
|
return true;
|
}
|
}
|
}
|
return false;
|
}
|
|
// 通过反射获取对象中的无人机boxSn字段
|
public static String getBoxSn(Object dto) {
|
try {
|
// 获取类的所有字段
|
Field[] fields = dto.getClass().getDeclaredFields();
|
for (Field field : fields) {
|
// 设置字段可访问
|
field.setAccessible(true);
|
|
// 判断是否是 boxSn 字段
|
if ("boxSn".equals(field.getName())) {
|
// 获取并返回 droneId 的值
|
return (String) field.get(dto);
|
}
|
}
|
} catch (IllegalAccessException e) {
|
e.printStackTrace();
|
}
|
return null; // 如果没有找到 droneId 字段，返回 null
|
}
|
}
|
