Writing config files

Author: mmontes11

main.go

@@ -10,7 +10,9 @@ import (
 	"os/signal"
 	"syscall"
 
+	"github.com/mariadb-operator/agent/pkg/filemanager"
 	"github.com/mariadb-operator/agent/pkg/logger"
+	"github.com/mariadb-operator/init/pkg/config"
 	mariadbv1alpha1 "github.com/mariadb-operator/mariadb-operator/api/v1alpha1"
 	"k8s.io/client-go/kubernetes"
 	"k8s.io/client-go/rest"
@@ -61,28 +63,48 @@ func main() {
 	if err != nil {
 		log.Fatalf("error creating logger: %v", err)
 	}
-	logger.Info("Staring MariaDB init")
+	logger.Info("Staring init")
 
-	config, err := config()
+	restConfig, err := restConfig()
 	if err != nil {
 		logger.Error(err, "Error getting Kubernetes config")
 		os.Exit(1)
 	}
-	clientset, err := kubernetes.NewForConfig(config)
+	clientset, err := kubernetes.NewForConfig(restConfig)
 	if err != nil {
-		logger.Error(err, "Error getting Kubernetes clientset")
+		logger.Error(err, "Error creating Kubernetes clientset")
 		os.Exit(1)
 	}
-
 	mdb, err := mariadb(ctx, mariadbName, mariadbNamespace, clientset)
 	if err != nil {
 		logger.Error(err, "Error getting MariaDB")
 		os.Exit(1)
 	}
-	logger.V(1).Info("got MariaDB", "mariadb", mdb)
+
+	fileManager, err := filemanager.NewFileManager(configDir, stateDir)
+	if err != nil {
+		logger.Error(err, "Error creating file manager")
+		os.Exit(1)
+	}
+	configBytes, err := config.NewConfigFile(mdb).Marshal()
+	if err != nil {
+		logger.Error(err, "Error getting galera config")
+		os.Exit(1)
+	}
+	logger.Info("Configuring Galera")
+	if err := fileManager.WriteConfigFile(config.ConfigFileName, configBytes); err != nil {
+		logger.Error(err, "Error writing galera config")
+		os.Exit(1)
+	}
+	logger.Info("Configuring bootstrap")
+	if err := fileManager.WriteConfigFile(config.BootstrapFileName, config.BootstrapFile); err != nil {
+		logger.Error(err, "Error writing bootstrap config")
+		os.Exit(1)
+	}
+	logger.Info("Init done")
 }
 
-func config() (*rest.Config, error) {
+func restConfig() (*rest.Config, error) {
 	if kubeconfig := os.Getenv("KUBECONFIG"); kubeconfig != "" {
 		return clientcmd.BuildConfigFromFlags("", kubeconfig)
 	}

make/dev.mk

@@ -20,8 +20,16 @@ cover: test ## Run tests and generate coverage.
 release: goreleaser ## Test release locally.
 	$(GORELEASER) release --snapshot --rm-dist
 
+
+.PHONY: dir
+dir: ## Create config and state directories for local development.
+	mkdir -p mariadb/config
+	mkdir -p mariadb/state
+
 export KUBECONFIG ?= $(HOME)/.kube/config
+export HOSTNAME ?= mariadb-galera-0
+export MARIADB_ROOT_PASSWORD ?= mariadb
 RUN_FLAGS ?= --log-dev --log-level=debug --log-time-encoder=iso8601 --mariadb-name=mariadb-galera --mariadb-namespace=default --config-dir=mariadb/config --state-dir=mariadb/state
 .PHONY: run
-run: ## Run init from your host.
+run: dir ## Run init from your host.
 	go run main.go $(RUN_FLAGS)

pkg/config/config.go

@@ -8,16 +8,20 @@ import (
 	"strings"
 	"text/template"
 
+	"github.com/mariadb-operator/agent/pkg/galera"
 	mariadbv1alpha1 "github.com/mariadb-operator/mariadb-operator/api/v1alpha1"
 	ctrlresources "github.com/mariadb-operator/mariadb-operator/controllers/resources"
 	"github.com/mariadb-operator/mariadb-operator/pkg/statefulset"
 )
 
 const (
-	BootstrapFile = `[galera]
-wsrep_new_cluster="ON"`
+	ConfigFileName    = "0-galera.cnf"
+	BootstrapFileName = galera.BootstrapFileName
 )
 
+var BootstrapFile = []byte(`[galera]
+wsrep_new_cluster="ON"`)
+
 type ConfigFile struct {
 	mariadb *mariadbv1alpha1.MariaDB
 }