package com.iotechn.unimall.data.open;
|
|
import cn.hutool.crypto.asymmetric.RSA;
|
import com.alibaba.fastjson.JSONObject;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.iotechn.unimall.data.domain.OpenPlatformClientDO;
|
import com.iotechn.unimall.data.domain.OpenPlatformNotifyDO;
|
import com.iotechn.unimall.data.mapper.OpenPlatformClientMapper;
|
import com.iotechn.unimall.data.mapper.OpenPlatformNotifyMapper;
|
import com.dobbinsoft.fw.support.component.open.OpenPlatformStorageStrategy;
|
import com.dobbinsoft.fw.support.component.open.model.OPClient;
|
import com.dobbinsoft.fw.support.component.open.model.OPClientPermission;
|
import com.dobbinsoft.fw.support.component.open.model.OPNotify;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Component;
|
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.util.StringUtils;
|
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
/**
|
* ClassName: OpenPlatformStorageDatabaseImpl
|
* Description: TODO 缓存
|
*
|
* @author: e-weichaozheng
|
* @date: 2021-04-25
|
*/
|
@Component
|
public class OpenPlatformStorageDatabaseImpl implements OpenPlatformStorageStrategy {
|
|
@Autowired
|
private OpenPlatformClientMapper openPlatformClientMapper;
|
|
@Autowired
|
private OpenPlatformNotifyMapper openPlatformNotifyMapper;
|
|
@Override
|
public boolean checkClientExists(String clientCode) {
|
return openPlatformClientMapper.selectCount(new QueryWrapper<OpenPlatformClientDO>().eq("code", clientCode)) > 0;
|
}
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public OPClient initClient(String clientCode, String notifyUrl, String publicKey) {
|
if (openPlatformClientMapper.selectCount(new QueryWrapper<OpenPlatformClientDO>().eq("code", clientCode)) > 0) {
|
return null;
|
}
|
RSA rsa = new RSA();
|
// 生成一对RSA密钥
|
String privateKeyBase64 = rsa.getPrivateKeyBase64();
|
String publicKeyBase64 = rsa.getPublicKeyBase64();
|
OpenPlatformClientDO openPlatformClientDO = new OpenPlatformClientDO();
|
openPlatformClientDO.setCode(clientCode);
|
openPlatformClientDO.setNotifyUrl(notifyUrl);
|
openPlatformClientDO.setPublicKey1(publicKeyBase64);
|
openPlatformClientDO.setPublicKey2(publicKey);
|
openPlatformClientDO.setPrivateKey1(privateKeyBase64);
|
openPlatformClientDO.setPermissionList("[]");
|
if (openPlatformClientMapper.insert(openPlatformClientDO) > 0) {
|
OPClient opClient = new OPClient();
|
opClient.setCode(clientCode);
|
opClient.setNotifyUrl(notifyUrl);
|
opClient.setPublicKey1(publicKeyBase64);
|
opClient.setPublicKey2(publicKey);
|
opClient.setPermissionList(new ArrayList<>());
|
openPlatformClientDO.setPrivateKey1(privateKeyBase64);
|
return opClient;
|
}
|
return null;
|
}
|
|
@Override
|
public boolean setClientNotifyUrl(String clientCode, String notifyUrl) {
|
OpenPlatformClientDO openPlatformClientDO = new OpenPlatformClientDO();
|
openPlatformClientDO.setNotifyUrl(notifyUrl);
|
return openPlatformClientMapper.update(openPlatformClientDO, new QueryWrapper<OpenPlatformClientDO>().eq("code", clientCode)) > 0;
|
}
|
|
@Override
|
public boolean setClientPublicKey(String clientCode, String publicKey) {
|
OpenPlatformClientDO openPlatformClientDO = new OpenPlatformClientDO();
|
openPlatformClientDO.setPublicKey2(publicKey);
|
return openPlatformClientMapper.update(openPlatformClientDO, new QueryWrapper<OpenPlatformClientDO>().eq("code", clientCode)) > 0;
|
}
|
|
@Override
|
public OPClient getClient(String clientCode) {
|
OpenPlatformClientDO openPlatformClientDO = openPlatformClientMapper.selectOne(new QueryWrapper<OpenPlatformClientDO>().eq("code", clientCode));
|
OPClient opClient = new OPClient();
|
opClient.setPublicKey1(openPlatformClientDO.getPublicKey1());
|
opClient.setPublicKey2(openPlatformClientDO.getPublicKey2());
|
opClient.setCode(openPlatformClientDO.getCode());
|
if (StringUtils.isEmpty(openPlatformClientDO.getPermissionList())) {
|
opClient.setPermissionList(new ArrayList<>());
|
} else {
|
List<OPClientPermission> permissionList = JSONObject.parseArray(openPlatformClientDO.getPermissionList(), OPClientPermission.class);
|
opClient.setPermissionList(permissionList);
|
}
|
return opClient;
|
}
|
|
@Override
|
public boolean setClientPermissionList(String clientCode, List<OPClientPermission> permissionList) {
|
OpenPlatformClientDO openPlatformClientDO = new OpenPlatformClientDO();
|
openPlatformClientDO.setPermissionList(JSONObject.toJSONString(permissionList));
|
return openPlatformClientMapper.update(openPlatformClientDO, new QueryWrapper<OpenPlatformClientDO>().eq("code", clientCode)) > 0;
|
}
|
|
@Override
|
public boolean checkApiPermission(String clientCode, String group, String method) {
|
OpenPlatformClientDO openPlatformClientDO = openPlatformClientMapper.selectOne(new QueryWrapper<OpenPlatformClientDO>().eq("code", clientCode));
|
List<OPClientPermission> permissionList = JSONObject.parseArray(openPlatformClientDO.getPermissionList(), OPClientPermission.class);
|
OPClientPermission permission = new OPClientPermission();
|
permission.setGroup(group);
|
permission.setMethod(method);
|
return permissionList.contains(permission);
|
}
|
|
@Override
|
public OPClient getClientWithPK(String clientCode) {
|
OpenPlatformClientDO openPlatformClientDO = openPlatformClientMapper.selectOne(new QueryWrapper<OpenPlatformClientDO>().eq("code", clientCode));
|
OPClient opClient = new OPClient();
|
opClient.setPublicKey1(openPlatformClientDO.getPublicKey1());
|
opClient.setPrivateKey1(openPlatformClientDO.getPrivateKey1());
|
opClient.setPublicKey2(openPlatformClientDO.getPublicKey2());
|
opClient.setCode(openPlatformClientDO.getCode());
|
if (StringUtils.isEmpty(openPlatformClientDO.getPermissionList())) {
|
opClient.setPermissionList(new ArrayList<>());
|
} else {
|
List<OPClientPermission> permissionList = JSONObject.parseArray(openPlatformClientDO.getPermissionList(), OPClientPermission.class);
|
opClient.setPermissionList(permissionList);
|
}
|
return opClient;
|
}
|
|
@Override
|
public String storeNotify(OPNotify opNotify) {
|
OpenPlatformNotifyDO openPlatformNotifyDO = new OpenPlatformNotifyDO();
|
openPlatformNotifyDO.setClientCode(opNotify.getClientCode());
|
openPlatformNotifyDO.setNextNotify(opNotify.getNextNotify());
|
openPlatformNotifyDO.setParams(JSONObject.toJSONString(opNotify.getParams()));
|
openPlatformNotifyDO.setStatus(opNotify.getStatus());
|
openPlatformNotifyDO.setTimes(opNotify.getTimes());
|
|
if (openPlatformNotifyMapper.insert(openPlatformNotifyDO) > 0) {
|
return openPlatformNotifyDO.getId() + "";
|
}
|
return null;
|
}
|
|
@Override
|
public List<OPNotify> getNeedNotify() {
|
QueryWrapper<OpenPlatformNotifyDO> queryWrapper = new QueryWrapper<OpenPlatformNotifyDO>().eq("status", 0).gt("next_notify", new Date());
|
Integer count = openPlatformNotifyMapper.selectCount(queryWrapper);
|
if (count == 0) {
|
return new ArrayList<>();
|
}
|
List<OpenPlatformNotifyDO> openPlatformNotifyDOS = openPlatformNotifyMapper.selectList(queryWrapper);
|
return openPlatformNotifyDOS.stream().map(item -> {
|
OPNotify opNotify = new OPNotify();
|
opNotify.setId(item.getId() + "");
|
opNotify.setStatus(item.getStatus());
|
opNotify.setTimes(item.getTimes());
|
opNotify.setNextNotify(item.getNextNotify());
|
opNotify.setClientCode(item.getClientCode());
|
opNotify.setParams(JSONObject.parseArray(item.getParams(), String.class));
|
return opNotify;
|
}).collect(Collectors.toList());
|
}
|
|
@Override
|
public boolean updateNotify(OPNotify opNotify) {
|
OpenPlatformNotifyDO openPlatformNotifyDO = new OpenPlatformNotifyDO();
|
openPlatformNotifyDO.setClientCode(opNotify.getClientCode());
|
openPlatformNotifyDO.setNextNotify(opNotify.getNextNotify());
|
openPlatformNotifyDO.setParams(opNotify.getParams() == null ? null : JSONObject.toJSONString(opNotify.getParams()));
|
openPlatformNotifyDO.setStatus(opNotify.getStatus());
|
openPlatformNotifyDO.setTimes(opNotify.getTimes());
|
openPlatformNotifyDO.setId(Long.parseLong(opNotify.getId()));
|
return openPlatformNotifyMapper.updateById(openPlatformNotifyDO) > 0;
|
}
|
|
@Override
|
public boolean customTryNotify() {
|
return false;
|
}
|
}
|
