uuid.go

// Package uuid is a micro plugin for adding a trace header
package uuid

import (
	"github.com/geiqin/go-micro/config/source/cli"
	"net/http"

	"github.com/geiqin/micro/plugin"
	"github.com/google/uuid"
	"github.com/micro/cli/v2"
)

type uuidPlugin struct{}

func (u *uuidPlugin) Flags() []cli.Flag {
	return nil
}

func (u *uuidPlugin) Commands() []*cli.Command {
	return nil
}

func (u *uuidPlugin) Handler() plugin.Handler {
	return func(h http.Handler) http.Handler {
		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			if td := r.Header.Get("X-Micro-Trace-Id"); len(td) == 0 {
				r.Header.Set("X-Micro-Trace-Id", uuid.New().String())
			}
			h.ServeHTTP(w, r)
		})
	}
}

func (u *uuidPlugin) Init(ctx *cli.Context) error {
	return nil
}

func (u *uuidPlugin) String() string {
	return "uuid"
}

func New() plugin.Plugin {
	return new(uuidPlugin)
}