package jnpf.base.controller;
|
|
import cn.dev33.satoken.same.SaSameUtil;
|
import cn.dev33.satoken.stp.StpUtil;
|
import jnpf.base.ActionResult;
|
import jnpf.base.UserInfo;
|
import jnpf.base.model.uploadedit.UploadEditLease;
|
import jnpf.base.model.uploadedit.UploadEditSaveCommand;
|
import jnpf.base.service.UploadEditDraftService;
|
import jnpf.database.util.TenantDataSourceUtil;
|
import jnpf.exception.DataException;
|
import jnpf.onlinedev.UploadEditApi;
|
import jnpf.util.NoDataSourceBind;
|
import jnpf.util.UserProvider;
|
import lombok.RequiredArgsConstructor;
|
import org.springframework.web.bind.annotation.*;
|
import java.util.function.Supplier;
|
|
@RestController
|
@RequestMapping("/Internal/visualdev/upload-edit")
|
@RequiredArgsConstructor
|
public class UploadEditInternalController implements UploadEditApi {
|
private final UploadEditDraftService service;
|
@Override @PostMapping("/status-access") @NoDataSourceBind
|
public ActionResult<Boolean> canReadStatus(@RequestParam("tenantId") String tenantId,
|
@RequestParam("ownerId") String ownerId, @RequestParam("leaseId") String leaseId) {
|
return asOwner(tenantId, ownerId, () -> ActionResult.success(service.canReadStatus(tenantId, ownerId, leaseId)));
|
}
|
|
@Override @PostMapping("/lease") @NoDataSourceBind
|
public ActionResult<UploadEditLease> lease(@RequestParam("tenantId") String tenantId,
|
@RequestParam("ownerId") String ownerId, @RequestParam("leaseId") String leaseId) {
|
return asOwner(tenantId, ownerId, () -> ActionResult.success(service.lease(tenantId, ownerId, leaseId)));
|
}
|
@Override @PostMapping("/save") @NoDataSourceBind
|
public ActionResult<Boolean> save(@RequestBody UploadEditSaveCommand command) {
|
return asOwner(command.getTenantId(), command.getOwnerId(), () -> ActionResult.success(service.save(command)));
|
}
|
|
private <T> T asOwner(String tenantId, String ownerId, Supplier<T> work) {
|
SaSameUtil.checkCurrentRequestToken();
|
if (tenantId == null || tenantId.isBlank() || ownerId == null || ownerId.isBlank()) throw new DataException("缺少编辑租约归属");
|
UserInfo previous = UserProvider.getLocalLoginUser();
|
boolean switched = false;
|
try {
|
TenantDataSourceUtil.switchTenant(tenantId);
|
UserInfo owner = new UserInfo();
|
owner.setUserId(ownerId);
|
owner.setTenantId(tenantId);
|
UserProvider.setLocalLoginUser(owner);
|
StpUtil.switchTo(UserProvider.splicingLoginId(ownerId));
|
switched = true;
|
return work.get();
|
} finally {
|
if (switched) StpUtil.endSwitch();
|
UserProvider.clearLocalUser();
|
if (previous != null) UserProvider.setLocalLoginUser(previous);
|
TenantDataSourceUtil.clearLocalTenantInfo();
|
}
|
}
|
}
|
