作者 王勇

再次对增加队列时,增添验证

... ... @@ -5,7 +5,6 @@ import com.sunyo.wlpt.message.bus.service.domain.BusQueue;
import com.sunyo.wlpt.message.bus.service.response.ResultJson;
import com.sunyo.wlpt.message.bus.service.service.BusQueueService;
import com.sunyo.wlpt.message.bus.service.service.UserInfoService;
import com.sunyo.wlpt.message.bus.service.utils.IdUtils;
import org.springframework.web.bind.annotation.*;
import javax.annotation.Resource;
... ... @@ -117,8 +116,7 @@ public class BusQueueController {
public ResultJson insertBusQueue(@RequestBody BusQueue busQueue) throws IOException, TimeoutException
{
String message = validateBusQueue(busQueue);
// 设置id
busQueue.setId(IdUtils.generateId());
//验证通过
return message == null
? busQueueService.insertSelective(busQueue)
... ...
package com.sunyo.wlpt.message.bus.service.mapper;
import com.sunyo.wlpt.message.bus.service.domain.BusQueue;
import com.sunyo.wlpt.message.bus.service.domain.UserInfo;
import org.apache.ibatis.annotations.Mapper;import org.apache.ibatis.annotations.Param;import java.util.List;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;
import java.util.List;
/**
* @author 子诚
... ... @@ -10,6 +14,7 @@ import org.apache.ibatis.annotations.Mapper;import org.apache.ibatis.annotations
*/
@Mapper
public interface UserInfoMapper {
/**
* delete by primary key
*
... ... @@ -80,4 +85,28 @@ public interface UserInfoMapper {
* @return 用户列表
*/
List<UserInfo> selectUserExist(@Param("username") String username);
/**
* 分页查询用户信息
*
* @param userInfo 用户信息 {@link UserInfo}
* @return
*/
List<UserInfo> selectUserInfoList(UserInfo userInfo);
/**
* 校验用户信息
*
* @param userInfo 用户信息 {@link UserInfo}
* @return
*/
List<UserInfo> validateUserInfo(UserInfo userInfo);
/**
* 根据队列中的用户名,虚拟主机名称和虚拟主机id,查询用户关系
*
* @param busQueue {@link BusQueue}
* @return
*/
List<UserInfo> selectByBusQueue(BusQueue busQueue);
}
\ No newline at end of file
... ...
... ... @@ -8,6 +8,7 @@ import com.sunyo.wlpt.message.bus.service.mapper.UserInfoMapper;
import com.sunyo.wlpt.message.bus.service.rabbit.utils.RabbitUtils;
import com.sunyo.wlpt.message.bus.service.response.ResultJson;
import com.sunyo.wlpt.message.bus.service.service.BusQueueService;
import com.sunyo.wlpt.message.bus.service.utils.IdUtils;
import io.netty.util.internal.StringUtil;
import org.springframework.context.annotation.Lazy;
import org.springframework.stereotype.Service;
... ... @@ -113,6 +114,11 @@ public class BusQueueServiceImpl implements BusQueueService {
if (userInfoMapper.selectUserExist(record.getUsername()).size() == 0) {
return new ResultJson<>("400", "该用户信息,不存在");
}
if (userInfoMapper.selectByBusQueue(record).size() == 0) {
return new ResultJson<>("400", "该用户关系,不存在");
}
// 设置id
record.setId(IdUtils.generateId());
rabbitUtils.toCreateQueue(record);
return busQueueMapper.insertSelective(record) > 0
? new ResultJson<>("200", "添加消息队列,成功")
... ...
package com.sunyo.wlpt.message.bus.service.utils;
import java.security.MessageDigest;
/**
* @author 子诚
* Description:
* 时间:2020/8/7 17:33
*/
public class Md5Utils {
/***
* MD5加码 生成32位md5码
*/
public static String string2Md5(String inStr)
{
MessageDigest md5 = null;
try {
md5 = MessageDigest.getInstance("MD5");
} catch (Exception e) {
System.out.println(e.toString());
e.printStackTrace();
return "";
}
char[] charArray = inStr.toCharArray();
byte[] byteArray = new byte[charArray.length];
for (int i = 0; i < charArray.length; i++) {
byteArray[i] = (byte) charArray[i];
}
byte[] md5Bytes = md5.digest(byteArray);
StringBuffer hexValue = new StringBuffer();
for (int i = 0; i < md5Bytes.length; i++) {
int val = ((int) md5Bytes[i]) & 0xff;
if (val < 16) {
hexValue.append("0");
}
hexValue.append(Integer.toHexString(val));
}
return hexValue.toString();
}
/**
* 加密解密算法 执行一次加密,两次解密
*/
public static String convertMD5(String inStr)
{
char[] a = inStr.toCharArray();
for (int i = 0; i < a.length; i++) {
a[i] = (char) (a[i] ^ 't');
}
String s = new String(a);
return s;
}
}