mirror of
https://github.com/chrislusf/seaweedfs
synced 2025-07-26 21:42:48 +02:00
aa66852304
* add ui for maintenance * valid config loading. fix workers page. * refactor * grpc between admin and workers * add a long-running bidirectional grpc call between admin and worker * use the grpc call to heartbeat * use the grpc call to communicate * worker can remove the http client * admin uses http port + 10000 as its default grpc port * one task one package * handles connection failures gracefully with exponential backoff * grpc with insecure tls * grpc with optional tls * fix detecting tls * change time config from nano seconds to seconds * add tasks with 3 interfaces * compiles reducing hard coded * remove a couple of tasks * remove hard coded references * reduce hard coded values * remove hard coded values * remove hard coded from templ * refactor maintenance package * fix import cycle * simplify * simplify * auto register * auto register factory * auto register task types * self register types * refactor * simplify * remove one task * register ui * lazy init executor factories * use registered task types * DefaultWorkerConfig remove hard coded task types * remove more hard coded * implement get maintenance task * dynamic task configuration * "System Settings" should only have system level settings * adjust menu for tasks * ensure menu not collapsed * render job configuration well * use templ for ui of task configuration * fix ordering * fix bugs * saving duration in seconds * use value and unit for duration * Delete WORKER_REFACTORING_PLAN.md * Delete maintenance.json * Delete custom_worker_example.go * remove address from workers * remove old code from ec task * remove creating collection button * reconnect with exponential backoff * worker use security.toml * start admin server with tls info from security.toml * fix "weed admin" cli description
82 lines
2.1 KiB
Go
82 lines
2.1 KiB
Go
package balance
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/seaweedfs/seaweedfs/weed/glog"
|
|
"github.com/seaweedfs/seaweedfs/weed/worker/tasks"
|
|
"github.com/seaweedfs/seaweedfs/weed/worker/types"
|
|
)
|
|
|
|
// Task implements balance operation to redistribute volumes across volume servers
|
|
type Task struct {
|
|
*tasks.BaseTask
|
|
server string
|
|
volumeID uint32
|
|
collection string
|
|
}
|
|
|
|
// NewTask creates a new balance task instance
|
|
func NewTask(server string, volumeID uint32, collection string) *Task {
|
|
task := &Task{
|
|
BaseTask: tasks.NewBaseTask(types.TaskTypeBalance),
|
|
server: server,
|
|
volumeID: volumeID,
|
|
collection: collection,
|
|
}
|
|
return task
|
|
}
|
|
|
|
// Execute executes the balance task
|
|
func (t *Task) Execute(params types.TaskParams) error {
|
|
glog.Infof("Starting balance task for volume %d on server %s (collection: %s)", t.volumeID, t.server, t.collection)
|
|
|
|
// Simulate balance operation with progress updates
|
|
steps := []struct {
|
|
name string
|
|
duration time.Duration
|
|
progress float64
|
|
}{
|
|
{"Analyzing cluster state", 2 * time.Second, 15},
|
|
{"Identifying optimal placement", 3 * time.Second, 35},
|
|
{"Moving volume data", 6 * time.Second, 75},
|
|
{"Updating cluster metadata", 2 * time.Second, 95},
|
|
{"Verifying balance", 1 * time.Second, 100},
|
|
}
|
|
|
|
for _, step := range steps {
|
|
if t.IsCancelled() {
|
|
return fmt.Errorf("balance task cancelled")
|
|
}
|
|
|
|
glog.V(1).Infof("Balance task step: %s", step.name)
|
|
t.SetProgress(step.progress)
|
|
|
|
// Simulate work
|
|
time.Sleep(step.duration)
|
|
}
|
|
|
|
glog.Infof("Balance task completed for volume %d on server %s", t.volumeID, t.server)
|
|
return nil
|
|
}
|
|
|
|
// Validate validates the task parameters
|
|
func (t *Task) Validate(params types.TaskParams) error {
|
|
if params.VolumeID == 0 {
|
|
return fmt.Errorf("volume_id is required")
|
|
}
|
|
if params.Server == "" {
|
|
return fmt.Errorf("server is required")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// EstimateTime estimates the time needed for the task
|
|
func (t *Task) EstimateTime(params types.TaskParams) time.Duration {
|
|
// Base time for balance operation
|
|
baseTime := 35 * time.Second
|
|
|
|
// Could adjust based on volume size or cluster state
|
|
return baseTime
|
|
}
|