基于Java探针的Mock框架.zip

/** * Mockit is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * <p> * Mockit is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * <p> * You should have received a copy of the GNU General Public License * along with Mockit. If not, see <http://www.gnu.org/licenses/>. */ package cn.thinkinginjava.mockit.admin.service.impl; import cn.thinkinginjava.mockit.admin.context.ResponseCallback; import cn.thinkinginjava.mockit.admin.mapper.MockitServiceRegistryMapper; import cn.thinkinginjava.mockit.admin.model.dto.BatchCommonDTO; import cn.thinkinginjava.mockit.admin.model.dto.MockitResult; import cn.thinkinginjava.mockit.admin.model.dto.MockitServiceRegistryDTO; import cn.thinkinginjava.mockit.admin.model.dto.Session; import cn.thinkinginjava.mockit.admin.model.entity.MockitMethodMockData; import cn.thinkinginjava.mockit.admin.model.entity.MockitServiceClass; import cn.thinkinginjava.mockit.admin.model.entity.MockitServiceMethod; import cn.thinkinginjava.mockit.admin.model.entity.MockitServiceRegistry; import cn.thinkinginjava.mockit.admin.model.enums.EnabledStatusEnum; import cn.thinkinginjava.mockit.admin.model.enums.OnlineStatusEnum; import cn.thinkinginjava.mockit.admin.model.vo.MockitServiceRegistryVO; import cn.thinkinginjava.mockit.admin.service.IMockitMethodMockDataService; import cn.thinkinginjava.mockit.admin.service.IMockitServiceClassService; import cn.thinkinginjava.mockit.admin.service.IMockitServiceMethodService; import cn.thinkinginjava.mockit.admin.service.IMockitServiceRegistryService; import cn.thinkinginjava.mockit.admin.session.SessionHolder; import cn.thinkinginjava.mockit.admin.utils.MessageUtil; import cn.thinkinginjava.mockit.common.constant.MockConstants; import cn.thinkinginjava.mockit.common.model.dto.CancelMockData; import cn.thinkinginjava.mockit.common.model.dto.Message; import cn.thinkinginjava.mockit.common.model.dto.MethodMockData; import cn.thinkinginjava.mockit.common.model.dto.MockData; import cn.thinkinginjava.mockit.common.model.enums.MessageTypeEnum; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.google.common.collect.Lists; import org.apache.commons.collections4.CollectionUtils; import org.apache.commons.collections4.MapUtils; import org.apache.commons.lang3.StringUtils; import org.apache.commons.lang3.time.DateFormatUtils; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.beans.BeanUtils; import org.springframework.stereotype.Service; import javax.annotation.Resource; import java.util.*; import java.util.concurrent.CompletableFuture; import java.util.stream.Collectors; /** * Implementation of the service for managing Mockit service registries. * This class extends the base ServiceImpl class and implements the IMockitServiceRegistryService interface. */ @Service public class MockitServiceRegistryServiceImpl extends ServiceImpl<MockitServiceRegistryMapper, MockitServiceRegistry> implements IMockitServiceRegistryService { private static final Logger logger = LoggerFactory.getLogger(MockitServiceRegistryServiceImpl.class); @Resource private IMockitServiceClassService iMockitServiceClassService; @Resource private IMockitServiceMethodService iMockitServiceMethodService; @Resource private IMockitMethodMockDataService iMockitMethodMockDataService; /** * Online method: Sets the given session as online * * @param session session */ @Override public void online(Session session) { QueryWrapper<MockitServiceRegistry> queryWrapper = getQueryWrapper(session); MockitServiceRegistry mockitServiceRegistryUpdate = new MockitServiceRegistry(); mockitServiceRegistryUpdate.setAlias(session.getAlias()); mockitServiceRegistryUpdate.setIp(session.getIp()); mockitServiceRegistryUpdate.setOnline(MockConstants.YES); mockitServiceRegistryUpdate.setEnabled(MockConstants.YES); mockitServiceRegistryUpdate.setCreateAt(new Date()); mockitServiceRegistryUpdate.setUpdateAt(new Date()); saveOrUpdate(mockitServiceRegistryUpdate, queryWrapper); } /** * Offline method: Sets the given session as offline * * @param session session */ @Override public void offline(Session session) { QueryWrapper<MockitServiceRegistry> queryWrapper = getQueryWrapper(session); MockitServiceRegistry mockitServiceRegistryUpdate = new MockitServiceRegistry(); mockitServiceRegistryUpdate.setAlias(session.getAlias()); mockitServiceRegistryUpdate.setIp(session.getIp()); mockitServiceRegistryUpdate.setOnline(MockConstants.NO); mockitServiceRegistryUpdate.setEnabled(MockConstants.NO); mockitServiceRegistryUpdate.setUpdateAt(new Date()); update(mockitServiceRegistryUpdate, queryWrapper); } /** * OfflineAll method: Sets the all sessions as offline */ @Override public void offlineAll() { QueryWrapper<MockitServiceRegistry> queryWrapper = new QueryWrapper<>(); queryWrapper.lambda().eq(MockitServiceRegistry::getOnline, MockConstants.YES); queryWrapper.lambda().eq(MockitServiceRegistry::getDeleted, MockConstants.NO); MockitServiceRegistry mockitServiceRegistryUpdate = new MockitServiceRegistry(); mockitServiceRegistryUpdate.setOnline(MockConstants.NO); mockitServiceRegistryUpdate.setUpdateAt(new Date()); update(mockitServiceRegistryUpdate, queryWrapper); } /** * AddService method：add new service * * @param mockitServiceRegistryDTO service info */ @Override public void addService(MockitServiceRegistryDTO mockitServiceRegistryDTO) { QueryWrapper<MockitServiceRegistry> queryWrapper = new QueryWrapper<>(); queryWrapper.lambda().eq(MockitServiceRegistry::getAlias, mockitServiceRegistryDTO.getAlias()); queryWrapper.lambda().eq(MockitServiceRegistry::getIp, mockitServiceRegistryDTO.getIp()); MockitServiceRegistry mockitServiceRegistry = new MockitServiceRegistry(); mockitServiceRegistry.setAlias(mockitServiceRegistryDTO.getAlias()); mockitServiceRegistry.setIp(mockitServiceRegistryDTO.getIp()); mockitServiceRegistry.setOnline(MockConstants.NO); mockitServiceRegistry.setEnabled(MockConstants.NO); mockitServiceRegistry.setDeleted(MockConstants.NO); mockitServiceRegistry.setCreateAt(new Date()); mockitServiceRegistry.setUpdateAt(new Date()); saveOrUpdate(mockitServiceRegistry, queryWrapper); } /** * DeleteService method：delete service * * @param mockitServiceRegistryDTO service info */ @Override public void deleteService(MockitServiceRegistryDTO mockitServiceRegistryDTO) { QueryWrapper<MockitServiceRegistry> queryWrapper = new QueryWrapper<>(); queryWrapper.lambda().eq(MockitServiceRegistry::getAlias, mockitServiceRegistryDTO.getAlias()); queryWrapper.lambda().eq(MockitServiceRegistry::getIp, mockitServiceRegistryDTO.getIp()); MockitServiceRegistry mockitServiceRegistry = new MockitServiceRegistry(); mockitServiceRegistry.setO