Files
my-rook-config/cmd/rook/cockroachdb/operator.go
T
Guy Margalit 2ce2382676 Refactor operator context init and allow to run local
Signed-off-by: Guy Margalit <guymguym@gmail.com>
Co-Authored-By: Sébastien Han <seb@redhat.com>
Co-Authored-By: Travis Nielsen <tnielsen@redhat.com>

This change is meant to allow running operators locally on a developer machine.
The idea is to allow faster development cycles by reducing the time and complexity of building -> deploying -> debugging on cluster.

For operators that rely only on kubernetes API this works easily - see cockroachdb and minio examples in development-flow doc.

The change includes:

- rook.NewContext() - Refactored to remove repeating initialization code that was copy-pasted in most of the operators in order to create the clusterd.Context and the Clientsets. Also it detects the mode of working in-cluster vs external and sets up the external mode with standard user config (~/.kube/config) and a job executor.
- rook.GetOperatorImage() - Refactor this repeating code in many operators to detect the operator pod image. Also added a global flag --operator-image that developers can use to override this when running locally.
- rook.TerminateOnError() - Added a convenient function.
2019-06-19 20:47:55 +03:00

55 lines
1.6 KiB
Go

/*
Copyright 2018 The Rook Authors. All rights reserved.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package cockroachdb
import (
"github.com/rook/rook/cmd/rook/rook"
operator "github.com/rook/rook/pkg/operator/cockroachdb"
"github.com/rook/rook/pkg/util/flags"
"github.com/spf13/cobra"
)
const containerName = "rook-cockroachdb-operator"
var operatorCmd = &cobra.Command{
Use: "operator",
Short: "Runs the cockroachdb operator to deploy and manage cockroachdb in kubernetes clusters",
Long: `Runs the cockroachdb operator to deploy and manage cockroachdb in kubernetes clusters.
https://github.com/rook/rook`,
}
func init() {
flags.SetFlagsFromEnv(operatorCmd.Flags(), rook.RookEnvVarPrefix)
flags.SetLoggingFlags(operatorCmd.Flags())
operatorCmd.RunE = startOperator
}
func startOperator(cmd *cobra.Command, args []string) error {
rook.SetLogLevel()
rook.LogStartupInfo(operatorCmd.Flags())
logger.Infof("starting cockroachdb operator")
context := rook.NewContext()
rookImage := rook.GetOperatorImage(context.Clientset, containerName)
op := operator.New(context, rookImage)
err := op.Run()
rook.TerminateOnError(err, "failed to run operator")
return nil
}