http.go

// Copyright 2020 Asim Aslam
//
// 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
//
//     https://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.
//
// Original source: github.com/micro/go-micro/v3/api/handler/http/http.go

// Package http is a http reverse proxy handler
package http

import (
	"bytes"
	"fmt"
	"io/ioutil"
	"math/rand"
	"net/http"
	"net/http/httputil"
	"net/url"
	"strconv"
	"strings"

	"github.com/2637309949/micro/v3/service/api"
	"github.com/2637309949/micro/v3/service/api/handler"
	"github.com/2637309949/micro/v3/service/errors"
	"github.com/2637309949/micro/v3/service/registry"
	xhttp "github.com/2637309949/micro/v3/util/http"
)

const (
	Handler = "http"
)

type httpHandler struct {
	options handler.Options

	// set with different initializer
	s *api.Service
}

func (h *httpHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	service, err := h.getService(r)
	if err != nil {
		err := errors.InternalServerError("go.micro.api", err.Error())
		xhttp.WriteError(w, r, err)
		return
	}

	if len(service) == 0 {
		er := errors.InternalServerError("go.micro.api", "not found service")
		xhttp.WriteError(w, r, er)
		return
	}

	rp, err := url.Parse(service)
	if err != nil {
		err := errors.InternalServerError("go.micro.api", err.Error())
		xhttp.WriteError(w, r, err)
		return
	}

	proxy := httputil.NewSingleHostReverseProxy(rp)
	proxy.ModifyResponse = func(r1 *http.Response) error {
		if strings.HasPrefix(r1.Header.Get("Content-Type"), "application/json") {
			bodyBytes, _ := ioutil.ReadAll(r1.Body)
			if len(bodyBytes) > 0 {
				bodyBytes = xhttp.Marshal(r.Context(), bodyBytes)
				r1.Header.Set("Content-Length", strconv.Itoa(len(bodyBytes)))
			}
			r1.Body = ioutil.NopCloser(bytes.NewBuffer(bodyBytes))
		}
		return nil
	}
	proxy.ServeHTTP(w, r)
}

// getService returns the service for this request from the selector
func (h *httpHandler) getService(r *http.Request) (string, error) {
	var service *api.Service

	if v, ok := r.Context().(handler.Context); ok {
		// we were given the service
		service = v.Service()
	} else if h.options.Router != nil {
		// try get service from router
		s, err := h.options.Router.Route(r)
		if err != nil {
			return "", err
		}
		service = s
	} else {
		// we have no way of routing the request
		return "", errors.InternalServerError("go.micro.api", "no route found")
	}

	// get the nodes for this service
	var nodes []*registry.Node
	for _, srv := range service.Services {
		nodes = append(nodes, srv.Nodes...)
	}

	// select a random node
	if len(nodes) == 0 {
		return "", errors.InternalServerError("go.micro.api", "no route found")
	}
	node := nodes[rand.Int()%len(nodes)]

	return fmt.Sprintf("http://%s", node.Address), nil
}

func (h *httpHandler) String() string {
	return "http"
}

// NewHandler returns a http proxy handler
func NewHandler(opts ...handler.Option) handler.Handler {
	options := handler.NewOptions(opts...)

	return &httpHandler{
		options: options,
	}
}