Compare commits
8 Commits
8417f8f795
...
feature/de
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
c5f957477f | ||
| ebcae55302 | |||
| e2ad42f0ac | |||
|
|
a80b7ac0a6 | ||
|
|
69a5a8c298 | ||
| 3f499163a7 | |||
| 07ac71974c | |||
| cdd7e3fd3a |
@@ -335,7 +335,7 @@ func (h *Handlers) HandleAppDeploy() http.HandlerFunc {
|
|||||||
deployCtx := context.WithoutCancel(request.Context())
|
deployCtx := context.WithoutCancel(request.Context())
|
||||||
|
|
||||||
go func(ctx context.Context, appToDeploy *models.App) {
|
go func(ctx context.Context, appToDeploy *models.App) {
|
||||||
deployErr := h.deploy.Deploy(ctx, appToDeploy, nil)
|
deployErr := h.deploy.Deploy(ctx, appToDeploy, nil, false)
|
||||||
if deployErr != nil {
|
if deployErr != nil {
|
||||||
h.log.Error(
|
h.log.Error(
|
||||||
"deployment failed",
|
"deployment failed",
|
||||||
@@ -354,6 +354,32 @@ func (h *Handlers) HandleAppDeploy() http.HandlerFunc {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// HandleCancelDeploy cancels an in-progress deployment for an app.
|
||||||
|
func (h *Handlers) HandleCancelDeploy() http.HandlerFunc {
|
||||||
|
return func(writer http.ResponseWriter, request *http.Request) {
|
||||||
|
appID := chi.URLParam(request, "id")
|
||||||
|
|
||||||
|
application, findErr := models.FindApp(request.Context(), h.db, appID)
|
||||||
|
if findErr != nil || application == nil {
|
||||||
|
http.NotFound(writer, request)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
cancelled := h.deploy.CancelDeploy(application.ID)
|
||||||
|
if cancelled {
|
||||||
|
h.log.Info("deployment cancelled by user", "app", application.Name)
|
||||||
|
}
|
||||||
|
|
||||||
|
http.Redirect(
|
||||||
|
writer,
|
||||||
|
request,
|
||||||
|
"/apps/"+application.ID,
|
||||||
|
http.StatusSeeOther,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// HandleAppDeployments returns the deployments history handler.
|
// HandleAppDeployments returns the deployments history handler.
|
||||||
func (h *Handlers) HandleAppDeployments() http.HandlerFunc {
|
func (h *Handlers) HandleAppDeployments() http.HandlerFunc {
|
||||||
tmpl := templates.GetParsed()
|
tmpl := templates.GetParsed()
|
||||||
@@ -382,22 +408,22 @@ func (h *Handlers) HandleAppDeployments() http.HandlerFunc {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// defaultLogTail is the default number of log lines to fetch.
|
// DefaultLogTail is the default number of log lines to fetch.
|
||||||
const defaultLogTail = "500"
|
const DefaultLogTail = "500"
|
||||||
|
|
||||||
// maxLogTail is the maximum allowed value for the tail parameter.
|
// maxLogTail is the maximum allowed value for the tail parameter.
|
||||||
const maxLogTail = 500
|
const maxLogTail = 500
|
||||||
|
|
||||||
// sanitizeTail validates and clamps the tail query parameter.
|
// SanitizeTail validates and clamps the tail query parameter.
|
||||||
// It returns a numeric string clamped to maxLogTail, or the default if invalid.
|
// It returns a numeric string clamped to maxLogTail, or the default if invalid.
|
||||||
func sanitizeTail(raw string) string {
|
func SanitizeTail(raw string) string {
|
||||||
if raw == "" {
|
if raw == "" {
|
||||||
return defaultLogTail
|
return DefaultLogTail
|
||||||
}
|
}
|
||||||
|
|
||||||
n, err := strconv.Atoi(raw)
|
n, err := strconv.Atoi(raw)
|
||||||
if err != nil || n < 1 {
|
if err != nil || n < 1 {
|
||||||
return defaultLogTail
|
return DefaultLogTail
|
||||||
}
|
}
|
||||||
|
|
||||||
if n > maxLogTail {
|
if n > maxLogTail {
|
||||||
@@ -428,7 +454,7 @@ func (h *Handlers) HandleAppLogs() http.HandlerFunc {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
tail := sanitizeTail(request.URL.Query().Get("tail"))
|
tail := SanitizeTail(request.URL.Query().Get("tail"))
|
||||||
|
|
||||||
logs, logsErr := h.docker.ContainerLogs(
|
logs, logsErr := h.docker.ContainerLogs(
|
||||||
request.Context(),
|
request.Context(),
|
||||||
|
|||||||
@@ -1,9 +0,0 @@
|
|||||||
package handlers
|
|
||||||
|
|
||||||
// ExportedSanitizeTail wraps sanitizeTail for external tests.
|
|
||||||
func ExportedSanitizeTail(input string) string {
|
|
||||||
return sanitizeTail(input)
|
|
||||||
}
|
|
||||||
|
|
||||||
// ExportedDefaultLogTail exports defaultLogTail for external tests.
|
|
||||||
const ExportedDefaultLogTail = defaultLogTail
|
|
||||||
@@ -684,6 +684,47 @@ func TestDeletePortOwnershipVerification(t *testing.T) {
|
|||||||
assert.NotNil(t, found, "port should still exist after IDOR attempt")
|
assert.NotNil(t, found, "port should still exist after IDOR attempt")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestHandleCancelDeployRedirects(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
testCtx := setupTestHandlers(t)
|
||||||
|
|
||||||
|
createdApp := createTestApp(t, testCtx, "cancel-deploy-app")
|
||||||
|
|
||||||
|
request := httptest.NewRequest(
|
||||||
|
http.MethodPost,
|
||||||
|
"/apps/"+createdApp.ID+"/deployments/cancel",
|
||||||
|
nil,
|
||||||
|
)
|
||||||
|
request = addChiURLParams(request, map[string]string{"id": createdApp.ID})
|
||||||
|
recorder := httptest.NewRecorder()
|
||||||
|
|
||||||
|
handler := testCtx.handlers.HandleCancelDeploy()
|
||||||
|
handler.ServeHTTP(recorder, request)
|
||||||
|
|
||||||
|
assert.Equal(t, http.StatusSeeOther, recorder.Code)
|
||||||
|
assert.Equal(t, "/apps/"+createdApp.ID, recorder.Header().Get("Location"))
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestHandleCancelDeployReturns404ForUnknownApp(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
testCtx := setupTestHandlers(t)
|
||||||
|
|
||||||
|
request := httptest.NewRequest(
|
||||||
|
http.MethodPost,
|
||||||
|
"/apps/nonexistent/deployments/cancel",
|
||||||
|
nil,
|
||||||
|
)
|
||||||
|
request = addChiURLParams(request, map[string]string{"id": "nonexistent"})
|
||||||
|
recorder := httptest.NewRecorder()
|
||||||
|
|
||||||
|
handler := testCtx.handlers.HandleCancelDeploy()
|
||||||
|
handler.ServeHTTP(recorder, request)
|
||||||
|
|
||||||
|
assert.Equal(t, http.StatusNotFound, recorder.Code)
|
||||||
|
}
|
||||||
|
|
||||||
func TestHandleWebhookReturns404ForUnknownSecret(t *testing.T) {
|
func TestHandleWebhookReturns404ForUnknownSecret(t *testing.T) {
|
||||||
t.Parallel()
|
t.Parallel()
|
||||||
|
|
||||||
|
|||||||
@@ -14,16 +14,16 @@ func TestSanitizeTail(t *testing.T) {
|
|||||||
input string
|
input string
|
||||||
expected string
|
expected string
|
||||||
}{
|
}{
|
||||||
{"empty uses default", "", handlers.ExportedDefaultLogTail},
|
{"empty uses default", "", handlers.DefaultLogTail},
|
||||||
{"valid small number", "50", "50"},
|
{"valid small number", "50", "50"},
|
||||||
{"valid max boundary", "500", "500"},
|
{"valid max boundary", "500", "500"},
|
||||||
{"exceeds max clamped", "501", "500"},
|
{"exceeds max clamped", "501", "500"},
|
||||||
{"very large clamped", "999999", "500"},
|
{"very large clamped", "999999", "500"},
|
||||||
{"non-numeric uses default", "abc", handlers.ExportedDefaultLogTail},
|
{"non-numeric uses default", "abc", handlers.DefaultLogTail},
|
||||||
{"all keyword uses default", "all", handlers.ExportedDefaultLogTail},
|
{"all keyword uses default", "all", handlers.DefaultLogTail},
|
||||||
{"negative uses default", "-1", handlers.ExportedDefaultLogTail},
|
{"negative uses default", "-1", handlers.DefaultLogTail},
|
||||||
{"zero uses default", "0", handlers.ExportedDefaultLogTail},
|
{"zero uses default", "0", handlers.DefaultLogTail},
|
||||||
{"float uses default", "1.5", handlers.ExportedDefaultLogTail},
|
{"float uses default", "1.5", handlers.DefaultLogTail},
|
||||||
{"one is valid", "1", "1"},
|
{"one is valid", "1", "1"},
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -31,7 +31,7 @@ func TestSanitizeTail(t *testing.T) {
|
|||||||
t.Run(tc.name, func(t *testing.T) {
|
t.Run(tc.name, func(t *testing.T) {
|
||||||
t.Parallel()
|
t.Parallel()
|
||||||
|
|
||||||
got := handlers.ExportedSanitizeTail(tc.input)
|
got := handlers.SanitizeTail(tc.input)
|
||||||
if got != tc.expected {
|
if got != tc.expected {
|
||||||
t.Errorf("sanitizeTail(%q) = %q, want %q", tc.input, got, tc.expected)
|
t.Errorf("sanitizeTail(%q) = %q, want %q", tc.input, got, tc.expected)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -19,6 +19,7 @@ const (
|
|||||||
DeploymentStatusDeploying DeploymentStatus = "deploying"
|
DeploymentStatusDeploying DeploymentStatus = "deploying"
|
||||||
DeploymentStatusSuccess DeploymentStatus = "success"
|
DeploymentStatusSuccess DeploymentStatus = "success"
|
||||||
DeploymentStatusFailed DeploymentStatus = "failed"
|
DeploymentStatusFailed DeploymentStatus = "failed"
|
||||||
|
DeploymentStatusCancelled DeploymentStatus = "cancelled"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Display constants.
|
// Display constants.
|
||||||
|
|||||||
@@ -68,6 +68,7 @@ func (s *Server) SetupRoutes() {
|
|||||||
r.Post("/apps/{id}", s.handlers.HandleAppUpdate())
|
r.Post("/apps/{id}", s.handlers.HandleAppUpdate())
|
||||||
r.Post("/apps/{id}/delete", s.handlers.HandleAppDelete())
|
r.Post("/apps/{id}/delete", s.handlers.HandleAppDelete())
|
||||||
r.Post("/apps/{id}/deploy", s.handlers.HandleAppDeploy())
|
r.Post("/apps/{id}/deploy", s.handlers.HandleAppDeploy())
|
||||||
|
r.Post("/apps/{id}/deployments/cancel", s.handlers.HandleCancelDeploy())
|
||||||
r.Get("/apps/{id}/deployments", s.handlers.HandleAppDeployments())
|
r.Get("/apps/{id}/deployments", s.handlers.HandleAppDeployments())
|
||||||
r.Get("/apps/{id}/deployments/{deploymentID}/logs", s.handlers.HandleDeploymentLogsAPI())
|
r.Get("/apps/{id}/deployments/{deploymentID}/logs", s.handlers.HandleDeploymentLogsAPI())
|
||||||
r.Get("/apps/{id}/deployments/{deploymentID}/download", s.handlers.HandleDeploymentLogDownload())
|
r.Get("/apps/{id}/deployments/{deploymentID}/download", s.handlers.HandleDeploymentLogDownload())
|
||||||
|
|||||||
@@ -10,7 +10,6 @@ import (
|
|||||||
"log/slog"
|
"log/slog"
|
||||||
"net/http"
|
"net/http"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/gorilla/sessions"
|
"github.com/gorilla/sessions"
|
||||||
"go.uber.org/fx"
|
"go.uber.org/fx"
|
||||||
@@ -269,7 +268,7 @@ func (svc *Service) DestroySession(
|
|||||||
return fmt.Errorf("failed to get session: %w", err)
|
return fmt.Errorf("failed to get session: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
session.Options.MaxAge = -1 * int(time.Second)
|
session.Options.MaxAge = -1
|
||||||
|
|
||||||
saveErr := session.Save(request, respWriter)
|
saveErr := session.Save(request, respWriter)
|
||||||
if saveErr != nil {
|
if saveErr != nil {
|
||||||
|
|||||||
@@ -369,3 +369,38 @@ func TestAuthenticate(testingT *testing.T) {
|
|||||||
assert.ErrorIs(t, err, auth.ErrInvalidCredentials)
|
assert.ErrorIs(t, err, auth.ErrInvalidCredentials)
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestDestroySessionMaxAge(testingT *testing.T) {
|
||||||
|
testingT.Parallel()
|
||||||
|
|
||||||
|
testingT.Run("sets MaxAge to exactly -1", func(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
svc, cleanup := setupTestService(t)
|
||||||
|
defer cleanup()
|
||||||
|
|
||||||
|
recorder := httptest.NewRecorder()
|
||||||
|
request := httptest.NewRequest(http.MethodGet, "/", nil)
|
||||||
|
|
||||||
|
err := svc.DestroySession(recorder, request)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
// Check the Set-Cookie header to verify MaxAge is -1 (immediate expiry).
|
||||||
|
// With MaxAge = -1, the cookie should have Max-Age=0 in the HTTP header
|
||||||
|
// (per http.Cookie semantics: negative MaxAge means delete now).
|
||||||
|
cookies := recorder.Result().Cookies()
|
||||||
|
require.NotEmpty(t, cookies, "expected a Set-Cookie header")
|
||||||
|
|
||||||
|
found := false
|
||||||
|
|
||||||
|
for _, c := range cookies {
|
||||||
|
if c.MaxAge < 0 {
|
||||||
|
found = true
|
||||||
|
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
assert.True(t, found, "expected a cookie with negative MaxAge (deletion)")
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|||||||
@@ -43,6 +43,8 @@ var (
|
|||||||
ErrContainerUnhealthy = errors.New("container unhealthy after 60 seconds")
|
ErrContainerUnhealthy = errors.New("container unhealthy after 60 seconds")
|
||||||
// ErrDeploymentInProgress indicates another deployment is already running.
|
// ErrDeploymentInProgress indicates another deployment is already running.
|
||||||
ErrDeploymentInProgress = errors.New("deployment already in progress for this app")
|
ErrDeploymentInProgress = errors.New("deployment already in progress for this app")
|
||||||
|
// ErrDeployCancelled indicates the deployment was cancelled by a newer deploy.
|
||||||
|
ErrDeployCancelled = errors.New("deployment cancelled by newer deploy")
|
||||||
// ErrBuildTimeout indicates the build phase exceeded the timeout.
|
// ErrBuildTimeout indicates the build phase exceeded the timeout.
|
||||||
ErrBuildTimeout = errors.New("build timeout exceeded")
|
ErrBuildTimeout = errors.New("build timeout exceeded")
|
||||||
// ErrDeployTimeout indicates the deploy phase exceeded the timeout.
|
// ErrDeployTimeout indicates the deploy phase exceeded the timeout.
|
||||||
@@ -205,15 +207,22 @@ type ServiceParams struct {
|
|||||||
Notify *notify.Service
|
Notify *notify.Service
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// activeDeploy tracks a running deployment so it can be cancelled.
|
||||||
|
type activeDeploy struct {
|
||||||
|
cancel context.CancelFunc
|
||||||
|
done chan struct{}
|
||||||
|
}
|
||||||
|
|
||||||
// Service provides deployment functionality.
|
// Service provides deployment functionality.
|
||||||
type Service struct {
|
type Service struct {
|
||||||
log *slog.Logger
|
log *slog.Logger
|
||||||
db *database.Database
|
db *database.Database
|
||||||
docker *docker.Client
|
docker *docker.Client
|
||||||
notify *notify.Service
|
notify *notify.Service
|
||||||
config *config.Config
|
config *config.Config
|
||||||
params *ServiceParams
|
params *ServiceParams
|
||||||
appLocks sync.Map // map[string]*sync.Mutex - per-app deployment locks
|
activeDeploys sync.Map // map[string]*activeDeploy - per-app active deployment tracking
|
||||||
|
appLocks sync.Map // map[string]*sync.Mutex - per-app deployment locks
|
||||||
}
|
}
|
||||||
|
|
||||||
// New creates a new deploy Service.
|
// New creates a new deploy Service.
|
||||||
@@ -274,12 +283,39 @@ func (svc *Service) GetLogFilePath(app *models.App, deployment *models.Deploymen
|
|||||||
return filepath.Join(svc.config.DataDir, "logs", hostname, app.Name, filename)
|
return filepath.Join(svc.config.DataDir, "logs", hostname, app.Name, filename)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Deploy deploys an app.
|
// HasActiveDeploy returns true if there is an active deployment for the given app.
|
||||||
|
func (svc *Service) HasActiveDeploy(appID string) bool {
|
||||||
|
_, ok := svc.activeDeploys.Load(appID)
|
||||||
|
|
||||||
|
return ok
|
||||||
|
}
|
||||||
|
|
||||||
|
// CancelDeploy cancels any in-progress deployment for the given app
|
||||||
|
// and waits for it to finish before returning. Returns true if a deployment
|
||||||
|
// was cancelled, false if there was nothing to cancel.
|
||||||
|
func (svc *Service) CancelDeploy(appID string) bool {
|
||||||
|
if !svc.HasActiveDeploy(appID) {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
svc.cancelActiveDeploy(appID)
|
||||||
|
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// Deploy deploys an app. If cancelExisting is true (e.g. webhook-triggered),
|
||||||
|
// any in-progress deploy for the same app will be cancelled before starting.
|
||||||
|
// If cancelExisting is false and a deploy is in progress, ErrDeploymentInProgress is returned.
|
||||||
func (svc *Service) Deploy(
|
func (svc *Service) Deploy(
|
||||||
ctx context.Context,
|
ctx context.Context,
|
||||||
app *models.App,
|
app *models.App,
|
||||||
webhookEventID *int64,
|
webhookEventID *int64,
|
||||||
|
cancelExisting bool,
|
||||||
) error {
|
) error {
|
||||||
|
if cancelExisting {
|
||||||
|
svc.cancelActiveDeploy(app.ID)
|
||||||
|
}
|
||||||
|
|
||||||
// Try to acquire per-app deployment lock
|
// Try to acquire per-app deployment lock
|
||||||
if !svc.tryLockApp(app.ID) {
|
if !svc.tryLockApp(app.ID) {
|
||||||
svc.log.Warn("deployment already in progress", "app", app.Name)
|
svc.log.Warn("deployment already in progress", "app", app.Name)
|
||||||
@@ -288,45 +324,80 @@ func (svc *Service) Deploy(
|
|||||||
}
|
}
|
||||||
defer svc.unlockApp(app.ID)
|
defer svc.unlockApp(app.ID)
|
||||||
|
|
||||||
|
// Set up cancellable context and register as active deploy
|
||||||
|
deployCtx, cancel := context.WithCancel(ctx)
|
||||||
|
done := make(chan struct{})
|
||||||
|
ad := &activeDeploy{cancel: cancel, done: done}
|
||||||
|
svc.activeDeploys.Store(app.ID, ad)
|
||||||
|
|
||||||
|
defer func() {
|
||||||
|
cancel()
|
||||||
|
close(done)
|
||||||
|
svc.activeDeploys.Delete(app.ID)
|
||||||
|
}()
|
||||||
|
|
||||||
// Fetch webhook event and create deployment record
|
// Fetch webhook event and create deployment record
|
||||||
webhookEvent := svc.fetchWebhookEvent(ctx, webhookEventID)
|
webhookEvent := svc.fetchWebhookEvent(deployCtx, webhookEventID)
|
||||||
|
|
||||||
deployment, err := svc.createDeploymentRecord(ctx, app, webhookEventID, webhookEvent)
|
// Use a background context for DB operations that must complete regardless of cancellation
|
||||||
|
bgCtx := context.WithoutCancel(deployCtx)
|
||||||
|
|
||||||
|
deployment, err := svc.createDeploymentRecord(bgCtx, app, webhookEventID, webhookEvent)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
svc.logWebhookPayload(ctx, deployment, webhookEvent)
|
svc.logWebhookPayload(bgCtx, deployment, webhookEvent)
|
||||||
|
|
||||||
err = svc.updateAppStatusBuilding(ctx, app)
|
err = svc.updateAppStatusBuilding(bgCtx, app)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
svc.notify.NotifyBuildStart(ctx, app, deployment)
|
svc.notify.NotifyBuildStart(bgCtx, app, deployment)
|
||||||
|
|
||||||
|
return svc.runBuildAndDeploy(deployCtx, bgCtx, app, deployment)
|
||||||
|
}
|
||||||
|
|
||||||
|
// runBuildAndDeploy executes the build and deploy phases, handling cancellation.
|
||||||
|
func (svc *Service) runBuildAndDeploy(
|
||||||
|
deployCtx context.Context,
|
||||||
|
bgCtx context.Context,
|
||||||
|
app *models.App,
|
||||||
|
deployment *models.Deployment,
|
||||||
|
) error {
|
||||||
// Build phase with timeout
|
// Build phase with timeout
|
||||||
imageID, err := svc.buildImageWithTimeout(ctx, app, deployment)
|
imageID, err := svc.buildImageWithTimeout(deployCtx, app, deployment)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
cancelErr := svc.checkCancelled(deployCtx, bgCtx, app, deployment)
|
||||||
|
if cancelErr != nil {
|
||||||
|
return cancelErr
|
||||||
|
}
|
||||||
|
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
svc.notify.NotifyBuildSuccess(ctx, app, deployment)
|
svc.notify.NotifyBuildSuccess(bgCtx, app, deployment)
|
||||||
|
|
||||||
// Deploy phase with timeout
|
// Deploy phase with timeout
|
||||||
err = svc.deployContainerWithTimeout(ctx, app, deployment, imageID)
|
err = svc.deployContainerWithTimeout(deployCtx, app, deployment, imageID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
cancelErr := svc.checkCancelled(deployCtx, bgCtx, app, deployment)
|
||||||
|
if cancelErr != nil {
|
||||||
|
return cancelErr
|
||||||
|
}
|
||||||
|
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
err = svc.updateAppRunning(ctx, app, imageID)
|
err = svc.updateAppRunning(bgCtx, app, imageID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
// Use context.WithoutCancel to ensure health check completes even if
|
// Use context.WithoutCancel to ensure health check completes even if
|
||||||
// the parent context is cancelled (e.g., HTTP request ends).
|
// the parent context is cancelled (e.g., HTTP request ends).
|
||||||
go svc.checkHealthAfterDelay(context.WithoutCancel(ctx), app, deployment)
|
go svc.checkHealthAfterDelay(bgCtx, app, deployment)
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
@@ -463,6 +534,43 @@ func (svc *Service) unlockApp(appID string) {
|
|||||||
svc.getAppLock(appID).Unlock()
|
svc.getAppLock(appID).Unlock()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// cancelActiveDeploy cancels any in-progress deployment for the given app
|
||||||
|
// and waits for it to finish before returning.
|
||||||
|
func (svc *Service) cancelActiveDeploy(appID string) {
|
||||||
|
val, ok := svc.activeDeploys.Load(appID)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
ad, ok := val.(*activeDeploy)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
svc.log.Info("cancelling in-progress deployment", "app_id", appID)
|
||||||
|
ad.cancel()
|
||||||
|
<-ad.done
|
||||||
|
}
|
||||||
|
|
||||||
|
// checkCancelled checks if the deploy context was cancelled (by a newer deploy)
|
||||||
|
// and if so, marks the deployment as cancelled. Returns ErrDeployCancelled or nil.
|
||||||
|
func (svc *Service) checkCancelled(
|
||||||
|
deployCtx context.Context,
|
||||||
|
bgCtx context.Context,
|
||||||
|
app *models.App,
|
||||||
|
deployment *models.Deployment,
|
||||||
|
) error {
|
||||||
|
if !errors.Is(deployCtx.Err(), context.Canceled) {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
svc.log.Info("deployment cancelled by newer deploy", "app", app.Name)
|
||||||
|
|
||||||
|
_ = deployment.MarkFinished(bgCtx, models.DeploymentStatusCancelled)
|
||||||
|
|
||||||
|
return ErrDeployCancelled
|
||||||
|
}
|
||||||
|
|
||||||
func (svc *Service) fetchWebhookEvent(
|
func (svc *Service) fetchWebhookEvent(
|
||||||
ctx context.Context,
|
ctx context.Context,
|
||||||
webhookEventID *int64,
|
webhookEventID *int64,
|
||||||
|
|||||||
133
internal/service/deploy/deploy_cancel_test.go
Normal file
133
internal/service/deploy/deploy_cancel_test.go
Normal file
@@ -0,0 +1,133 @@
|
|||||||
|
package deploy_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"log/slog"
|
||||||
|
"sync"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
|
||||||
|
"git.eeqj.de/sneak/upaas/internal/service/deploy"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestCancelActiveDeploy_NoExisting(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
svc := deploy.NewTestService(slog.Default())
|
||||||
|
|
||||||
|
// Should not panic or block when no active deploy exists
|
||||||
|
svc.CancelActiveDeploy("nonexistent-app")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCancelActiveDeploy_CancelsAndWaits(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
svc := deploy.NewTestService(slog.Default())
|
||||||
|
|
||||||
|
ctx, cancel := context.WithCancel(context.Background())
|
||||||
|
done := make(chan struct{})
|
||||||
|
|
||||||
|
svc.RegisterActiveDeploy("app-1", cancel, done)
|
||||||
|
|
||||||
|
// Simulate a running deploy that respects cancellation
|
||||||
|
var deployFinished bool
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
<-ctx.Done()
|
||||||
|
|
||||||
|
deployFinished = true
|
||||||
|
|
||||||
|
close(done)
|
||||||
|
}()
|
||||||
|
|
||||||
|
svc.CancelActiveDeploy("app-1")
|
||||||
|
assert.True(t, deployFinished, "deploy should have finished after cancellation")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCancelActiveDeploy_BlocksUntilDone(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
svc := deploy.NewTestService(slog.Default())
|
||||||
|
|
||||||
|
ctx, cancel := context.WithCancel(context.Background())
|
||||||
|
done := make(chan struct{})
|
||||||
|
|
||||||
|
svc.RegisterActiveDeploy("app-2", cancel, done)
|
||||||
|
|
||||||
|
// Simulate slow cleanup after cancellation
|
||||||
|
go func() {
|
||||||
|
<-ctx.Done()
|
||||||
|
time.Sleep(50 * time.Millisecond)
|
||||||
|
close(done)
|
||||||
|
}()
|
||||||
|
|
||||||
|
start := time.Now()
|
||||||
|
|
||||||
|
svc.CancelActiveDeploy("app-2")
|
||||||
|
|
||||||
|
elapsed := time.Since(start)
|
||||||
|
|
||||||
|
assert.GreaterOrEqual(t, elapsed, 50*time.Millisecond,
|
||||||
|
"cancelActiveDeploy should block until the deploy finishes")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestTryLockApp_PreventsConcurrent(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
svc := deploy.NewTestService(slog.Default())
|
||||||
|
|
||||||
|
assert.True(t, svc.TryLockApp("app-1"), "first lock should succeed")
|
||||||
|
assert.False(t, svc.TryLockApp("app-1"), "second lock should fail")
|
||||||
|
|
||||||
|
svc.UnlockApp("app-1")
|
||||||
|
|
||||||
|
assert.True(t, svc.TryLockApp("app-1"), "lock after unlock should succeed")
|
||||||
|
|
||||||
|
svc.UnlockApp("app-1")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCancelActiveDeploy_AllowsNewDeploy(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
svc := deploy.NewTestService(slog.Default())
|
||||||
|
|
||||||
|
// Simulate an active deploy holding the lock
|
||||||
|
ctx, cancel := context.WithCancel(context.Background())
|
||||||
|
done := make(chan struct{})
|
||||||
|
|
||||||
|
svc.RegisterActiveDeploy("app-3", cancel, done)
|
||||||
|
|
||||||
|
// Lock the app as if a deploy is in progress
|
||||||
|
assert.True(t, svc.TryLockApp("app-3"))
|
||||||
|
|
||||||
|
// Simulate deploy goroutine: release lock on cancellation
|
||||||
|
var mu sync.Mutex
|
||||||
|
|
||||||
|
released := false
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
<-ctx.Done()
|
||||||
|
|
||||||
|
svc.UnlockApp("app-3")
|
||||||
|
|
||||||
|
mu.Lock()
|
||||||
|
released = true
|
||||||
|
mu.Unlock()
|
||||||
|
|
||||||
|
close(done)
|
||||||
|
}()
|
||||||
|
|
||||||
|
// Cancel should cause the old deploy to release its lock
|
||||||
|
svc.CancelActiveDeploy("app-3")
|
||||||
|
|
||||||
|
mu.Lock()
|
||||||
|
assert.True(t, released)
|
||||||
|
mu.Unlock()
|
||||||
|
|
||||||
|
// Now a new deploy should be able to acquire the lock
|
||||||
|
assert.True(t, svc.TryLockApp("app-3"), "should be able to lock after cancellation")
|
||||||
|
|
||||||
|
svc.UnlockApp("app-3")
|
||||||
|
}
|
||||||
33
internal/service/deploy/export_test.go
Normal file
33
internal/service/deploy/export_test.go
Normal file
@@ -0,0 +1,33 @@
|
|||||||
|
package deploy
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"log/slog"
|
||||||
|
)
|
||||||
|
|
||||||
|
// NewTestService creates a Service with minimal dependencies for testing.
|
||||||
|
func NewTestService(log *slog.Logger) *Service {
|
||||||
|
return &Service{
|
||||||
|
log: log,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// CancelActiveDeploy exposes cancelActiveDeploy for testing.
|
||||||
|
func (svc *Service) CancelActiveDeploy(appID string) {
|
||||||
|
svc.cancelActiveDeploy(appID)
|
||||||
|
}
|
||||||
|
|
||||||
|
// RegisterActiveDeploy registers an active deploy for testing.
|
||||||
|
func (svc *Service) RegisterActiveDeploy(appID string, cancel context.CancelFunc, done chan struct{}) {
|
||||||
|
svc.activeDeploys.Store(appID, &activeDeploy{cancel: cancel, done: done})
|
||||||
|
}
|
||||||
|
|
||||||
|
// TryLockApp exposes tryLockApp for testing.
|
||||||
|
func (svc *Service) TryLockApp(appID string) bool {
|
||||||
|
return svc.tryLockApp(appID)
|
||||||
|
}
|
||||||
|
|
||||||
|
// UnlockApp exposes unlockApp for testing.
|
||||||
|
func (svc *Service) UnlockApp(appID string) {
|
||||||
|
svc.unlockApp(appID)
|
||||||
|
}
|
||||||
@@ -143,7 +143,7 @@ func (svc *Service) triggerDeployment(
|
|||||||
// even if the HTTP request context is cancelled.
|
// even if the HTTP request context is cancelled.
|
||||||
deployCtx := context.WithoutCancel(ctx)
|
deployCtx := context.WithoutCancel(ctx)
|
||||||
|
|
||||||
deployErr := svc.deploy.Deploy(deployCtx, app, &eventID)
|
deployErr := svc.deploy.Deploy(deployCtx, app, &eventID, true)
|
||||||
if deployErr != nil {
|
if deployErr != nil {
|
||||||
svc.log.Error("deployment failed", "error", deployErr, "app", appName)
|
svc.log.Error("deployment failed", "error", deployErr, "app", appName)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -40,6 +40,10 @@
|
|||||||
<span x-text="deploying ? 'Deploying...' : 'Deploy Now'"></span>
|
<span x-text="deploying ? 'Deploying...' : 'Deploy Now'"></span>
|
||||||
</button>
|
</button>
|
||||||
</form>
|
</form>
|
||||||
|
<form method="POST" action="/apps/{{.App.ID}}/deployments/cancel" class="inline" x-show="deploying" x-cloak x-data="confirmAction('Cancel the current deployment?')" @submit="confirm($event)">
|
||||||
|
{{ .CSRFField }}
|
||||||
|
<button type="submit" class="btn-danger">Cancel Deploy</button>
|
||||||
|
</form>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user