/*
 * Copyright 2017 NAVER Corp.
 *
 * 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.navercorp.pinpoint.web.cluster;

import com.navercorp.pinpoint.thrift.dto.command.TCommandTransferResponse;
import com.navercorp.pinpoint.thrift.dto.command.TRouteResult;
import com.navercorp.pinpoint.thrift.io.DeserializerFactory;
import com.navercorp.pinpoint.thrift.io.HeaderTBaseDeserializer;
import com.navercorp.pinpoint.thrift.util.SerializationUtils;
import org.apache.commons.lang3.ArrayUtils;
import org.apache.thrift.TBase;

/**
 * @Author Taejin Koo
 */
public class DefaultPinpointRouteResponse implements PinpointRouteResponse {

    private final byte[] payload;

    private TRouteResult routeResult;
    private TBase response;

    private boolean isParsed;

    public DefaultPinpointRouteResponse(byte[] payload) {
        this.payload = payload;
    }

    public void parse(DeserializerFactory<HeaderTBaseDeserializer> commandDeserializerFactory) {
        if (!isParsed) {
            if (ArrayUtils.isEmpty(payload)) {
                routeResult = TRouteResult.EMPTY_RESPONSE;
                return;
            }

            TBase object = deserialize(commandDeserializerFactory, payload, null);

            if (object == null) {
                routeResult = TRouteResult.NOT_SUPPORTED_RESPONSE;
            } else if (object instanceof  TCommandTransferResponse) {
                TCommandTransferResponse commandResponse = (TCommandTransferResponse) object;
                TRouteResult routeResult = commandResponse.getRouteResult();
                if (routeResult == null) {
                    this.routeResult = TRouteResult.UNKNOWN;
                } else {
                    this.routeResult = routeResult;
                }

                response = deserialize(commandDeserializerFactory, commandResponse.getPayload(), null);
            } else {
                routeResult = TRouteResult.UNKNOWN;
                response = object;
            }
            isParsed = true;
        }
    }

    @Override
    public TRouteResult getRouteResult() {
        assertParsed();
        return routeResult;
    }

    @Override
    public TBase getResponse() {
        assertParsed();
        return response;
    }

    @Override
    public <R extends TBase> R getResponse(Class<R> clazz) {
        TBase response = getResponse();

        if (clazz.isInstance(response)) {
            return (R) response;
        }

        throw new ClassCastException("Not expected " + clazz + " type.");
    }

    @Override
    public <R extends TBase> R getResponse(Class<R> clazz, R defaultValue) {
        TBase response = getResponse();

        if (clazz.isInstance(response)) {
            return (R) response;
        }

        return defaultValue;
    }

    private void assertParsed() {
        if (!isParsed) {
            throw new IllegalStateException("not yet parsed.");
        }
    }

    private TBase deserialize(DeserializerFactory<HeaderTBaseDeserializer> commandDeserializerFactory, byte[] objectData, TBase defaultValue) {
        return SerializationUtils.deserialize(objectData, commandDeserializerFactory, defaultValue);
    }


}