Merge remote-tracking branch 'upstream/feature/1.7.1' into feature/1.7.1

# Conflicts:
#	neutrino-proxy-admin/src/views/proxy/portMapping.vue
#	neutrino-proxy-server/src/main/java/fun/asgc/neutrino/proxy/server/service/PortPoolService.java
This commit is contained in:
Yohanes
2023-03-20 10:11:47 +08:00
139 changed files with 6124 additions and 434 deletions
@@ -45,7 +45,7 @@ import java.util.List;
@Slf4j
@Component
public class DBInitialize implements EventListener<AppLoadEndEvent> {
private static List<String> initDataTableNameList = Lists.newArrayList("user", "license", "port_pool", "port_mapping", "job_info");
private static List<String> initDataTableNameList = Lists.newArrayList("user", "license", "port_group", "port_pool", "port_mapping", "job_info");
@Inject
private DbConfig dbConfig;
@@ -0,0 +1,12 @@
package fun.asgc.neutrino.proxy.server.constant;
/**
* @author: aoshiguchen
* @date: 2023/3/18
*/
public interface Constants {
/**
* 默认的端口分组ID
*/
int DEFAULT_PORT_GROUP_ID = 1;
}
@@ -57,6 +57,10 @@ public enum ExceptionConstant {
// 调度管理(15000)
JOB_INFO_NOT_EXIST(15000, "调度管理记录不存在"),
SYSTEM_ERROR(500, "系统异常"),
PORT_GROUP_NAME_ALREADY_EXIST(16000,"端口分组名称[{}]已经存在"),
PORT_GROUP_NAME_DOES_NOT_EXIST(16001,"端口分组不存在"),
DEFAULT_GROUP_FORBID_DELETE(16002,"默认分组禁止删除")
;
private int code;
@@ -0,0 +1,73 @@
package fun.asgc.neutrino.proxy.server.controller;
import fun.asgc.neutrino.proxy.server.base.page.PageInfo;
import fun.asgc.neutrino.proxy.server.base.page.PageQuery;
import fun.asgc.neutrino.proxy.server.base.rest.Authorization;
import fun.asgc.neutrino.proxy.server.controller.req.*;
import fun.asgc.neutrino.proxy.server.controller.res.*;
import fun.asgc.neutrino.proxy.server.service.PortGroupService;
import fun.asgc.neutrino.proxy.server.service.PortPoolService;
import fun.asgc.neutrino.proxy.server.util.ParamCheckUtil;
import org.apache.commons.lang3.StringUtils;
import org.noear.solon.annotation.*;
import java.util.List;
/**
* 端口分组控制层
*/
@Mapping("/port-group")
@Controller
public class PortGroupController {
@Inject
private PortGroupService portGroupService;
@Post
@Mapping("/create")
public PortGroupCreateRes create(PortGroupCreateReq req) {
ParamCheckUtil.checkNotNull(req, "req");
ParamCheckUtil.checkNotEmpty(req.getName(), "name");
ParamCheckUtil.checkNotNull(req.getPossessorType(), "possessorType");
ParamCheckUtil.checkNotNull(req.getPossessorId(), "possessorId");
return portGroupService.create(req);
}
@Get
@Mapping("/page")
public PageInfo<PortGroupListRes> page(PageQuery pageQuery, PortGroupListReq req) {
ParamCheckUtil.checkNotNull(pageQuery, "pageQuery");
return portGroupService.page(pageQuery, req);
}
@Get
@Mapping("/list")
public List<PortGroupListRes> list(PortGroupListReq req) {
return portGroupService.list(req);
}
@Post
@Mapping("/update/enable-status")
public PortGroupUpdateEnableStatusRes updateEnableStatus(PortGroupUpdateEnableStatusReq req) {
ParamCheckUtil.checkNotNull(req, "req");
ParamCheckUtil.checkNotNull(req.getId(), "id");
ParamCheckUtil.checkNotNull(req.getEnable(), "enable");
return portGroupService.updateEnableStatus(req);
}
@Post
@Mapping("/delete")
@Authorization(onlyAdmin = true)
public void delete(PortGroupDeleteReq req) {
ParamCheckUtil.checkNotNull(req, "req");
ParamCheckUtil.checkNotNull(req.getId(), "id");
portGroupService.delete(req.getId());
}
}
@@ -24,13 +24,8 @@ package fun.asgc.neutrino.proxy.server.controller;
import fun.asgc.neutrino.proxy.server.base.page.PageInfo;
import fun.asgc.neutrino.proxy.server.base.page.PageQuery;
import fun.asgc.neutrino.proxy.server.base.rest.Authorization;
import fun.asgc.neutrino.proxy.server.controller.req.PortPoolCreateReq;
import fun.asgc.neutrino.proxy.server.controller.req.PortPoolDeleteReq;
import fun.asgc.neutrino.proxy.server.controller.req.PortPoolListReq;
import fun.asgc.neutrino.proxy.server.controller.req.PortPoolUpdateEnableStatusReq;
import fun.asgc.neutrino.proxy.server.controller.res.PortPoolCreateRes;
import fun.asgc.neutrino.proxy.server.controller.res.PortPoolListRes;
import fun.asgc.neutrino.proxy.server.controller.res.PortPoolUpdateEnableStatusRes;
import fun.asgc.neutrino.proxy.server.controller.req.*;
import fun.asgc.neutrino.proxy.server.controller.res.*;
import fun.asgc.neutrino.proxy.server.service.PortPoolService;
import fun.asgc.neutrino.proxy.server.util.ParamCheckUtil;
import org.noear.solon.annotation.*;
@@ -68,6 +63,7 @@ public class PortPoolController {
public PortPoolCreateRes create(PortPoolCreateReq req) {
ParamCheckUtil.checkNotNull(req, "req");
ParamCheckUtil.checkNotNull(req.getPort(), "port");
ParamCheckUtil.checkNotNull(req.getGroupId(), "groupId");
return portPoolService.create(req);
}
@@ -92,4 +88,28 @@ public class PortPoolController {
portPoolService.delete(req.getId());
}
@Get
@Mapping("/get-available-port-list")
public List<PortPoolListRes> getAvailablePortList(AvailablePortListReq req) {
ParamCheckUtil.checkNotNull(req, "req");
ParamCheckUtil.checkNotNull(req.getLicenseId(), "licenseId");
return portPoolService.getAvailablePortList(req);
}
@Get
@Mapping("/get-by-group")
public List<PortPoolListRes> portListByGroupId(String groupId) {
return portPoolService.portListByGroupId(groupId);
}
@Put
@Mapping("/update-group")
public PortPoolUpdateGroupRes updateGroup(PortPoolUpdateGroupReq req) {
ParamCheckUtil.checkNotNull(req, "req");
ParamCheckUtil.checkNotNull(req.getGroupId(), "groupId");
ParamCheckUtil.checkNotEmpty(req.getPortIdList(), "portIdList");
return portPoolService.updateGroup(req);
}
}
@@ -0,0 +1,15 @@
package fun.asgc.neutrino.proxy.server.controller.req;
import lombok.Data;
/**
* 获取可用端口请求
*/
@Data
public class AvailablePortListReq {
/**
* licenseId
*/
private Integer licenseId;
}
@@ -29,5 +29,6 @@ import lombok.Data;
*/
@Data
public class LicenseFlowReportReq {
private Integer userId;
private Integer licenseId;
}
@@ -31,4 +31,7 @@ import lombok.Data;
@Data
public class LicenseListReq {
private Integer userId;
private Integer isOnline;
private Integer enable;
}
@@ -0,0 +1,27 @@
package fun.asgc.neutrino.proxy.server.controller.req;
import lombok.Data;
/**
* 端口分组创建请求
*
*
*/
@Data
public class PortGroupCreateReq {
/**
* 分组名称
*/
private String name ;
/**
* 所有者类型 (0、全局共享 1、用户所有 2License所有)
*/
private Integer possessorType ;
/**
* 所有者id(当type为0时 固定为-1、当type为1时为用户id 、当type为2时为licenseid)
*/
private Integer possessorId ;
}
@@ -0,0 +1,11 @@
package fun.asgc.neutrino.proxy.server.controller.req;
import lombok.Data;
/**
* 删除端口分组请求
*/
@Data
public class PortGroupDeleteReq {
private Integer id;
}
@@ -0,0 +1,27 @@
package fun.asgc.neutrino.proxy.server.controller.req;
import lombok.Data;
/**
* 端口分组查询请求
*
*
*/
@Data
public class PortGroupListReq {
/**
* 分组名称
*/
private String name ;
/**
* 所有者类型 (0、全局共享 1、用户所有 2License所有)
*/
private Integer possessorType ;
/**
* 所有者id(当type为0时 固定为-1、当type为1时为用户id 、当type为2时为licenseid)
*/
private Integer possessorId ;
}
@@ -0,0 +1,13 @@
package fun.asgc.neutrino.proxy.server.controller.req;
import lombok.Data;
/**
* 修改端口分组请求
*/
@Data
public class PortGroupUpdateEnableStatusReq {
private Integer id;
private Integer enable;
}
@@ -31,4 +31,6 @@ import lombok.Data;
@Data
public class PortPoolCreateReq {
private Integer port;
private Integer groupId;
}
@@ -0,0 +1,17 @@
package fun.asgc.neutrino.proxy.server.controller.req;
import lombok.Data;
import java.util.List;
/**
* 批量修改端口分组请求
*/
@Data
public class PortPoolUpdateGroupReq {
private String groupId;
private List<Integer> portIdList;
}
@@ -29,5 +29,8 @@ import lombok.Data;
*/
@Data
public class UserFlowReportReq {
/**
* 用户ID
*/
private Integer userId;
}
@@ -0,0 +1,7 @@
package fun.asgc.neutrino.proxy.server.controller.res;
import lombok.Data;
@Data
public class AvailablePortListRes {
}
@@ -22,6 +22,7 @@
package fun.asgc.neutrino.proxy.server.controller.res;
import lombok.Data;
import lombok.experimental.Accessors;
import java.util.Date;
@@ -29,6 +30,7 @@ import java.util.Date;
* @author: aoshiguchen
* @date: 2022/12/21
*/
@Accessors(chain = true)
@Data
public class LicenseFlowReportRes {
/**
@@ -48,31 +50,27 @@ public class LicenseFlowReportRes {
*/
private String licenseName;
/**
* 写入字节数
* 上行流量字节数
*/
private Long writeBytes;
private Long upFlowBytes;
/**
* 读取字节数
* 下行流量字节数
*/
private Long readBytes;
private Long downFlowBytes;
/**
* 写入流量描述
* 总流量字节数
*/
private String writeFlowStr;
private Long totalFlowBytes;
/**
* 读取流量描述
* 上行流量描述
*/
private String readFlowStr;
private String upFlowDesc;
/**
* 流量描述
* 下行流量描述
*/
private String flowStr;
private String downFlowDesc;
/**
* 报表时间
* 总流量描述
*/
private Date date;
/**
* 创建时间
*/
private Date createTime;
private String totalFlowDesc;
}
@@ -0,0 +1,4 @@
package fun.asgc.neutrino.proxy.server.controller.res;
public class PortGroupCreateRes {
}
@@ -0,0 +1,50 @@
package fun.asgc.neutrino.proxy.server.controller.res;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableId;
import java.util.Date;
public class PortGroupListRes {
/**
* 主键
*/
@TableId(type = IdType.AUTO)
private Integer id;
/**
* 分组名称
*/
private String name;
/**
* 所有者类型 (0、全局共享 1、用户所有 2License所有)
*/
private Integer possessorType;
/**
* 所有者id(当type为0时 固定为-1、当type为1时为用户id 、当type为2时为licenseid)
*/
private Integer possessorId;
/**
* 是否启用(1、启用 2、禁用)
*/
private Integer enable;
/**
* 创建时间
*/
private Date createTime;
/**
* 更新时间
*/
private Date updateTime;
/**
* 来源
*/
private String possessor;
}
@@ -0,0 +1,7 @@
package fun.asgc.neutrino.proxy.server.controller.res;
import lombok.Data;
@Data
public class PortGroupUpdateEnableStatusRes {
}
@@ -49,4 +49,14 @@ public class PortPoolListRes {
* 更新时间
*/
private Date updateTime;
/**
* 分组类型
*/
private Integer possessorType;
/**
* 分组
*/
private String groupName;
}
@@ -0,0 +1,4 @@
package fun.asgc.neutrino.proxy.server.controller.res;
public class PortPoolUpdateGroupRes {
}
@@ -22,6 +22,7 @@
package fun.asgc.neutrino.proxy.server.controller.res;
import lombok.Data;
import lombok.experimental.Accessors;
import java.util.Date;
@@ -29,6 +30,7 @@ import java.util.Date;
* @author: aoshiguchen
* @date: 2022/12/21
*/
@Accessors(chain = true)
@Data
public class UserFlowReportRes {
/**
@@ -40,39 +42,27 @@ public class UserFlowReportRes {
*/
private String userName;
/**
* 历史写入字节数
* 上行流量字节数
*/
private Long historyWriteBytes;
private Long upFlowBytes;
/**
* 历史读取字节数
* 下行流量字节数
*/
private Long historyReadBytes;
private Long downFlowBytes;
/**
* 写入字节数
* 总流量字节数
*/
private Long writeBytes;
private Long totalFlowBytes;
/**
* 读取字节数
* 上行流量描述
*/
private Long readBytes;
private String upFlowDesc;
/**
* 写入流量描述
* 下行流量描述
*/
private String writeFlowStr;
private String downFlowDesc;
/**
* 读取流量描述
* 流量描述
*/
private String readFlowStr;
/**
* 流量描述
*/
private String flowStr;
/**
* 报表时间
*/
private Date date;
/**
* 创建时间
*/
private Date createTime;
private String totalFlowDesc;
}
@@ -0,0 +1,27 @@
package fun.asgc.neutrino.proxy.server.dal;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import fun.asgc.neutrino.proxy.server.controller.req.PortGroupListReq;
import fun.asgc.neutrino.proxy.server.controller.res.PortGroupListRes;
import fun.asgc.neutrino.proxy.server.dal.entity.PortGroupDO;
import org.apache.ibatis.annotations.Mapper;
import java.util.Date;
import java.util.List;
@Mapper
public interface PortGroupMapper extends BaseMapper<PortGroupDO> {
List<PortGroupListRes> selectPortGroupListResList(PortGroupListReq res);
default void updateEnableStatus(Integer id, Integer enable, Date now){
this.update(null, Wrappers.lambdaUpdate(PortGroupDO.class)
.eq(PortGroupDO::getId,id)
.set(PortGroupDO::getEnable,enable)
.set(PortGroupDO::getUpdateTime,now)
);
}
}
@@ -24,10 +24,15 @@ package fun.asgc.neutrino.proxy.server.dal;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import fun.asgc.neutrino.proxy.server.controller.req.PortPoolListReq;
import fun.asgc.neutrino.proxy.server.controller.res.AvailablePortListRes;
import fun.asgc.neutrino.proxy.server.controller.res.PortPoolListRes;
import fun.asgc.neutrino.proxy.server.dal.entity.PortPoolDO;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;
import java.util.Date;
import java.util.List;
/**
*
@@ -53,4 +58,15 @@ public interface PortPoolMapper extends BaseMapper<PortPoolDO> {
default PortPoolDO findById(Integer id) {
return this.selectById(id);
}
default List<PortPoolDO> getByGroupId(String groupId){
return this.selectList(
new LambdaQueryWrapper<PortPoolDO>()
.eq(PortPoolDO::getGroupId, groupId)
);
}
List<PortPoolListRes> selectResList(PortPoolListReq req);
List<PortPoolListRes> getAvailablePortList(@Param("licenseId") Integer licenseId,@Param("userId") Integer userId);
}
@@ -0,0 +1,32 @@
package fun.asgc.neutrino.proxy.server.dal;
import fun.asgc.neutrino.proxy.server.controller.req.LicenseFlowReportReq;
import fun.asgc.neutrino.proxy.server.controller.req.UserFlowReportReq;
import fun.asgc.neutrino.proxy.server.controller.res.LicenseFlowReportRes;
import fun.asgc.neutrino.proxy.server.controller.res.UserFlowReportRes;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;
import java.util.List;
/**
* @author: aoshiguchen
* @date: 2023/3/19
*/
@Mapper
public interface ReportMapper {
/**
* 基于用户维度的流量报表
* @param req
* @return
*/
List<UserFlowReportRes> userFlowReportList(@Param("req") UserFlowReportReq req);
/**
* 基于license维度的流量报表
* @param req
* @return
*/
List<LicenseFlowReportRes> licenseFLowReportList(@Param("req")LicenseFlowReportReq req);
}
@@ -0,0 +1,57 @@
package fun.asgc.neutrino.proxy.server.dal.entity;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import lombok.Data;
import lombok.ToString;
import lombok.experimental.Accessors;
import java.util.Date;
/**
* 端口组
*/
@ToString
@Accessors(chain = true)
@Data
@TableName("port_group")
public class PortGroupDO {
/**
* 主键
*/
@TableId(type = IdType.AUTO)
private Integer id;
/**
* 分组名称
*/
private String name;
/**
* 所有者类型 (0、全局共享 1、用户所有 2License所有)
*/
private Integer possessorType;
/**
* 所有者id(当type为0时 固定为-1、当type为1时为用户id 、当type为2时为licenseid)
*/
private Integer possessorId;
/**
* 是否启用(1、启用 2、禁用)
*/
private Integer enable;
/**
* 创建时间
*/
private Date createTime;
/**
* 更新时间
*/
private Date updateTime;
}
@@ -42,6 +42,12 @@ import java.util.Date;
public class PortPoolDO {
@TableId(type = IdType.AUTO)
private Integer id;
/**
* 分组id
*/
private Integer groupId;
/**
* 端口
*/
@@ -51,7 +51,10 @@ public class LicenseService implements Lifecycle {
public PageInfo<LicenseListRes> page(PageQuery pageQuery, LicenseListReq req) {
Page<LicenseListRes> result = PageHelper.startPage(pageQuery.getCurrent(), pageQuery.getSize());
List<LicenseDO> list = licenseMapper.selectList(new LambdaQueryWrapper<LicenseDO>()
.orderByAsc(LicenseDO::getId)
.eq(req.getUserId() != null, LicenseDO::getUserId, req.getUserId())
.eq(req.getIsOnline() != null, LicenseDO::getIsOnline, req.getIsOnline())
.eq(req.getEnable() != null, LicenseDO::getEnable, req.getEnable())
.orderByAsc(Arrays.asList(LicenseDO::getUserId, LicenseDO::getId))
);
List<LicenseListRes> respList = mapperFacade.mapAsList(list, LicenseListRes.class);
if (CollectionUtils.isEmpty(list)) {
@@ -0,0 +1,105 @@
package fun.asgc.neutrino.proxy.server.service;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.github.pagehelper.Page;
import com.github.pagehelper.PageHelper;
import fun.asgc.neutrino.proxy.server.base.page.PageInfo;
import fun.asgc.neutrino.proxy.server.base.page.PageQuery;
import fun.asgc.neutrino.proxy.server.base.rest.ServiceException;
import fun.asgc.neutrino.proxy.server.base.rest.SystemContextHolder;
import fun.asgc.neutrino.proxy.server.constant.Constants;
import fun.asgc.neutrino.proxy.server.constant.EnableStatusEnum;
import fun.asgc.neutrino.proxy.server.constant.ExceptionConstant;
import fun.asgc.neutrino.proxy.server.controller.req.PortGroupCreateReq;
import fun.asgc.neutrino.proxy.server.controller.req.PortGroupListReq;
import fun.asgc.neutrino.proxy.server.controller.req.PortGroupUpdateEnableStatusReq;
import fun.asgc.neutrino.proxy.server.controller.res.*;
import fun.asgc.neutrino.proxy.server.dal.PortGroupMapper;
import fun.asgc.neutrino.proxy.server.dal.PortPoolMapper;
import fun.asgc.neutrino.proxy.server.dal.entity.LicenseDO;
import fun.asgc.neutrino.proxy.server.dal.entity.PortGroupDO;
import fun.asgc.neutrino.proxy.server.dal.entity.PortMappingDO;
import fun.asgc.neutrino.proxy.server.dal.entity.PortPoolDO;
import fun.asgc.neutrino.proxy.server.util.ParamCheckUtil;
import ma.glasnost.orika.MapperFacade;
import org.apache.ibatis.solon.annotation.Db;
import org.noear.solon.annotation.Component;
import org.noear.solon.annotation.Inject;
import java.util.Date;
import java.util.List;
import java.util.Objects;
/**
* 端口分组服务
*/
@Component
public class PortGroupService {
@Inject
private MapperFacade mapperFacade;
@Db
private PortGroupMapper portGroupMapper;
@Db
private PortPoolMapper portPoolMapper;
public PortGroupCreateRes create(PortGroupCreateReq req) {
PortGroupDO portGroupDO = portGroupMapper.selectOne(Wrappers.lambdaQuery(PortGroupDO.class)
.eq(PortGroupDO::getName, req.getName()));
if (Objects.nonNull(portGroupDO)) {
throw ServiceException.create(ExceptionConstant.PORT_GROUP_NAME_ALREADY_EXIST, req.getName());
}
Date now = new Date();
portGroupDO = new PortGroupDO();
portGroupDO.setName(req.getName());
portGroupDO.setPossessorType(req.getPossessorType());
portGroupDO.setPossessorId(req.getPossessorId());
portGroupDO.setEnable(EnableStatusEnum.ENABLE.getStatus());
portGroupDO.setCreateTime(now);
portGroupDO.setUpdateTime(now);
portGroupMapper.insert(portGroupDO);
return new PortGroupCreateRes();
}
public PageInfo<PortGroupListRes> page(PageQuery pageQuery, PortGroupListReq req) {
Page<PortGroupListRes> result = PageHelper.startPage(pageQuery.getCurrent(), pageQuery.getSize());
List<PortGroupListRes> list = portGroupMapper.selectPortGroupListResList(req);
return PageInfo.of(list, result.getTotal(), pageQuery.getCurrent(), pageQuery.getSize());
}
public List<PortGroupListRes> list(PortGroupListReq req) {
List<PortGroupListRes> list = portGroupMapper.selectPortGroupListResList(req);
return list;
}
public PortGroupUpdateEnableStatusRes updateEnableStatus(PortGroupUpdateEnableStatusReq req) {
PortGroupDO portGroupDO = portGroupMapper.selectById(req.getId());
ParamCheckUtil.checkNotNull(portGroupDO, ExceptionConstant.PORT_GROUP_NAME_DOES_NOT_EXIST);
if (!SystemContextHolder.isAdmin()) {
ParamCheckUtil.checkExpression(false, ExceptionConstant.NO_PERMISSION_VISIT);
}
portGroupMapper.updateEnableStatus(req.getId(), req.getEnable(), new Date());
return new PortGroupUpdateEnableStatusRes();
}
public void delete(Integer id) {
if (id == Constants.DEFAULT_PORT_GROUP_ID) {
throw ServiceException.create(ExceptionConstant.DEFAULT_GROUP_FORBID_DELETE);
}
PortGroupDO portGroupDO = portGroupMapper.selectById(id);
//检验分组是否存在
ParamCheckUtil.checkNotNull(portGroupDO, ExceptionConstant.PORT_GROUP_NAME_DOES_NOT_EXIST);
//删除
portGroupMapper.deleteById(id);
//修改绑定此分组的端口到默认分组
portPoolMapper.update(null, Wrappers.lambdaUpdate(PortPoolDO.class)
.eq(PortPoolDO::getGroupId, portGroupDO.getId())
.set(PortPoolDO::getGroupId, Constants.DEFAULT_PORT_GROUP_ID)
.set(PortPoolDO::getUpdateTime, new Date())
);
}
}
@@ -1,16 +1,16 @@
/**
* Copyright (c) 2022 aoshiguchen
* <p>
*
* 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:
* <p>
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
* <p>
*
* 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
@@ -22,77 +22,93 @@
package fun.asgc.neutrino.proxy.server.service;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.github.pagehelper.Page;
import com.github.pagehelper.PageHelper;
import fun.asgc.neutrino.proxy.server.base.page.PageInfo;
import fun.asgc.neutrino.proxy.server.base.page.PageQuery;
import fun.asgc.neutrino.proxy.server.base.rest.ServiceException;
import fun.asgc.neutrino.proxy.server.base.rest.SystemContextHolder;
import fun.asgc.neutrino.proxy.server.constant.EnableStatusEnum;
import fun.asgc.neutrino.proxy.server.constant.ExceptionConstant;
import fun.asgc.neutrino.proxy.server.controller.req.PortPoolCreateReq;
import fun.asgc.neutrino.proxy.server.controller.req.PortPoolListReq;
import fun.asgc.neutrino.proxy.server.controller.req.PortPoolUpdateEnableStatusReq;
import fun.asgc.neutrino.proxy.server.controller.res.PortPoolCreateRes;
import fun.asgc.neutrino.proxy.server.controller.res.PortPoolListRes;
import fun.asgc.neutrino.proxy.server.controller.res.PortPoolUpdateEnableStatusRes;
import fun.asgc.neutrino.proxy.server.controller.req.*;
import fun.asgc.neutrino.proxy.server.controller.res.*;
import fun.asgc.neutrino.proxy.server.dal.PortGroupMapper;
import fun.asgc.neutrino.proxy.server.dal.PortMappingMapper;
import fun.asgc.neutrino.proxy.server.dal.PortPoolMapper;
import fun.asgc.neutrino.proxy.server.dal.entity.PortGroupDO;
import fun.asgc.neutrino.proxy.server.dal.entity.PortMappingDO;
import fun.asgc.neutrino.proxy.server.dal.entity.PortPoolDO;
import fun.asgc.neutrino.proxy.server.dal.entity.UserDO;
import fun.asgc.neutrino.proxy.server.util.ParamCheckUtil;
import ma.glasnost.orika.MapperFacade;
import org.apache.ibatis.solon.annotation.Db;
import org.noear.solon.annotation.Component;
import org.noear.solon.annotation.Inject;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;
/**
*
* @author: aoshiguchen
* @date: 2022/8/7
*/
@Component
public class PortPoolService {
@Inject
private MapperFacade mapperFacade;
@Db
private PortPoolMapper portPoolMapper;
@Inject
private VisitorChannelService visitorChannelService;
@Inject
private MapperFacade mapperFacade;
@Db
private PortPoolMapper portPoolMapper;
@Inject
private VisitorChannelService visitorChannelService;
@Db
private PortMappingMapper portMappingMapper;
private PortGroupMapper portGroupMapper;
@Db
private PortMappingMapper portMappingMapper;
public PageInfo<PortPoolListRes> page(PageQuery pageQuery, PortPoolListReq req) {
Page<PortPoolListRes> result = PageHelper.startPage(pageQuery.getCurrent(), pageQuery.getSize());
List<PortPoolDO> list = portPoolMapper.selectList(new LambdaQueryWrapper<PortPoolDO>().orderByAsc(PortPoolDO::getId));
List<PortPoolListRes> respList = mapperFacade.mapAsList(list, PortPoolListRes.class);
return PageInfo.of(respList, result.getTotal(), pageQuery.getCurrent(), pageQuery.getSize());
List<PortPoolListRes> list = portPoolMapper.selectResList(req);
return PageInfo.of(list, result.getTotal(), pageQuery.getCurrent(), pageQuery.getSize());
}
public List<PortPoolListRes> list(PortPoolListReq req) {
List<PortPoolDO> list = portPoolMapper.selectList(new LambdaQueryWrapper<PortPoolDO>().eq(PortPoolDO::getEnable, EnableStatusEnum.ENABLE.getStatus()));
return mapperFacade.mapAsList(this.filterUsedPorts(list), PortPoolListRes.class);
List<PortPoolDO> list = portPoolMapper.selectList(new LambdaQueryWrapper<PortPoolDO>()
.eq(PortPoolDO::getEnable, EnableStatusEnum.ENABLE.getStatus())
);
return mapperFacade.mapAsList(this.filterUsedPorts(list), PortPoolListRes.class);
}
private List<PortPoolDO> filterUsedPorts(List<PortPoolDO> list) {
//Gets the used ports
List<PortMappingDO> usePorts = portMappingMapper.selectList(new LambdaQueryWrapper<PortMappingDO>().orderByAsc(PortMappingDO::getId));
private List<PortPoolDO> filterUsedPorts(List<PortPoolDO> list) {
//Gets the used ports
List<PortMappingDO> usePorts = portMappingMapper.selectList(new LambdaQueryWrapper<PortMappingDO>().orderByAsc(PortMappingDO::getId));
List<Integer> serverPorts = usePorts.stream().map(item -> item.getServerPort()).collect(Collectors.toList());
List<Integer> serverPorts = usePorts.stream().map(item -> item.getServerPort()).collect(Collectors.toList());
return list.stream().filter(item -> !serverPorts.contains(item.getPort())).collect(Collectors.toList());
}
return list.stream().filter(item -> !serverPorts.contains(item.getPort())).collect(Collectors.toList());
}
public PortPoolCreateRes create(PortPoolCreateReq req) {
public PortPoolCreateRes create(PortPoolCreateReq req) {
PortPoolDO oldPortPoolDO = portPoolMapper.findByPort(req.getPort());
ParamCheckUtil.checkMustNull(oldPortPoolDO, ExceptionConstant.PORT_CANNOT_REPEAT);
PortGroupDO portGroupDO = portGroupMapper.selectById(req.getGroupId());
ParamCheckUtil.checkNotNull(portGroupDO, ExceptionConstant.PORT_GROUP_NAME_DOES_NOT_EXIST);
Date now = new Date();
portPoolMapper.insert(new PortPoolDO().setPort(req.getPort()).setEnable(EnableStatusEnum.ENABLE.getStatus()).setCreateTime(now).setUpdateTime(now));
portPoolMapper.insert(new PortPoolDO()
.setPort(req.getPort())
.setGroupId(req.getGroupId())
.setEnable(EnableStatusEnum.ENABLE.getStatus())
.setCreateTime(now)
.setUpdateTime(now)
);
// 更新visitorChannel
visitorChannelService.updateVisitorChannelByPortPool(req.getPort(), EnableStatusEnum.ENABLE.getStatus());
@@ -120,4 +136,27 @@ public class PortPoolService {
visitorChannelService.updateVisitorChannelByPortPool(portPoolDO.getPort(), EnableStatusEnum.DISABLE.getStatus());
}
public List<PortPoolListRes> portListByGroupId(String groupId) {
List<PortPoolDO> portPoolDOList = portPoolMapper.getByGroupId(groupId);
List<PortPoolListRes> portPoolListReList = mapperFacade.mapAsList(portPoolDOList, PortPoolListRes.class);
return portPoolListReList;
}
public PortPoolUpdateGroupRes updateGroup(PortPoolUpdateGroupReq req) {
PortGroupDO portGroupDO = portGroupMapper.selectById(req.getGroupId());
if (Objects.isNull(portGroupDO)) {
throw ServiceException.create(ExceptionConstant.PARAMS_INVALID);
}
portPoolMapper.update(null, Wrappers.lambdaUpdate(PortPoolDO.class)
.in(PortPoolDO::getId, req.getPortIdList())
.set(PortPoolDO::getGroupId, req.getGroupId())
.set(PortPoolDO::getUpdateTime, new Date())
);
return new PortPoolUpdateGroupRes();
}
public List<PortPoolListRes> getAvailablePortList(AvailablePortListReq req) {
UserDO user = SystemContextHolder.getUser();
return portPoolMapper.getAvailablePortList(req.getLicenseId(), user.getId());
}
}
@@ -1,13 +1,25 @@
package fun.asgc.neutrino.proxy.server.service;
import cn.hutool.core.collection.CollectionUtil;
import com.github.pagehelper.Page;
import com.github.pagehelper.PageHelper;
import com.google.common.collect.Lists;
import fun.asgc.neutrino.proxy.server.base.page.PageInfo;
import fun.asgc.neutrino.proxy.server.base.page.PageQuery;
import fun.asgc.neutrino.proxy.server.controller.req.LicenseFlowReportReq;
import fun.asgc.neutrino.proxy.server.controller.req.UserFlowReportReq;
import fun.asgc.neutrino.proxy.server.controller.res.JobLogListRes;
import fun.asgc.neutrino.proxy.server.controller.res.LicenseFlowReportRes;
import fun.asgc.neutrino.proxy.server.controller.res.UserFlowReportRes;
import fun.asgc.neutrino.proxy.server.dal.ReportMapper;
import fun.asgc.neutrino.proxy.server.util.FormatUtil;
import lombok.extern.slf4j.Slf4j;
import ma.glasnost.orika.MapperFacade;
import org.apache.ibatis.solon.annotation.Db;
import org.noear.solon.annotation.Component;
import org.noear.solon.annotation.Inject;
import java.util.List;
/**
* @author: aoshiguchen
@@ -16,6 +28,10 @@ import org.noear.solon.annotation.Component;
@Slf4j
@Component
public class ReportService {
@Inject
private MapperFacade mapperFacade;
@Db
private ReportMapper reportMapper;
/**
* 用户流量报表分页
@@ -24,8 +40,10 @@ public class ReportService {
* @return
*/
public PageInfo<UserFlowReportRes> userFlowReportPage(PageQuery pageQuery, UserFlowReportReq req) {
// TODO
return null;
Page<UserFlowReportRes> result = PageHelper.startPage(pageQuery.getCurrent(), pageQuery.getSize());
List<UserFlowReportRes> list = reportMapper.userFlowReportList(req);
fillUserFlowReport(list);
return PageInfo.of(list, result.getTotal(), pageQuery.getCurrent(), pageQuery.getSize());
}
/**
@@ -35,7 +53,43 @@ public class ReportService {
* @return
*/
public PageInfo<LicenseFlowReportRes> licenseFlowReportPage(PageQuery pageQuery, LicenseFlowReportReq req) {
// TODO
return null;
Page<LicenseFlowReportRes> result = PageHelper.startPage(pageQuery.getCurrent(), pageQuery.getSize());
List<LicenseFlowReportRes> list = reportMapper.licenseFLowReportList(req);
fillLicenseFlowReport(list);
return PageInfo.of(list, 25L, pageQuery.getCurrent(), pageQuery.getSize());
}
private void fillUserFlowReport(List<UserFlowReportRes> list) {
if (CollectionUtil.isEmpty(list)) {
return;
}
for (UserFlowReportRes item : list) {
long upFlowBytes = (null == item.getUpFlowBytes()) ? 0 : item.getUpFlowBytes();
long downFlowBytes = (null == item.getDownFlowBytes()) ? 0 : item.getDownFlowBytes();
long totalFlowBytes = upFlowBytes + downFlowBytes;
item.setUpFlowBytes(upFlowBytes);
item.setDownFlowBytes(downFlowBytes);
item.setTotalFlowBytes(totalFlowBytes);
item.setUpFlowDesc(FormatUtil.getSizeDescByByteCount(upFlowBytes));
item.setDownFlowDesc(FormatUtil.getSizeDescByByteCount(downFlowBytes));
item.setTotalFlowDesc(FormatUtil.getSizeDescByByteCount(totalFlowBytes));
}
}
private void fillLicenseFlowReport(List<LicenseFlowReportRes> list) {
if (CollectionUtil.isEmpty(list)) {
return;
}
for (LicenseFlowReportRes item : list) {
long upFlowBytes = (null == item.getUpFlowBytes()) ? 0 : item.getUpFlowBytes();
long downFlowBytes = (null == item.getDownFlowBytes()) ? 0 : item.getDownFlowBytes();
long totalFlowBytes = upFlowBytes + downFlowBytes;
item.setUpFlowBytes(upFlowBytes);
item.setDownFlowBytes(downFlowBytes);
item.setTotalFlowBytes(totalFlowBytes);
item.setUpFlowDesc(FormatUtil.getSizeDescByByteCount(upFlowBytes));
item.setDownFlowDesc(FormatUtil.getSizeDescByByteCount(downFlowBytes));
item.setTotalFlowDesc(FormatUtil.getSizeDescByByteCount(totalFlowBytes));
}
}
}
@@ -0,0 +1,49 @@
package fun.asgc.neutrino.proxy.server.util;
/**
* @author: aoshiguchen
* @date: 2023/3/19
*/
public class FormatUtil {
private static final String[] SIZE_UNINTS = {"B", "KB", "MB", "GB", "TB"};
private static final int SIZE_SYSTEM = 1024;
/**
* 根据字节数获取大小描述
* 1、小于1024字节的以B为单位
* 2、小于1024KB的以KB为单位
* 3、小于1024M的以MB为单位
* 4、小于1024G的以GB为单位
* 5、其他以TB为单位
* @param byteCount
* @return
*/
public static String getSizeDescByByteCount(long byteCount){
if(byteCount <= 0){
return "0B";
}
double res = byteCount;
int index = 0;
while (index < SIZE_UNINTS.length && res >= SIZE_SYSTEM){
res /= SIZE_SYSTEM;
index++;
}
if(index >= SIZE_UNINTS.length){
index = SIZE_UNINTS.length - 1;
res *= 1024;
}
return trimZero(String.format("%.2f", res)) + SIZE_UNINTS[index];
}
private static String trimZero(String s) {
if (s.indexOf(".") > 0) {
// 去掉多余的0
s = s.replaceAll("0+?$", "");
// 如最后一位是.则去掉
s = s.replaceAll("[.]$", "");
}
return s;
}
}