mirror of
https://github.com/gitpod-io/gitpod.git
synced 2025-12-08 17:36:30 +00:00
27de293182
To allow overriding the default scheduler when testing the reconciler locally. Change default to 1 hour.
81 lines
2.2 KiB
Go
81 lines
2.2 KiB
Go
// Copyright (c) 2022 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 (
|
|
"github.com/gitpod-io/gitpod/common-go/baseserver"
|
|
"github.com/gitpod-io/gitpod/common-go/log"
|
|
"github.com/gitpod-io/gitpod/usage/pkg/controller"
|
|
"github.com/gitpod-io/gitpod/usage/pkg/db"
|
|
"github.com/gitpod-io/gitpod/usage/pkg/stripe"
|
|
"github.com/spf13/cobra"
|
|
"net"
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(run())
|
|
}
|
|
|
|
func run() *cobra.Command {
|
|
var (
|
|
verbose bool
|
|
apiKeyFile string
|
|
schedule time.Duration
|
|
)
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "run",
|
|
Short: "Starts the service",
|
|
Version: Version,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
log.Init(ServiceName, Version, true, verbose)
|
|
|
|
conn, err := db.Connect(db.ConnectionParams{
|
|
User: os.Getenv("DB_USERNAME"),
|
|
Password: os.Getenv("DB_PASSWORD"),
|
|
Host: net.JoinHostPort(os.Getenv("DB_HOST"), os.Getenv("DB_PORT")),
|
|
Database: "gitpod",
|
|
})
|
|
if err != nil {
|
|
log.WithError(err).Fatal("Failed to establish database connection.")
|
|
}
|
|
|
|
err = stripe.Authenticate(apiKeyFile)
|
|
if err != nil {
|
|
log.WithError(err).Fatal("Failed to initialize stripe client.")
|
|
}
|
|
|
|
ctrl, err := controller.New(schedule, controller.NewUsageReconciler(conn))
|
|
if err != nil {
|
|
log.WithError(err).Fatal("Failed to initialize usage controller.")
|
|
}
|
|
|
|
err = ctrl.Start()
|
|
if err != nil {
|
|
log.WithError(err).Fatal("Failed to start usage controller.")
|
|
}
|
|
defer ctrl.Stop()
|
|
|
|
srv, err := baseserver.New("usage")
|
|
if err != nil {
|
|
log.WithError(err).Fatal("Failed to initialize server.")
|
|
}
|
|
|
|
err = srv.ListenAndServe()
|
|
if err != nil {
|
|
log.WithError(err).Fatal("Failed to listen and serve.")
|
|
}
|
|
},
|
|
}
|
|
|
|
cmd.Flags().BoolVar(&verbose, "verbose", false, "Toggle verbose logging (debug level)")
|
|
cmd.Flags().DurationVar(&schedule, "schedule", 1*time.Hour, "The schedule on which the reconciler should run")
|
|
cmd.Flags().StringVar(&apiKeyFile, "api-key-file", "/stripe-secret/apikeys", "Location of the stripe credentials file on disk")
|
|
|
|
return cmd
|
|
}
|