feat: 一些调整

This commit is contained in:
曹鹏飞 2025-02-08 19:19:48 +08:00
parent be2f5bf5b5
commit 816c07ca2f
3 changed files with 14 additions and 0 deletions

View File

@ -6,6 +6,7 @@ import cn.hutool.core.util.StrUtil;
import com.nflg.mobilebroken.common.util.AdminUserUtil; import com.nflg.mobilebroken.common.util.AdminUserUtil;
import com.nflg.mobilebroken.common.util.SaTokenAdminUtil; import com.nflg.mobilebroken.common.util.SaTokenAdminUtil;
import com.nflg.mobilebroken.common.util.SaTokenAppUtil; import com.nflg.mobilebroken.common.util.SaTokenAppUtil;
import com.nflg.mobilebroken.repository.service.IAdminApiService;
import com.nflg.mobilebroken.repository.service.IAdminRoleService; import com.nflg.mobilebroken.repository.service.IAdminRoleService;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
@ -22,6 +23,9 @@ public class StpInterfaceImpl implements StpInterface {
@Resource @Resource
private IAdminRoleService adminRoleService; private IAdminRoleService adminRoleService;
@Resource
private IAdminApiService adminApiService;
@Override @Override
public List<String> getPermissionList(Object loginId, String loginType) { public List<String> getPermissionList(Object loginId, String loginType) {
if (StrUtil.equals(loginType, SaTokenAdminUtil.getLoginType())){ if (StrUtil.equals(loginType, SaTokenAdminUtil.getLoginType())){
@ -30,6 +34,7 @@ public class StpInterfaceImpl implements StpInterface {
List<String> roleCodes = (List<String>) SaTokenAdminUtil.getExtra("roles"); List<String> roleCodes = (List<String>) SaTokenAdminUtil.getExtra("roles");
urls = adminRoleService.getUrlsByRoleCodes(roleCodes); urls = adminRoleService.getUrlsByRoleCodes(roleCodes);
AdminUserUtil.setPermissions(urls); AdminUserUtil.setPermissions(urls);
urls.addAll(adminApiService.getPublics());
} }
return urls; return urls;
} }

View File

@ -4,6 +4,7 @@ import com.baomidou.mybatisplus.extension.service.IService;
import com.nflg.mobilebroken.common.pojo.vo.ApiVO; import com.nflg.mobilebroken.common.pojo.vo.ApiVO;
import com.nflg.mobilebroken.repository.entity.AdminApi; import com.nflg.mobilebroken.repository.entity.AdminApi;
import java.util.Collection;
import java.util.List; import java.util.List;
/** /**
@ -17,4 +18,6 @@ import java.util.List;
public interface IAdminApiService extends IService<AdminApi> { public interface IAdminApiService extends IService<AdminApi> {
List<ApiVO> getAll(); List<ApiVO> getAll();
Collection<String> getPublics();
} }

View File

@ -9,6 +9,7 @@ import com.nflg.mobilebroken.repository.mapper.AdminApiMapper;
import com.nflg.mobilebroken.repository.service.IAdminApiService; import com.nflg.mobilebroken.repository.service.IAdminApiService;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import java.util.Collection;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.stream.Collectors; import java.util.stream.Collectors;
@ -35,4 +36,9 @@ public class AdminApiServiceImpl extends ServiceImpl<AdminApiMapper, AdminApi> i
.setItems(Convert.toList(ApiItemVO.class, e.getValue()))) .setItems(Convert.toList(ApiItemVO.class, e.getValue())))
.collect(Collectors.toList()); .collect(Collectors.toList());
} }
@Override
public Collection<String> getPublics() {
return lambdaQuery().eq(AdminApi::getIsPublic, true).list().stream().map(AdminApi::getUrl).collect(Collectors.toList());
}
} }