cn.thinkinginjava.mockit.client.handler.message.CancelMockMessageHandler Maven / Gradle / Ivy

/**
 * 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.
 * 
 * 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.
 * 
 * You should have received a copy of the GNU General Public License
 * along with Mockit. If not, see .
 */

package cn.thinkinginjava.mockit.client.handler.message;

import cn.thinkinginjava.mockit.client.annotation.MessageType;
import cn.thinkinginjava.mockit.common.model.dto.CancelMockData;
import cn.thinkinginjava.mockit.common.model.dto.Message;
import cn.thinkinginjava.mockit.common.model.enums.MessageTypeEnum;
import cn.thinkinginjava.mockit.common.utils.GsonUtil;
import cn.thinkinginjava.mockit.core.transformer.CancelMockClassFileTransformer;
import cn.thinkinginjava.mockit.core.transformer.manager.MockTransformerManager;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;
import org.apache.commons.lang3.StringUtils;

import static cn.thinkinginjava.mockit.common.constant.MockConstants.SUCCESS;

/**
 * Implementation of the OptionStrategy interface for handling the cancel mock option.
 */
@MessageType(MessageTypeEnum.CANCEL_MOCK)
public class CancelMockMessageHandler implements MessageHandler {

    /**
     * Executes the option strategy with the given text.
     *
     * @param text The text containing the option information.
     * @throws Exception if an error occurs during execution.
     */
    @Override
    public void handle(ChannelHandlerContext ctx, String text) throws Exception {
        Message repMessage = GsonUtil.fromJsonToMessage(text, CancelMockData.class);
        CancelMockData cancelMockData = repMessage.getData();
        if (cancelMockData == null || StringUtils.isEmpty(cancelMockData.getClassName())) {
            return;
        }
        Class targetClass = loadTargetClass(cancelMockData.getClassName());
        CancelMockClassFileTransformer cancelTransformer = new CancelMockClassFileTransformer(cancelMockData.getClassName());
        MockTransformerManager.reduction(cancelTransformer, targetClass);
        Message resMessage = new Message<>();
        resMessage.setData(SUCCESS);
        resMessage.setRequestId(repMessage.getRequestId());
        String json = GsonUtil.toJson(resMessage);
        TextWebSocketFrame responseFrame = new TextWebSocketFrame(json);
        ctx.writeAndFlush(responseFrame);
    }

    /**
     * Loads the target class based on the stored MockData.
     *
     * @param className The class name.
     * @return The loaded target class.
     * @throws ClassNotFoundException if the target class is not found.
     */
    private Class loadTargetClass(String className) throws ClassNotFoundException {
        return Class.forName(className);
    }
}