gitee.com/sy_183/go-common@v1.0.5-0.20231205030221-958cfe129b47/config.v2/handler.go

package config

import "reflect"

type PostHandler interface {
	PostHandle() error
}

func postHandleStruct(v reflect.Value) error {
	t := v.Type()
	l := t.NumField()
	for i := 0; i < l; i++ {
		f := t.Field(i)
		if f.IsExported() {
			if err := postHandle(v.Field(i)); err != nil {
				return err
			}
		}
	}
	return nil
}

func postHandleList(v reflect.Value) error {
	if v.IsNil() {
		return nil
	}
	l := v.Len()
	for i := 0; i < l; i++ {
		if err := postHandle(v.Index(i)); err != nil {
			return err
		}
	}
	return nil
}

func postHandleMap(v reflect.Value) error {
	if v.IsNil() {
		return nil
	}
	iter := v.MapRange()
	for iter.Next() {
		if err := postHandle(iter.Key()); err != nil {
			return err
		}
		if err := postHandle(iter.Value()); err != nil {
			return err
		}
	}
	return nil
}

func postHandle(v reflect.Value) error {
	for v.Kind() == reflect.Ptr {
		if v.IsNil() {
			return nil
		}
		v = v.Elem()
	}

	if v.CanAddr() {
		i := v.Addr().Interface()
		if h, ok := i.(PostHandler); ok {
			if err := h.PostHandle(); err != nil {
				return err
			}
			return nil
		}
	}

	switch v.Kind() {
	case reflect.Struct:
		return postHandleStruct(v)
	case reflect.Slice, reflect.Array:
		return postHandleList(v)
	case reflect.Map:
		return postHandleMap(v)
	default:
		return nil
	}
}

func PostHandle(c any) error {
	v := reflect.ValueOf(c)
	if v.IsValid() {
		return postHandle(v)
	}
	return nil
}