cgo invoke params

This commit is contained in:
liwei committed 2024-12-05 22:15:12 +08:00
1 parent 4b4c03fd90
commit 63296f6950
5 files changed
+29 -36

No files matched your search

+9 -4
View File
@@ -2,11 +2,11 @@ package api
import (
"context"
"encoding/json"
"fmt"
"github.com/GopeedLab/gopeed/pkg/api/model"
"github.com/GopeedLab/gopeed/pkg/base"
"github.com/GopeedLab/gopeed/pkg/download"
"github.com/GopeedLab/gopeed/pkg/util"
"net"
"net/http"
"reflect"
@@ -345,8 +345,8 @@ func (i *Instance) Close() *model.Result[any] {
}
type Request struct {
Method string `json:"method"`
Params []string `json:"params"`
Method string `json:"method"`
Params []any `json:"params"`
}
// Invoke support dynamic call method
@@ -364,6 +364,11 @@ func Invoke(instance *Instance, request *Request) (ret any) {
in := make([]reflect.Value, numIn)
for i := 0; i < numIn; i++ {
paramType := fn.Type().In(i)
arg := args[i]
if arg == nil {
in[i] = reflect.Zero(fn.Type().In(i))
continue
}
var param reflect.Value
var paramPtr any
if paramType.Kind() == reflect.Ptr {
@@ -373,7 +378,7 @@ func Invoke(instance *Instance, request *Request) (ret any) {
param = reflect.New(paramType).Elem()
paramPtr = param.Addr().Interface()
}
if err := json.Unmarshal([]byte(args[i]), paramPtr); err != nil {
if err := util.MapToStruct(arg, paramPtr); err != nil {
panic(err)
}
in[i] = param
+14 -17
View File
@@ -10,26 +10,20 @@ import (
var instance, _ = Create(nil)
func TestInvoke(t *testing.T) {
doTestInvoke(t, "Info", []any{})
doTestInvoke(t, "GetTasks", nil)
doTestInvoke(t, "GetTasks", []any{})
doTestInvoke(t, "GetTasks", []any{nil})
doTestInvoke(t, "GetTasks", []any{&download.TaskFilter{}})
doTestInvoke(t, "Info", []any{}, model.CodeOk)
doTestInvoke(t, "Info1", []any{}, model.CodeError)
doTestInvoke(t, "CreateTask", []any{nil}, model.CodeError)
doTestInvoke(t, "GetTasks", nil, model.CodeError)
doTestInvoke(t, "GetTasks", []any{}, model.CodeError)
doTestInvoke(t, "GetTasks", []any{"{abc:123"}, model.CodeError)
doTestInvoke(t, "GetTasks", []any{&download.TaskFilter{}}, model.CodeOk)
doTestInvoke(t, "SwitchExtension", []any{"test", &model.SwitchExtension{Status: true}}, model.CodeError)
}
func doTestInvoke(t *testing.T, method string, params []any) *model.Result[any] {
reqParams := make([]string, len(params))
for i, p := range params {
bytes, err := json.Marshal(p)
if err != nil {
t.Fatal(err)
}
reqParams[i] = string(bytes)
}
func doTestInvoke(t *testing.T, method string, params []any, expectCode model.RespCode) {
result := Invoke(instance, &Request{
Method: method,
Params: reqParams,
Params: params,
})
buf, err := json.Marshal(result)
if err != nil {
@@ -40,5 +34,8 @@ func doTestInvoke(t *testing.T, method string, params []any) *model.Result[any]
if err := json.Unmarshal(buf, &res); err != nil {
t.Fatal(err)
}
return &res
if res.Code != expectCode {
t.Fatalf("Invoke method [%s] failed, expect code [%d], got [%d], msg: %s", method, expectCode, res.Code, res.Msg)
}
}