我可以用一雙額外的眼睛來解決這個挑戰，游樂場在這里
最終目標是將函式和結構公共函式注冊到活動管理器中，并通過函式名執行它們，因此類似于：
pool := map[string]interface{
"Sample": func(ctx context.Context) error,
"Sample2": func(ctx context.Context, args ...interface{}) error,
"SampleFromStruct": func(ctx context.Context) error,
"Sample2FromStruct": func(ctx context.Context, args ...interface{}) error,
}
功能看起來像：
func Sample(ctx context.Context) error {
fmt.Println("exec Sample")
return nil
}
func Sample2(ctx context.Context, args interface{}) error {
arguments := struct {
Foo string `json:"foo"`
Bar string `json:"bar"`
}{}
b, err := json.Marshal(args)
if err != nil {
return err
}
if err := json.Unmarshal(b, &arguments); err != nil {
return err
}
fmt.Println("exec Sample2 with args", arguments)
return nil
}
// and same but with struct
type ActivityInStruct struct {
Bar string
}
func (a *ActivityInStruct) SampleInStruct(ctx context.Context) error {
fmt.Println("Value of Bar", a.Bar)
return Sample(ctx)
}
func (a *ActivityInStruct) Sample2InStruct(ctx context.Context, args interface{}) error {
fmt.Println("Value of Bar", a.Bar)
return Sample2(ctx, args)
}
說到這里，我讓它與具有以下實作的函式一起作業：
type activityManager struct {
fnStorage map[string]interface{}
}
func (lm *activityManager) Register(fn interface{}) error {
fnName := strings.Split((runtime.FuncForPC(reflect.ValueOf(fn).Pointer()).Name()), ".")
name := fnName[len(fnName)-1]
lm.fnStorage[name] = fn
return nil
}
func (lm *activityManager) Exec(ctx context.Context, fnName string, args ...interface{}) error {
fn, ok := lm.fnStorage[fnName]
if !ok {
return fmt.Errorf("activity %s not found", fnName)
}
if signatureCtx, ok := fn.(func(context.Context) error); ok {
return signatureCtx(ctx)
}
if signatureWithArgument, ok := fn.(func(context.Context, interface{}) error); ok {
return signatureWithArgument(ctx, args[0])
}
return fmt.Errorf("signature for %s not supported", fnName)
}
所以執行看起來像這樣：
func NewManager() *activityManager {
return &activityManager{
fnStorage: map[string]interface{}{},
}
}
/*...*/
ctx := context.Background()
manager := NewManager()
manager.Register(Sample)
manager.Register(Sample2)
if err := manager.Exec(ctx, "Sample"); err != nil {
fmt.Println("Sample error", err.Error())
return
}
args1 := map[string]interface{}{
"foo": "isFoo",
"bar": "isBar",
}
if err := manager.Exec(ctx, "Sample2", args1); err != nil {
fmt.Println("Sample2 error", err.Error())
return
}
但是，要注冊這樣的內容：
func (lm *activityManager) RegisterStruct(fn interface{}) error {
t := reflect.TypeOf(fn)
for i := 0; i < t.NumMethod(); i {
m := t.Method(i)
if m.IsExported() {
/*
This won't work cause m.Type are
func(*main.ActivityInStruct, context.Context, interface {}) error
func(*main.ActivityInStruct, context.Context) error
instead of
func(context.Context, interface {}) error
func(context.Context) error
*/
lm.fnStorage[m.Name] = m.Func
}
}
return nil
}
/* Register Activity from Public methods in struct */
activitiesStruct := &ActivityInStruct{
Bar: "I'm the Bar",
}
manager.RegisterStruct(activitiesStruct)
我不能讓它作業，因為反射會顯示像這樣的方法簽名，func(*main.ActivityInStruct, context.Context, interface {}) error
知道如何解決嗎？完整的游樂場在這里
uj5u.com熱心網友回復：
呼叫Value.Method獲取方法值。
func (lm *activityManager) RegisterStruct(fn interface{}) error {
v := reflect.ValueOf(fn)
t := v.Type()
for i := 0; i < t.NumMethod(); i {
m := t.Method(i)
if m.IsExported() {
lm.fnStorage[m.Name] = v.Method(i).Interface()
}
}
return nil
}
轉載請註明出處，本文鏈接：https://www.uj5u.com/net/508322.html