You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
1.1 KiB
43 lines
1.1 KiB
5 years ago
|
package admin
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/HFO4/cloudreve/pkg/aria2"
|
||
|
|
"github.com/HFO4/cloudreve/pkg/serializer"
|
||
|
|
"net/url"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Aria2TestService aria2连接测试服务
|
||
|
|
type Aria2TestService struct {
|
||
|
|
Server string `json:"server" binding:"required"`
|
||
|
|
Token string `json:"token"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// Test 测试aria2连接
|
||
|
|
func (service *Aria2TestService) Test() serializer.Response {
|
||
|
|
testRPC := aria2.RPCService{}
|
||
|
|
|
||
|
|
// 解析RPC服务地址
|
||
|
|
server, err := url.Parse(service.Server)
|
||
|
|
if err != nil {
|
||
|
|
return serializer.ParamErr("无法解析 aria2 RPC 服务地址, "+err.Error(), nil)
|
||
|
|
}
|
||
|
|
server.Path = "/jsonrpc"
|
||
|
|
|
||
|
|
if err := testRPC.Init(server.String(), service.Token, 5, map[string]interface{}{}); err != nil {
|
||
|
|
return serializer.ParamErr("无法初始化连接, "+err.Error(), nil)
|
||
|
|
}
|
||
|
|
|
||
|
|
defer testRPC.Caller.Close()
|
||
|
|
|
||
|
|
info, err := testRPC.Caller.GetVersion()
|
||
|
|
if err != nil {
|
||
|
|
return serializer.ParamErr("无法请求 RPC 服务, "+err.Error(), nil)
|
||
|
|
}
|
||
|
|
|
||
|
|
if info.Version == "" {
|
||
|
|
return serializer.ParamErr("RPC 服务返回非预期响应", nil)
|
||
|
|
}
|
||
|
|
|
||
|
|
return serializer.Response{Data: info.Version}
|
||
|
|
}
|