jhynsoo/worlabel
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refactor: Aspect 최적화

Browse Source
This commit is contained in:
김용수 2024-09-23 22:24:18 +09:00
parent 211d77481d
commit e2fd9ace95
1 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
backend/src/main/java/com/worlabel/global/aspect/PrivilegeCheckAspect.java
View File

@ -30,27 +30,34 @@ public class PrivilegeCheckAspect {
// CheckPrivilege 어노테이션이 붙은 메서드가 실행되기전 실행
@Before("@annotation(checkPrivilege)")
public void checkPrivilege(JoinPoint joinPoint, CheckPrivilege checkPrivilege) {
int memberId = getMemberId();
int projectId = getProjectId(joinPoint);
checkPrivilegeUnauthorized(memberId, projectId, checkPrivilege.value());
}
private int getProjectId(JoinPoint joinPoint) {
MethodSignature signature = (MethodSignature) joinPoint.getSignature();
Method method = signature.getMethod();
Object[] args = joinPoint.getArgs();
Parameter[] parameters = method.getParameters();
Object principal = SecurityContextHolder.getContext()
.getAuthentication()
.getPrincipal();
int memberId = ((AuthMemberDto) principal).getId();
Integer projectId = null;
for (int paramIdx = 0; paramIdx < parameters.length; paramIdx++) {
String paramName = parameters[paramIdx].getName();
if (paramName.equals("projectId")) {
projectId = (Integer) args[paramIdx];
break;
return (Integer) args[paramIdx];
}
}
checkPrivilegeUnauthorized(memberId, projectId, checkPrivilege.value());
throw new CustomException(ErrorCode.SERVER_ERROR);
}
private int getMemberId() {
Object principal = SecurityContextHolder.getContext()
.getAuthentication()
.getPrincipal();
return ((AuthMemberDto) principal).getId();
}
public void checkPrivilegeUnauthorized(final Integer memberId, final Integer projectId, final PrivilegeType privilegeType) {