caller.go

package muses

import (
	"io/ioutil"
	"reflect"

	"github.com/yitume/muses/pkg/common"
)

// 通过反射取包里面的值
var orderCallerList = []callerAttr{
	{common.ModAppName},
	{common.ModLoggerName},
	{common.ModPromName},
	{common.ModMysqlName},
	{common.ModRedisName},
	{common.ModMongoName},
	{common.ModGinSessionName},
	{common.ModEchoSessionName},
	{common.ModTplBeegoName},
	{common.ModStatName},
	{common.ModGinName},
}

type callerAttr struct {
	Name string
}

// Container from file.
func parseFile(path string) ([]byte, error) {
	// read file to []byte
	b, err := ioutil.ReadFile(path)
	if err != nil {
		return b, err
	}
	return b, nil
}

func sortCallers(callers []common.CallerFunc) (callerSort []common.Caller, err error) {
	callerMap := make(map[string]common.Caller)
	callerSort = make([]common.Caller, 0)
	for _, caller := range callers {
		obj := caller()
		name := getCallerName(obj)
		callerMap[name] = obj
	}

	for _, value := range orderCallerList {
		caller, ok := callerMap[value.Name]
		if ok {
			// 如果存在于map，加入到排序里的caller sort
			callerSort = append(callerSort, caller)
		}
	}
	return
}

func getCallerName(caller common.Caller) string {
	return reflect.ValueOf(caller).Elem().FieldByName("Name").String()
}