com.qcloud.cos.internal.ResponseHeaderHandlerChain Maven / Gradle / Ivy

/*
 * Copyright 2010-2019 Amazon.com, Inc. or its affiliates. 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.
 * A copy of the License is located at
 *
 *  http://aws.amazon.com/apache2.0
 *
 * or in the "license" file accompanying this file. This file 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.
 
 * According to cos feature, we modify some class，comment, field name, etc.
 */


package com.qcloud.cos.internal;

import java.io.InputStream;
import java.util.Arrays;
import java.util.List;

import com.qcloud.cos.http.CosHttpResponse;

/**
 * An XML response handler that can also process an arbitrary number of headers
 * in the response.
 */
public class ResponseHeaderHandlerChain  extends COSXmlResponseHandler {

    private final List> headerHandlers;
    
    public ResponseHeaderHandlerChain(Unmarshaller responseUnmarshaller, HeaderHandler... headerHandlers) {
        super(responseUnmarshaller);
        this.headerHandlers = Arrays.asList(headerHandlers);
    }

    @Override
    public CosServiceResponse handle(CosHttpResponse response) throws Exception {
        CosServiceResponse cseResponse = super.handle(response);
        
        T result = cseResponse.getResult();
        if (result != null) {
            for (HeaderHandler handler : headerHandlers) {
                handler.handle(result, response);
            }
        }
        
        return cseResponse;
    }
}