forked from github/dataease
Merge pull request #10448 from dataease/pr@dev-v2@perf_share_ticket
perf(仪表板): 漏提代码
This commit is contained in:
fit2cloud-chenyw
GitHub
commit
94f6bff735
@ -0,0 +1,108 @@
|
||||
package io.dataease.share.dao.auto.entity;
|
||||
|
||||
import com.baomidou.mybatisplus.annotation.TableName;
|
||||
import java.io.Serializable;
|
||||
|
||||
/**
|
||||
* <p>
|
||||
*
|
||||
* </p>
|
||||
*
|
||||
* @author fit2cloud
|
||||
* @since 2024-06-21
|
||||
*/
|
||||
@TableName("core_share_ticket")
|
||||
public class CoreShareTicket implements Serializable {
|
||||
|
||||
private static final long serialVersionUID = 1L;
|
||||
|
||||
/**
|
||||
* ID
|
||||
*/
|
||||
private Long id;
|
||||
|
||||
/**
|
||||
* 分享uuid
|
||||
*/
|
||||
private String uuid;
|
||||
|
||||
/**
|
||||
* ticket
|
||||
*/
|
||||
private String ticket;
|
||||
|
||||
/**
|
||||
* ticket有效期
|
||||
*/
|
||||
private Long exp;
|
||||
|
||||
/**
|
||||
* ticket参数
|
||||
*/
|
||||
private String args;
|
||||
|
||||
/**
|
||||
* 首次访问时间
|
||||
*/
|
||||
private Long accessTime;
|
||||
|
||||
public Long getId() {
|
||||
return id;
|
||||
}
|
||||
|
||||
public void setId(Long id) {
|
||||
this.id = id;
|
||||
}
|
||||
|
||||
public String getUuid() {
|
||||
return uuid;
|
||||
}
|
||||
|
||||
public void setUuid(String uuid) {
|
||||
this.uuid = uuid;
|
||||
}
|
||||
|
||||
public String getTicket() {
|
||||
return ticket;
|
||||
}
|
||||
|
||||
public void setTicket(String ticket) {
|
||||
this.ticket = ticket;
|
||||
}
|
||||
|
||||
public Long getExp() {
|
||||
return exp;
|
||||
}
|
||||
|
||||
public void setExp(Long exp) {
|
||||
this.exp = exp;
|
||||
}
|
||||
|
||||
public String getArgs() {
|
||||
return args;
|
||||
}
|
||||
|
||||
public void setArgs(String args) {
|
||||
this.args = args;
|
||||
}
|
||||
|
||||
public Long getAccessTime() {
|
||||
return accessTime;
|
||||
}
|
||||
|
||||
public void setAccessTime(Long accessTime) {
|
||||
this.accessTime = accessTime;
|
||||
}
|
||||
|
||||
@Override
|
||||
public String toString() {
|
||||
return "CoreShareTicket{" +
|
||||
"id = " + id +
|
||||
", uuid = " + uuid +
|
||||
", ticket = " + ticket +
|
||||
", exp = " + exp +
|
||||
", args = " + args +
|
||||
", accessTime = " + accessTime +
|
||||
"}";
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,18 @@
|
||||
package io.dataease.share.dao.auto.mapper;
|
||||
|
||||
import io.dataease.share.dao.auto.entity.CoreShareTicket;
|
||||
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
|
||||
import org.apache.ibatis.annotations.Mapper;
|
||||
|
||||
/**
|
||||
* <p>
|
||||
* Mapper 接口
|
||||
* </p>
|
||||
*
|
||||
* @author fit2cloud
|
||||
* @since 2024-06-21
|
||||
*/
|
||||
@Mapper
|
||||
public interface CoreShareTicketMapper extends BaseMapper<CoreShareTicket> {
|
||||
|
||||
}
|
||||
@ -0,0 +1,153 @@
|
||||
package io.dataease.share.manage;
|
||||
|
||||
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
||||
import io.dataease.api.xpack.share.request.TicketCreator;
|
||||
import io.dataease.api.xpack.share.request.TicketDelRequest;
|
||||
import io.dataease.api.xpack.share.request.TicketSwitchRequest;
|
||||
import io.dataease.api.xpack.share.vo.TicketVO;
|
||||
import io.dataease.api.xpack.share.vo.TicketValidVO;
|
||||
import io.dataease.commons.utils.CodingUtil;
|
||||
import io.dataease.exception.DEException;
|
||||
import io.dataease.share.dao.auto.entity.CoreShareTicket;
|
||||
import io.dataease.share.dao.auto.entity.XpackShare;
|
||||
import io.dataease.share.dao.auto.mapper.CoreShareTicketMapper;
|
||||
import io.dataease.share.dao.auto.mapper.XpackShareMapper;
|
||||
import io.dataease.share.dao.ext.mapper.XpackShareExtMapper;
|
||||
import io.dataease.utils.AuthUtils;
|
||||
import io.dataease.utils.BeanUtils;
|
||||
import io.dataease.utils.IDUtils;
|
||||
import jakarta.annotation.Resource;
|
||||
import org.apache.commons.collections4.CollectionUtils;
|
||||
import org.apache.commons.lang3.ObjectUtils;
|
||||
import org.apache.commons.lang3.StringUtils;
|
||||
import org.springframework.stereotype.Component;
|
||||
import org.springframework.transaction.annotation.Transactional;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
@Component
|
||||
public class ShareTicketManage {
|
||||
|
||||
@Resource
|
||||
private CoreShareTicketMapper coreShareTicketMapper;
|
||||
|
||||
@Resource
|
||||
private XpackShareMapper xpackShareMapper;
|
||||
|
||||
@Resource
|
||||
private XpackShareExtMapper xpackShareExtMapper;
|
||||
|
||||
public CoreShareTicket getByTicket(String ticket) {
|
||||
QueryWrapper<CoreShareTicket> queryWrapper = new QueryWrapper<>();
|
||||
queryWrapper.eq("ticket", ticket);
|
||||
return coreShareTicketMapper.selectOne(queryWrapper);
|
||||
}
|
||||
|
||||
public String saveTicket(TicketCreator creator) {
|
||||
String ticket = creator.getTicket();
|
||||
if (StringUtils.isNotBlank(ticket)) {
|
||||
CoreShareTicket ticketEntity = getByTicket(ticket);
|
||||
if (ObjectUtils.isNotEmpty(ticketEntity)) {
|
||||
if (creator.isGenerateNew()) {
|
||||
ticketEntity.setAccessTime(null);
|
||||
ticketEntity.setTicket(CodingUtil.shortUuid());
|
||||
}
|
||||
ticketEntity.setArgs(creator.getArgs());
|
||||
ticketEntity.setExp(creator.getExp());
|
||||
ticketEntity.setUuid(creator.getUuid());
|
||||
coreShareTicketMapper.updateById(ticketEntity);
|
||||
return ticketEntity.getTicket();
|
||||
}
|
||||
}
|
||||
ticket = CodingUtil.shortUuid();
|
||||
CoreShareTicket linkTicket = new CoreShareTicket();
|
||||
linkTicket.setId(IDUtils.snowID());
|
||||
linkTicket.setTicket(ticket);
|
||||
linkTicket.setArgs(creator.getArgs());
|
||||
linkTicket.setExp(creator.getExp());
|
||||
linkTicket.setUuid(creator.getUuid());
|
||||
coreShareTicketMapper.insert(linkTicket);
|
||||
return ticket;
|
||||
}
|
||||
|
||||
public void deleteTicket(TicketDelRequest request) {
|
||||
String ticket = request.getTicket();
|
||||
if (StringUtils.isBlank(ticket)) {
|
||||
DEException.throwException("ticket为必填参数");
|
||||
}
|
||||
QueryWrapper<CoreShareTicket> queryWrapper = new QueryWrapper<>();
|
||||
queryWrapper.eq("ticket", ticket);
|
||||
coreShareTicketMapper.delete(queryWrapper);
|
||||
}
|
||||
|
||||
public void switchRequire(TicketSwitchRequest request) {
|
||||
String resourceId = request.getResourceId();
|
||||
Boolean require = request.getRequire();
|
||||
QueryWrapper<XpackShare> queryWrapper = new QueryWrapper<>();
|
||||
queryWrapper.eq("resource_id", resourceId);
|
||||
queryWrapper.eq("creator", AuthUtils.getUser().getUserId());
|
||||
XpackShare xpackShare = xpackShareMapper.selectOne(queryWrapper);
|
||||
xpackShare.setTicketRequire(require);
|
||||
xpackShareMapper.updateById(xpackShare);
|
||||
}
|
||||
|
||||
public List<TicketVO> query(Long resourceId) {
|
||||
QueryWrapper<XpackShare> queryWrapper = new QueryWrapper<>();
|
||||
queryWrapper.eq("resource_id", resourceId);
|
||||
queryWrapper.eq("creator", AuthUtils.getUser().getUserId());
|
||||
XpackShare xpackShare = xpackShareMapper.selectOne(queryWrapper);
|
||||
if (ObjectUtils.isEmpty(xpackShare)) return null;
|
||||
String uuid = xpackShare.getUuid();
|
||||
if (StringUtils.isBlank(uuid)) return null;
|
||||
QueryWrapper<CoreShareTicket> ticketQueryWrapper = new QueryWrapper<>();
|
||||
ticketQueryWrapper.eq("uuid", uuid);
|
||||
List<CoreShareTicket> coreShareTickets = coreShareTicketMapper.selectList(ticketQueryWrapper);
|
||||
if (CollectionUtils.isEmpty(coreShareTickets)) return null;
|
||||
return coreShareTickets.stream().map(item -> BeanUtils.copyBean(new TicketVO(), item)).toList();
|
||||
}
|
||||
|
||||
@Transactional
|
||||
public void updateByUuidChange(String originalUuid, String newUuid) {
|
||||
xpackShareExtMapper.updateTicketUuid(originalUuid, newUuid);
|
||||
}
|
||||
|
||||
@Transactional
|
||||
public void deleteByShare(String uuid) {
|
||||
QueryWrapper<CoreShareTicket> ticketQueryWrapper = new QueryWrapper<>();
|
||||
ticketQueryWrapper.eq("uuid", uuid);
|
||||
coreShareTicketMapper.delete(ticketQueryWrapper);
|
||||
}
|
||||
|
||||
public TicketValidVO validateTicket(String ticket, XpackShare share) {
|
||||
TicketValidVO vo = new TicketValidVO();
|
||||
if (StringUtils.isBlank(ticket)) {
|
||||
vo.setTicketValid(!share.getTicketRequire());
|
||||
return vo;
|
||||
}
|
||||
CoreShareTicket linkTicket = getByTicket(ticket);
|
||||
if (ObjectUtils.isEmpty(linkTicket)) {
|
||||
vo.setTicketValid(false);
|
||||
return vo;
|
||||
}
|
||||
vo.setTicketValid(true);
|
||||
vo.setArgs(linkTicket.getArgs());
|
||||
Long accessTime = linkTicket.getAccessTime();
|
||||
long now = System.currentTimeMillis();
|
||||
if (ObjectUtils.isEmpty(accessTime)) {
|
||||
accessTime = now;
|
||||
vo.setTicketExp(false);
|
||||
linkTicket.setAccessTime(accessTime);
|
||||
coreShareTicketMapper.updateById(linkTicket);
|
||||
return vo;
|
||||
}
|
||||
Long exp = linkTicket.getExp();
|
||||
if (ObjectUtils.isEmpty(exp) || exp.equals(0L)) {
|
||||
vo.setTicketExp(false);
|
||||
return vo;
|
||||
}
|
||||
long expTime = exp * 60L * 1000L;
|
||||
long time = now - accessTime;
|
||||
vo.setTicketExp(time > expTime);
|
||||
return vo;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,40 @@
|
||||
package io.dataease.share.server;
|
||||
|
||||
import io.dataease.api.xpack.share.ShareTicketApi;
|
||||
import io.dataease.api.xpack.share.request.TicketCreator;
|
||||
import io.dataease.api.xpack.share.request.TicketDelRequest;
|
||||
import io.dataease.api.xpack.share.request.TicketSwitchRequest;
|
||||
import io.dataease.api.xpack.share.vo.TicketVO;
|
||||
import io.dataease.share.manage.ShareTicketManage;
|
||||
import jakarta.annotation.Resource;
|
||||
import org.springframework.web.bind.annotation.RequestMapping;
|
||||
import org.springframework.web.bind.annotation.RestController;
|
||||
import java.util.List;
|
||||
|
||||
@RestController
|
||||
@RequestMapping("/ticket")
|
||||
public class ShareTicketServer implements ShareTicketApi {
|
||||
|
||||
@Resource
|
||||
private ShareTicketManage shareTicketManage;
|
||||
|
||||
@Override
|
||||
public String saveTicket(TicketCreator creator) {
|
||||
return shareTicketManage.saveTicket(creator);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void deleteTicket(TicketDelRequest request) {
|
||||
shareTicketManage.deleteTicket(request);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void switchRequire(TicketSwitchRequest request) {
|
||||
shareTicketManage.switchRequire(request);
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<TicketVO> query(Long resourceId) {
|
||||
return shareTicketManage.query(resourceId);
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user