97 lines
2.3 KiB
Go
97 lines
2.3 KiB
Go
package deploy
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"log/slog"
|
|
"mrvacommander/pkg/artifactstore"
|
|
"mrvacommander/pkg/qldbstore"
|
|
"mrvacommander/pkg/queue"
|
|
"os"
|
|
"strconv"
|
|
)
|
|
|
|
func validateEnvVars(requiredEnvVars []string) {
|
|
missing := false
|
|
|
|
for _, envVar := range requiredEnvVars {
|
|
if _, ok := os.LookupEnv(envVar); !ok {
|
|
slog.Error("Missing required environment variable", "key", envVar)
|
|
missing = true
|
|
}
|
|
}
|
|
|
|
if missing {
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func InitRabbitMQ(isAgent bool) (queue.Queue, error) {
|
|
requiredEnvVars := []string{
|
|
"MRVA_RABBITMQ_HOST",
|
|
"MRVA_RABBITMQ_PORT",
|
|
"MRVA_RABBITMQ_USER",
|
|
"MRVA_RABBITMQ_PASSWORD",
|
|
}
|
|
validateEnvVars(requiredEnvVars)
|
|
|
|
rmqHost := os.Getenv("MRVA_RABBITMQ_HOST")
|
|
rmqPort := os.Getenv("MRVA_RABBITMQ_PORT")
|
|
rmqUser := os.Getenv("MRVA_RABBITMQ_USER")
|
|
rmqPass := os.Getenv("MRVA_RABBITMQ_PASSWORD")
|
|
|
|
rmqPortAsInt, err := strconv.ParseInt(rmqPort, 10, 16)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to parse RabbitMQ port: %v", err)
|
|
}
|
|
|
|
log.Println("Initializing RabbitMQ queue")
|
|
|
|
rabbitMQQueue, err := queue.NewRabbitMQQueue(rmqHost, int16(rmqPortAsInt), rmqUser, rmqPass, isAgent)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to initialize RabbitMQ: %v", err)
|
|
}
|
|
|
|
return rabbitMQQueue, nil
|
|
}
|
|
|
|
func InitMinIOArtifactStore() (artifactstore.Store, error) {
|
|
requiredEnvVars := []string{
|
|
"ARTIFACT_MINIO_ENDPOINT",
|
|
"ARTIFACT_MINIO_ID",
|
|
"ARTIFACT_MINIO_SECRET",
|
|
}
|
|
validateEnvVars(requiredEnvVars)
|
|
|
|
endpoint := os.Getenv("ARTIFACT_MINIO_ENDPOINT")
|
|
id := os.Getenv("ARTIFACT_MINIO_ID")
|
|
secret := os.Getenv("ARTIFACT_MINIO_SECRET")
|
|
|
|
store, err := artifactstore.NewMinIOArtifactStore(endpoint, id, secret)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to initialize artifact store: %v", err)
|
|
}
|
|
|
|
return store, nil
|
|
}
|
|
|
|
func InitMinIOCodeQLDatabaseStore() (qldbstore.Store, error) {
|
|
requiredEnvVars := []string{
|
|
"QLDB_MINIO_ENDPOINT",
|
|
"QLDB_MINIO_ID",
|
|
"QLDB_MINIO_SECRET",
|
|
}
|
|
validateEnvVars(requiredEnvVars)
|
|
|
|
endpoint := os.Getenv("QLDB_MINIO_ENDPOINT")
|
|
id := os.Getenv("QLDB_MINIO_ID")
|
|
secret := os.Getenv("QLDB_MINIO_SECRET")
|
|
|
|
store, err := qldbstore.NewMinIOCodeQLDatabaseStore(endpoint, id, secret)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to initialize ql database storage: %v", err)
|
|
}
|
|
|
|
return store, nil
|
|
}
|