APIPark/module/system/dto/input.go

package system_dto

import (
	"net/url"
	"reflect"
)

type InputSetting struct {
	InvokeAddress *string `json:"invoke_address" key:"system.node.invoke_address"`
	SitePrefix    *string `json:"site_prefix" key:"system.setting.site_prefix"`
	OllamaAddress *string `json:"ollama_address" key:"system.ai_model.ollama_address"`
}

func (i *InputSetting) Validate() error {
	if i.InvokeAddress != nil {
		_, err := url.Parse(*i.InvokeAddress)
		if err != nil {
			return err
		}
	}

	if i.OllamaAddress != nil {
		_, err := url.Parse(*i.OllamaAddress)
		if err != nil {
			return err
		}
	}
	return nil
}

func ToKeyMap(i interface{}) map[string]string {
	result := make(map[string]string)
	val := reflect.ValueOf(i)
	typ := reflect.TypeOf(i)
	if typ.Kind() == reflect.Ptr {
		val = val.Elem()
		typ = typ.Elem()
	}
	switch typ.Kind() {
	case reflect.Struct:
		{
			for i := 0; i < typ.NumField(); i++ {
				f := typ.Field(i)
				v := val.Field(i)
				if f.Type.Kind() == reflect.Ptr {
					if v.IsNil() {
						continue
					}
					v = v.Elem()
				}

				if f.Tag.Get("key") != "" {
					result[f.Tag.Get("key")] = v.String()
				}

			}
		}
	}
	return result
}