com.saucesubfresh.rpc.server.process.AbstractMessageProcess Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of open-rpc-server Show documentation
There is a newer version: 2.1.0
/*
 * Copyright © 2022 Open-Light-Rpc Authors. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.saucesubfresh.rpc.server.process;

import com.saucesubfresh.rpc.core.Message;
import com.saucesubfresh.rpc.core.enums.PacketType;
import com.saucesubfresh.rpc.core.enums.ResponseStatus;
import com.saucesubfresh.rpc.core.exception.UnSupportMessageException;
import com.saucesubfresh.rpc.core.grpc.proto.MessageRequest;
import com.saucesubfresh.rpc.core.transport.MessageRequestBody;
import com.saucesubfresh.rpc.core.transport.MessageResponseBody;
import com.saucesubfresh.rpc.core.utils.json.JSON;
import com.saucesubfresh.rpc.core.utils.serialize.ProtostuffUtils;
import com.saucesubfresh.rpc.server.ServerConfiguration;
import com.saucesubfresh.rpc.server.callback.ResponseWriter;
import com.saucesubfresh.rpc.server.registry.RegistryService;
import lombok.extern.slf4j.Slf4j;

/**
 * @author lijunping on 2022/3/18
 */
@Slf4j
public abstract class AbstractMessageProcess implements MessageProcess{

    private final ServerConfiguration configuration;
    private final RegistryService registryService;

    protected AbstractMessageProcess(ServerConfiguration configuration, RegistryService registryService) {
        this.configuration = configuration;
        this.registryService = registryService;
    }

    @Override
    public void process(MessageRequest request, ResponseWriter responseWriter) {
        String requestJsonBody = request.getBody();
        MessageRequestBody requestBody = JSON.parse(requestJsonBody, MessageRequestBody.class);
        Message message = requestBody.getMessage();
        PacketType command = message.getCommand();

        MessageResponseBody responseBody = new MessageResponseBody();
        responseBody.setServerId(requestBody.getServerId());
        responseBody.setRequestId(requestBody.getRequestId());

        if (!command.isInner()){
            doProcess(message, responseBody, responseWriter);
            return;
        }

        try {
            handlerInnerMessage(command, responseBody);
        } catch (Exception e) {
            log.error(e.getMessage(), e);
            responseBody.setMsg(e.getMessage());
            responseBody.setStatus(ResponseStatus.ERROR);
        } finally {
            responseWriter.write(responseBody);
        }
    }

    private void handlerInnerMessage(PacketType command, MessageResponseBody responseBody){
        switch (command){
            case REGISTER:
                registryService.register(configuration.getServerAddress(), configuration.getServerPort());
                break;
            case DEREGISTER:
                registryService.deRegister(configuration.getServerAddress(), configuration.getServerPort());
                break;
            case PING:
                responseBody.setBody(ProtostuffUtils.serialize(PacketType.PONG.name()));
                break;
            default:
                throw new UnSupportMessageException("UnSupport message packet" + command);
        }
    }

    protected abstract void doProcess(Message message, MessageResponseBody responseBody, ResponseWriter responseWriter);
}