org.dromara.jpom.service.h2db.BaseGlobalOrWorkspaceService Maven / Gradle / Ivy
/*
* The MIT License (MIT)
*
* Copyright (c) 2019 Code Technology Studio
*
* Permission is hereby granted, free of charge, to any person obtaining a copy of
* this software and associated documentation files (the "Software"), to deal in
* the Software without restriction, including without limitation the rights to
* use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
* the Software, and to permit persons to whom the Software is furnished to do so,
* subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
* FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
* COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
* IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
package org.dromara.jpom.service.h2db;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.convert.Convert;
import cn.hutool.core.util.StrUtil;
import cn.hutool.db.Entity;
import cn.hutool.extra.servlet.ServletUtil;
import org.dromara.jpom.common.BaseServerController;
import org.dromara.jpom.common.ServerConst;
import org.dromara.jpom.model.BaseWorkspaceModel;
import org.dromara.jpom.model.PageResultDto;
import org.dromara.jpom.model.user.UserModel;
import org.springframework.util.Assert;
import javax.servlet.http.HttpServletRequest;
import java.util.List;
import java.util.Map;
import java.util.function.Consumer;
/**
* 支持全局贡献的数据
*
* @author bwcx_jzy
* @since 2023/03/23
*/
public abstract class BaseGlobalOrWorkspaceService extends BaseWorkspaceService {
@Override
public T getByKey(String keyValue, HttpServletRequest request) {
String workspace = this.getCheckUserWorkspace(request);
return super.getByKey(keyValue, true, entity -> entity.set("workspaceId", CollUtil.newArrayList(workspace, ServerConst.WORKSPACE_GLOBAL)));
}
/**
* 获取列表
*
* @param request 请求对象
* @return page
*/
@Override
public PageResultDto listPage(HttpServletRequest request) {
// 验证工作空间权限
Map paramMap = ServletUtil.getParamMap(request);
String workspaceId = this.getCheckUserWorkspace(request);
paramMap.put("workspaceId:in", workspaceId + StrUtil.COMMA + ServerConst.WORKSPACE_GLOBAL);
return super.listPage(paramMap);
}
// public PageResultDto listPage(HttpServletRequest request, Map paramMap) {
// String workspaceId = this.getCheckUserWorkspace(request);
// paramMap.put("workspaceId:in", workspaceId + StrUtil.COMMA + ServerConst.WORKSPACE_GLOBAL);
// return super.listPage(paramMap, false);
// }
@Override
public List listByWorkspace(HttpServletRequest request) {
String workspaceId = this.getCheckUserWorkspace(request);
Entity entity = Entity.create();
entity.set("workspaceId", CollUtil.newArrayList(workspaceId, ServerConst.WORKSPACE_GLOBAL));
List entities = super.queryList(entity);
return super.entityToBeanList(entities);
}
/**
* 更新数据,根据ID+工作空间ID
*
* @param info 更新的数据
* @param request 请求
* @return 影响行数
*/
public int updateById(T info, HttpServletRequest request) {
String workspaceId = this.getCheckUserWorkspace(request);
return super.updateById(info, entity -> entity.set("workspaceId", CollUtil.newArrayList(workspaceId, ServerConst.WORKSPACE_GLOBAL)));
}
@Override
public int delByKey(String keyValue, HttpServletRequest request) {
String workspaceId = this.getCheckUserWorkspace(request);
return super.delByKey(keyValue, entity -> entity.set("workspaceId", CollUtil.newArrayList(workspaceId, ServerConst.WORKSPACE_GLOBAL)));
}
@Override
public int delByWorkspace(HttpServletRequest request, Consumer consumer) {
String workspaceId = this.getCheckUserWorkspace(request);
return super.delByKey(null, entity -> {
entity.set("workspaceId", CollUtil.newArrayList(workspaceId, ServerConst.WORKSPACE_GLOBAL));
consumer.accept(entity);
int size = entity.size();
Assert.state(size > 1, "没有添加任何参数");
});
}
@Override
public T getByKey(String keyValue, UserModel userModel) {
return super.getByKey(keyValue, userModel);
}
/**
* 查询数据,并判断管理和创建人
*
* @param keyValue 主机id
* @param request 请求信息
* @return data
*/
public T getByKeyAndGlobal(String keyValue, HttpServletRequest request) {
return this.getByKeyAndGlobal(keyValue, request, "数据不存在");
}
/**
* 查询数据,并判断管理和创建人
*
* @param keyValue 主机id
* @param request 请求信息
* @return data
*/
public T getByKeyAndGlobal(String keyValue, HttpServletRequest request, String errorMsg) {
T byKey = this.getByKey(keyValue, request);
Assert.notNull(byKey, errorMsg);
UserModel userModel = BaseServerController.getUserByThreadLocal();
Assert.notNull(userModel, "当前未登录不能操作此数据");
if (StrUtil.equals(byKey.getWorkspaceId(), ServerConst.WORKSPACE_GLOBAL) && !userModel.isSystemUser()) {
// 是全局共享数据,并且不是管理员
Assert.state(StrUtil.equals(userModel.getId(), byKey.getCreateUser()), "没有当前数据权限,需要管理员或者数据创建人才操作该数据");
}
return byKey;
}
public String covertGlobalWorkspace(HttpServletRequest request) {
Map paramMap = ServletUtil.getParamMap(request);
boolean global = Convert.toBool(paramMap.get("global"), false);
// 判断是否为全局模式
return global ? ServerConst.WORKSPACE_GLOBAL : this.getCheckUserWorkspace(request);
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy