mirror of
https://github.com/gitpod-io/gitpod.git
synced 2025-12-08 17:36:30 +00:00
0f190439c2
* [installer] Setup seperate image builder deployment for mk2 * [image-builder] Route to wsman-mk2 * [wsman-mk2] Set default class * [installer] Update render tests * [ws-manager-mk2] Use image builder secret * [image-builder] Update render tests
83 lines
2.2 KiB
Go
83 lines
2.2 KiB
Go
// Copyright (c) 2021 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 cmd
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/gitpod-io/gitpod/common-go/baseserver"
|
|
|
|
"github.com/gitpod-io/gitpod/common-go/log"
|
|
"github.com/gitpod-io/gitpod/image-builder/api"
|
|
"github.com/gitpod-io/gitpod/image-builder/pkg/orchestrator"
|
|
"github.com/gitpod-io/gitpod/image-builder/pkg/resolve"
|
|
|
|
"github.com/opentracing/opentracing-go"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// runCmd represents the run command
|
|
var runCmd = &cobra.Command{
|
|
Use: "run",
|
|
Short: "Starts the image-builder service",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
cfg := getConfig()
|
|
log.WithField("config", cfg).Info("Starting image-builder-mk3")
|
|
|
|
srv, err := baseserver.New("image-builder-mk3",
|
|
baseserver.WithConfig(cfg.Server),
|
|
baseserver.WithVersion(Version),
|
|
)
|
|
if err != nil {
|
|
log.WithError(err).Fatal("Failed to setup server.")
|
|
}
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
span, ctx := opentracing.StartSpanFromContext(ctx, "/cmd/Run")
|
|
defer span.Finish()
|
|
|
|
service, err := orchestrator.NewOrchestratingBuilder(cfg.Orchestrator)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
if cfg.RefCache.Interval != "" && len(cfg.RefCache.Refs) > 0 {
|
|
interval, err := time.ParseDuration(cfg.RefCache.Interval)
|
|
if err != nil {
|
|
log.WithError(err).WithField("interval", cfg.RefCache.Interval).Fatal("interval is not a valid duration")
|
|
}
|
|
|
|
resolver := &resolve.PrecachingRefResolver{
|
|
Resolver: &resolve.StandaloneRefResolver{},
|
|
Candidates: cfg.RefCache.Refs,
|
|
}
|
|
go resolver.StartCaching(ctx, interval)
|
|
service.RefResolver = resolver
|
|
}
|
|
|
|
err = service.RegisterMetrics(srv.MetricsRegistry())
|
|
if err != nil {
|
|
log.WithError(err).Fatal("Failed to register metrics.")
|
|
}
|
|
|
|
err = service.Start(ctx)
|
|
if err != nil {
|
|
log.WithError(err).Fatal("Failed to start orchestrator service.")
|
|
}
|
|
|
|
api.RegisterImageBuilderServer(srv.GRPC(), service)
|
|
|
|
err = srv.ListenAndServe()
|
|
if err != nil {
|
|
log.WithError(err).Fatal("Failed to start server")
|
|
}
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(runCmd)
|
|
}
|