mirror of
https://github.com/gitpod-io/gitpod.git
synced 2025-12-08 17:36:30 +00:00
c4f9a428aa
* [ws-daemon] Simplify session management * [ws-daemon] Remove duplicated code * [ws-daemon] Fix names for tracing
102 lines
2.8 KiB
Go
102 lines
2.8 KiB
Go
// Copyright (c) 2023 Gitpod GmbH. All rights reserved.
|
|
// Licensed under the GNU Affero General Public License (AGPL).
|
|
// See License.AGPL.txt in the project root for license information.
|
|
|
|
package controller
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"path/filepath"
|
|
"sync"
|
|
|
|
"github.com/gitpod-io/gitpod/common-go/log"
|
|
"github.com/gitpod-io/gitpod/common-go/tracing"
|
|
"github.com/opentracing/opentracing-go"
|
|
|
|
"github.com/gitpod-io/gitpod/ws-daemon/pkg/internal/session"
|
|
)
|
|
|
|
type WorkspaceFactory func(ctx context.Context, instanceID string) (ws *session.Workspace, err error)
|
|
|
|
type WorkspaceProvider struct {
|
|
Location string
|
|
hooks map[session.WorkspaceState][]session.WorkspaceLivecycleHook
|
|
workspaces sync.Map
|
|
}
|
|
|
|
func NewWorkspaceProvider(location string, hooks map[session.WorkspaceState][]session.WorkspaceLivecycleHook) *WorkspaceProvider {
|
|
return &WorkspaceProvider{
|
|
Location: location,
|
|
hooks: hooks,
|
|
}
|
|
}
|
|
|
|
func (wf *WorkspaceProvider) NewWorkspace(ctx context.Context, instanceID, location string, create WorkspaceFactory) (ws *session.Workspace, err error) {
|
|
span, ctx := opentracing.StartSpanFromContext(ctx, "WorkspaceProvider.NewWorkspace")
|
|
tracing.ApplyOWI(span, log.OWI("", "", instanceID))
|
|
defer tracing.FinishSpan(span, &err)
|
|
|
|
ws, err = create(ctx, location)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if ws.NonPersistentAttrs == nil {
|
|
ws.NonPersistentAttrs = make(map[string]interface{})
|
|
}
|
|
|
|
err = wf.runLifecycleHooks(ctx, ws, session.WorkspaceInitializing)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
wf.workspaces.Store(instanceID, ws)
|
|
|
|
return ws, nil
|
|
}
|
|
|
|
func (wf *WorkspaceProvider) Remove(ctx context.Context, instanceID string) {
|
|
span, _ := opentracing.StartSpanFromContext(ctx, "WorkspaceProvider.Remove")
|
|
tracing.ApplyOWI(span, log.OWI("", "", instanceID))
|
|
defer tracing.FinishSpan(span, nil)
|
|
|
|
wf.workspaces.Delete(instanceID)
|
|
}
|
|
|
|
func (wf *WorkspaceProvider) GetAndConnect(ctx context.Context, instanceID string) (*session.Workspace, error) {
|
|
ws, ok := wf.workspaces.Load(instanceID)
|
|
if !ok {
|
|
// if the workspace is not in memory ws-daemon probabably has been restarted
|
|
// in that case we reload it from disk
|
|
path := filepath.Join(wf.Location, fmt.Sprintf("%s.workspace.json", instanceID))
|
|
loadWs, err := session.LoadWorkspace(ctx, path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ws = loadWs
|
|
}
|
|
|
|
err := wf.runLifecycleHooks(ctx, ws.(*session.Workspace), session.WorkspaceReady)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
wf.workspaces.Store(instanceID, ws)
|
|
|
|
return ws.(*session.Workspace), nil
|
|
}
|
|
|
|
func (s *WorkspaceProvider) runLifecycleHooks(ctx context.Context, ws *session.Workspace, state session.WorkspaceState) error {
|
|
hooks := s.hooks[state]
|
|
log.WithFields(ws.OWI()).WithField("state", state).WithField("hooks", len(hooks)).Info("running lifecycle hooks")
|
|
|
|
for _, h := range hooks {
|
|
err := h(ctx, ws)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|