fix: cpuset
This commit is contained in:
@@ -62,19 +62,24 @@ func (r *DockerRuntime) Create(ctx context.Context, server *model.Server) error
|
||||
fmt.Sprintf("%s:/acc/game:rw", hostServerPath),
|
||||
}
|
||||
|
||||
hostConfig := &container.HostConfig{
|
||||
Binds: binds,
|
||||
PortBindings: portBindings,
|
||||
RestartPolicy: container.RestartPolicy{Name: "unless-stopped"},
|
||||
// NET_RAW is required for tcpdump packet capture in debug mode (DEBUG_MODE=1).
|
||||
CapAdd: []string{"NET_RAW"},
|
||||
}
|
||||
if cpuset := env.GetACCCpuset(); cpuset != "" {
|
||||
hostConfig.CpusetCpus = cpuset
|
||||
}
|
||||
|
||||
resp, err := r.client.ContainerCreate(ctx, &container.Config{
|
||||
Image: image,
|
||||
ExposedPorts: exposedPorts,
|
||||
Env: []string{
|
||||
fmt.Sprintf("SERVER_PORT=%d", server.Port),
|
||||
},
|
||||
}, &container.HostConfig{
|
||||
Binds: binds,
|
||||
PortBindings: portBindings,
|
||||
RestartPolicy: container.RestartPolicy{Name: "unless-stopped"},
|
||||
// NET_RAW is required for tcpdump packet capture in debug mode (DEBUG_MODE=1).
|
||||
CapAdd: []string{"NET_RAW"},
|
||||
}, nil, nil, server.ServiceName)
|
||||
}, hostConfig, nil, nil, server.ServiceName)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to create container %s: %v", server.ServiceName, err)
|
||||
}
|
||||
|
||||
@@ -100,4 +100,7 @@ func registerDockerProviders(c *dig.Container) {
|
||||
c.Provide(func(client *dockerclient.Client) platform.LogStreamer {
|
||||
return NewDockerLogStreamer(client)
|
||||
})
|
||||
|
||||
c.Provide(NewSystemService)
|
||||
|
||||
}
|
||||
|
||||
115
local/service/system_service.go
Normal file
115
local/service/system_service.go
Normal file
@@ -0,0 +1,115 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"acc-server-manager/local/model"
|
||||
"acc-server-manager/local/platform"
|
||||
"acc-server-manager/local/repository"
|
||||
"acc-server-manager/local/utl/env"
|
||||
"acc-server-manager/local/utl/logging"
|
||||
"context"
|
||||
"fmt"
|
||||
"io"
|
||||
|
||||
"github.com/docker/docker/api/types/container"
|
||||
dockerimage "github.com/docker/docker/api/types/image"
|
||||
dockerclient "github.com/docker/docker/client"
|
||||
)
|
||||
|
||||
type SystemService struct {
|
||||
dockerClient *dockerclient.Client
|
||||
repo *repository.ServerRepository
|
||||
runtime platform.ServerRuntime
|
||||
serverService *ServerService
|
||||
}
|
||||
|
||||
func NewSystemService(
|
||||
dockerClient *dockerclient.Client,
|
||||
repo *repository.ServerRepository,
|
||||
runtime platform.ServerRuntime,
|
||||
serverService *ServerService,
|
||||
) *SystemService {
|
||||
return &SystemService{
|
||||
dockerClient: dockerClient,
|
||||
repo: repo,
|
||||
runtime: runtime,
|
||||
serverService: serverService,
|
||||
}
|
||||
}
|
||||
|
||||
func (s *SystemService) MigrateImage(ctx context.Context, stopRunning bool) (*model.MigrationResult, error) {
|
||||
image := env.GetACCImage()
|
||||
|
||||
reader, err := s.dockerClient.ImagePull(ctx, image, dockerimage.PullOptions{})
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to pull image %s: %v", image, err)
|
||||
}
|
||||
if _, err := io.Copy(io.Discard, reader); err != nil {
|
||||
logging.Warn("Error draining image pull response for %s: %v", image, err)
|
||||
}
|
||||
reader.Close()
|
||||
logging.Info("Pulled image %s", image)
|
||||
|
||||
servers, err := s.repo.GetAll(ctx, &model.ServerFilter{})
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to list servers: %v", err)
|
||||
}
|
||||
|
||||
result := &model.MigrationResult{Image: image}
|
||||
|
||||
for _, server := range *servers {
|
||||
statusStr, _ := s.runtime.Status(ctx, server.ID)
|
||||
isRunning := statusStr == model.StatusRunning.String()
|
||||
|
||||
if isRunning && !stopRunning {
|
||||
result.Skipped = append(result.Skipped, model.ServerMigrationEntry{
|
||||
ID: server.ID.String(),
|
||||
Name: server.Name,
|
||||
WasRunning: true,
|
||||
})
|
||||
continue
|
||||
}
|
||||
|
||||
entry := model.ServerMigrationEntry{
|
||||
ID: server.ID.String(),
|
||||
Name: server.Name,
|
||||
WasRunning: isRunning,
|
||||
}
|
||||
|
||||
if isRunning {
|
||||
if _, err := s.runtime.Stop(ctx, server.ID); err != nil {
|
||||
result.Failed = append(result.Failed, model.ServerMigrationError{
|
||||
ID: server.ID.String(),
|
||||
Name: server.Name,
|
||||
Error: fmt.Sprintf("stop failed: %v", err),
|
||||
})
|
||||
continue
|
||||
}
|
||||
}
|
||||
|
||||
if server.ContainerID != "" {
|
||||
if err := s.dockerClient.ContainerRemove(ctx, server.ContainerID, container.RemoveOptions{Force: true}); err != nil {
|
||||
logging.Warn("Failed to remove container %s for server %s: %v", server.ContainerID, server.ID, err)
|
||||
}
|
||||
}
|
||||
server.ContainerID = ""
|
||||
if err := s.repo.Update(ctx, &server); err != nil {
|
||||
logging.Warn("Failed to clear ContainerID for server %s: %v", server.ID, err)
|
||||
}
|
||||
|
||||
if isRunning {
|
||||
if _, err := s.runtime.Start(ctx, server.ID); err != nil {
|
||||
result.Failed = append(result.Failed, model.ServerMigrationError{
|
||||
ID: server.ID.String(),
|
||||
Name: server.Name,
|
||||
Error: fmt.Sprintf("restart failed: %v", err),
|
||||
})
|
||||
continue
|
||||
}
|
||||
s.serverService.StartAccServerRuntime(&server)
|
||||
}
|
||||
|
||||
result.Migrated = append(result.Migrated, entry)
|
||||
}
|
||||
|
||||
return result, nil
|
||||
}
|
||||
Reference in New Issue
Block a user