056de22edb
* Feat: retrieve nodes from data table * Feat: master node ping slave node in REST API * Feat: master send scheduled ping request * Feat: inactive nodes recover loop * Modify: remove database operations from aria2 RPC caller implementation * Feat: init aria2 client in master node * Feat: Round Robin load balancer * Feat: create and monitor aria2 task in master node * Feat: salve receive and handle heartbeat * Fix: Node ID will be 0 in download record generated in older version * Feat: sign request headers with all `X-` prefix * Feat: API call to slave node will carry meta data in headers * Feat: call slave aria2 rpc method from master * Feat: get slave aria2 task status Feat: encode slave response data using gob * Feat: aria2 callback to master node / cancel or select task to slave node * Fix: use dummy aria2 client when caller initialize failed in master node * Feat: slave aria2 status event callback / salve RPC auth * Feat: prototype for slave driven filesystem * Feat: retry for init aria2 client in master node * Feat: init request client with global options * Feat: slave receive async task from master * Fix: competition write in request header * Refactor: dependency initialize order * Feat: generic message queue implementation * Feat: message queue implementation * Feat: master waiting slave transfer result * Feat: slave transfer file in stateless policy * Feat: slave transfer file in slave policy * Feat: slave transfer file in local policy * Feat: slave transfer file in OneDrive policy * Fix: failed to initialize update checker http client * Feat: list slave nodes for dashboard * Feat: test aria2 rpc connection in slave * Feat: add and save node * Feat: add and delete node in node pool * Fix: temp file cannot be removed when aria2 task fails * Fix: delete node in admin panel * Feat: edit node and get node info * Modify: delete unused settings
71 lines
1.9 KiB
Go
71 lines
1.9 KiB
Go
package admin
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
model "github.com/cloudreve/Cloudreve/v3/models"
|
|
"net/url"
|
|
"time"
|
|
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/aria2"
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/auth"
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/request"
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/serializer"
|
|
)
|
|
|
|
// Aria2TestService aria2连接测试服务
|
|
type Aria2TestService struct {
|
|
Server string `json:"server" binding:"required"`
|
|
RPC string `json:"rpc" binding:"required"`
|
|
Secret string `json:"secret" binding:"required"`
|
|
Token string `json:"token"`
|
|
Type model.ModelType `json:"type"`
|
|
}
|
|
|
|
// Test 测试aria2连接
|
|
func (service *Aria2TestService) TestMaster() serializer.Response {
|
|
res, err := aria2.TestRPCConnection(service.RPC, service.Token, 5)
|
|
if err != nil {
|
|
return serializer.ParamErr(err.Error(), err)
|
|
}
|
|
|
|
if res.Version == "" {
|
|
return serializer.ParamErr("RPC 服务返回非预期响应", nil)
|
|
}
|
|
|
|
return serializer.Response{Data: res.Version}
|
|
}
|
|
|
|
func (service *Aria2TestService) TestSlave() serializer.Response {
|
|
slave, err := url.Parse(service.Server)
|
|
if err != nil {
|
|
return serializer.ParamErr("无法解析从机端地址，"+err.Error(), nil)
|
|
}
|
|
|
|
controller, _ := url.Parse("/api/v3/slave/ping/aria2")
|
|
|
|
// 请求正文
|
|
service.Type = model.MasterNodeType
|
|
bodyByte, _ := json.Marshal(service)
|
|
|
|
r := request.NewClient()
|
|
res, err := r.Request(
|
|
"POST",
|
|
slave.ResolveReference(controller).String(),
|
|
bytes.NewReader(bodyByte),
|
|
request.WithTimeout(time.Duration(10)*time.Second),
|
|
request.WithCredential(
|
|
auth.HMACAuth{SecretKey: []byte(service.Secret)},
|
|
int64(model.GetIntSetting("slave_api_timeout", 60)),
|
|
),
|
|
).DecodeResponse()
|
|
if err != nil {
|
|
return serializer.ParamErr("无连接到从机，"+err.Error(), nil)
|
|
}
|
|
|
|
if res.Code != 0 {
|
|
return serializer.ParamErr("成功接到从机，但是从机返回："+res.Msg, nil)
|
|
}
|
|
|
|
return serializer.Response{Data: res.Data.(string)}
|
|
}
|