Compare commits
1 Commits
feature/de
...
8417f8f795
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
8417f8f795 |
@@ -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, false)
|
deployErr := h.deploy.Deploy(ctx, appToDeploy, nil)
|
||||||
if deployErr != nil {
|
if deployErr != nil {
|
||||||
h.log.Error(
|
h.log.Error(
|
||||||
"deployment failed",
|
"deployment failed",
|
||||||
@@ -354,32 +354,6 @@ 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()
|
||||||
@@ -408,22 +382,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 {
|
||||||
@@ -454,7 +428,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(),
|
||||||
|
|||||||
9
internal/handlers/export_test.go
Normal file
9
internal/handlers/export_test.go
Normal file
@@ -0,0 +1,9 @@
|
|||||||
|
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,47 +684,6 @@ 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.DefaultLogTail},
|
{"empty uses default", "", handlers.ExportedDefaultLogTail},
|
||||||
{"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.DefaultLogTail},
|
{"non-numeric uses default", "abc", handlers.ExportedDefaultLogTail},
|
||||||
{"all keyword uses default", "all", handlers.DefaultLogTail},
|
{"all keyword uses default", "all", handlers.ExportedDefaultLogTail},
|
||||||
{"negative uses default", "-1", handlers.DefaultLogTail},
|
{"negative uses default", "-1", handlers.ExportedDefaultLogTail},
|
||||||
{"zero uses default", "0", handlers.DefaultLogTail},
|
{"zero uses default", "0", handlers.ExportedDefaultLogTail},
|
||||||
{"float uses default", "1.5", handlers.DefaultLogTail},
|
{"float uses default", "1.5", handlers.ExportedDefaultLogTail},
|
||||||
{"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.SanitizeTail(tc.input)
|
got := handlers.ExportedSanitizeTail(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,7 +19,6 @@ 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,7 +68,6 @@ 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,6 +10,7 @@ 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"
|
||||||
@@ -268,7 +269,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
|
session.Options.MaxAge = -1 * int(time.Second)
|
||||||
|
|
||||||
saveErr := session.Save(request, respWriter)
|
saveErr := session.Save(request, respWriter)
|
||||||
if saveErr != nil {
|
if saveErr != nil {
|
||||||
|
|||||||
@@ -369,38 +369,3 @@ 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,8 +43,6 @@ 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.
|
||||||
@@ -207,12 +205,6 @@ 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
|
||||||
@@ -221,7 +213,6 @@ type Service struct {
|
|||||||
notify *notify.Service
|
notify *notify.Service
|
||||||
config *config.Config
|
config *config.Config
|
||||||
params *ServiceParams
|
params *ServiceParams
|
||||||
activeDeploys sync.Map // map[string]*activeDeploy - per-app active deployment tracking
|
|
||||||
appLocks sync.Map // map[string]*sync.Mutex - per-app deployment locks
|
appLocks sync.Map // map[string]*sync.Mutex - per-app deployment locks
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -283,39 +274,12 @@ 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)
|
||||||
}
|
}
|
||||||
|
|
||||||
// HasActiveDeploy returns true if there is an active deployment for the given app.
|
// Deploy deploys an 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)
|
||||||
@@ -324,80 +288,45 @@ 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(deployCtx, webhookEventID)
|
webhookEvent := svc.fetchWebhookEvent(ctx, webhookEventID)
|
||||||
|
|
||||||
// Use a background context for DB operations that must complete regardless of cancellation
|
deployment, err := svc.createDeploymentRecord(ctx, app, webhookEventID, webhookEvent)
|
||||||
bgCtx := context.WithoutCancel(deployCtx)
|
|
||||||
|
|
||||||
deployment, err := svc.createDeploymentRecord(bgCtx, app, webhookEventID, webhookEvent)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
svc.logWebhookPayload(bgCtx, deployment, webhookEvent)
|
svc.logWebhookPayload(ctx, deployment, webhookEvent)
|
||||||
|
|
||||||
err = svc.updateAppStatusBuilding(bgCtx, app)
|
err = svc.updateAppStatusBuilding(ctx, app)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
svc.notify.NotifyBuildStart(bgCtx, app, deployment)
|
svc.notify.NotifyBuildStart(ctx, 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(deployCtx, app, deployment)
|
imageID, err := svc.buildImageWithTimeout(ctx, 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(bgCtx, app, deployment)
|
svc.notify.NotifyBuildSuccess(ctx, app, deployment)
|
||||||
|
|
||||||
// Deploy phase with timeout
|
// Deploy phase with timeout
|
||||||
err = svc.deployContainerWithTimeout(deployCtx, app, deployment, imageID)
|
err = svc.deployContainerWithTimeout(ctx, 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(bgCtx, app, imageID)
|
err = svc.updateAppRunning(ctx, 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(bgCtx, app, deployment)
|
go svc.checkHealthAfterDelay(context.WithoutCancel(ctx), app, deployment)
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
@@ -534,43 +463,6 @@ 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,
|
||||||
|
|||||||
@@ -1,133 +0,0 @@
|
|||||||
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")
|
|
||||||
}
|
|
||||||
@@ -1,33 +0,0 @@
|
|||||||
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, true)
|
deployErr := svc.deploy.Deploy(deployCtx, app, &eventID)
|
||||||
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,10 +40,6 @@
|
|||||||
<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