pkg/client/response.go

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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 client

import (
	"io"
)

type Response interface {
	IsStream() bool
}

// UnaryResponse response from endpoint
type UnaryResponse struct {
	Data []byte
}

func (r *UnaryResponse) IsStream() bool {
	return false
}

// NewByteResponse create response contains a []byte
func NewByteResponse(data []byte) *UnaryResponse {
	return &UnaryResponse{Data: data}
}

// StreamResponse response from endpoint
type StreamResponse struct {
	Stream io.ReadCloser
}

func (r *StreamResponse) IsStream() bool {
	return true
}

// NewStreamResponse create response contains a stream
func NewStreamResponse(stream io.ReadCloser) *StreamResponse {
	return &StreamResponse{Stream: stream}
}