heruvym/cmd/validator/main.go

123 lines
2.7 KiB
Go
Raw Normal View History

2024-12-02 12:12:52 +00:00
package main
import (
"bytes"
2024-12-02 12:12:52 +00:00
"context"
"fmt"
"gitea.pena/PenaSide/common/validate"
"gitea.pena/PenaSide/heruvym/internal/initialize"
minioDal "gitea.pena/PenaSide/heruvym/internal/repository/minio"
2024-12-09 10:36:43 +00:00
"github.com/caarlos0/env/v8"
2024-12-02 12:12:52 +00:00
"github.com/minio/minio-go/v7"
"github.com/pioz/faker"
2024-12-02 12:12:52 +00:00
"log"
)
func main() {
cfg, err := loadConfig()
if err != nil {
log.Fatalf("error loading config: %v", err)
}
if cfg.ClientHttpUrl == "" {
log.Fatalf("error client http url is empty")
}
if cfg.AdminHttpUrl == "" {
log.Fatalf("error admin http url is empty")
2024-12-02 12:12:52 +00:00
}
err = validate.ValidateMongo(cfg.ExternalCfg.Database)
if err != nil {
log.Fatalf("error validating mongo: %v", err)
}
err = validate.ValidateTgToken(cfg.TelegramToken)
if err != nil {
log.Fatalf("error validating telegram token: %v", err)
}
if cfg.TelegramChannelID == 0 {
log.Fatalf("telegram channel id must be set")
}
err = validateRedis(cfg)
if err != nil {
log.Fatalf("error validating redis: %v", err)
}
err = validateS3(cfg)
if err != nil {
log.Fatalf("error validating s3: %v", err)
}
}
func loadConfig() (initialize.Config, error) {
2024-12-09 10:36:43 +00:00
var config initialize.Config
if err := env.Parse(&config); err != nil {
2024-12-09 12:37:22 +00:00
return config, err
2024-12-02 12:12:52 +00:00
}
return config, nil
}
func validateRedis(cfg initialize.Config) error {
if cfg.RedisHost == "" {
return fmt.Errorf("redis host is empty")
}
_, err := initialize.Redis(context.TODO(), cfg)
if err != nil {
return err
}
return nil
}
func validateS3(cfg initialize.Config) error {
if cfg.S3Endpoint == "" {
return fmt.Errorf("s3 endpoint is empty")
}
if cfg.S3AccessKey == "" {
return fmt.Errorf("s3 access key is empty")
}
if cfg.S3SecretKey == "" {
return fmt.Errorf("s3 secret key is empty")
}
ctx := context.TODO()
client, err := initialize.Minio(cfg)
if err != nil {
return fmt.Errorf("error connect s3 in validate: %w", err)
2024-12-02 12:12:52 +00:00
}
objectName := faker.String()
content := []byte(faker.String())
contentType := "text/plain"
2024-12-02 12:12:52 +00:00
exist, err := client.BucketExists(ctx, minioDal.Bucket)
2024-12-02 12:12:52 +00:00
if err != nil {
return fmt.Errorf("error find s3 bucket in validate: %w", err)
2024-12-02 12:12:52 +00:00
}
// не создаем бакет так как бакет могут не дать создать
2024-12-02 12:12:52 +00:00
if !exist {
2024-12-06 08:42:03 +00:00
log.Println("bucket does not exist")
return nil
2024-12-02 12:12:52 +00:00
}
_, err = client.PutObject(ctx, minioDal.Bucket, objectName,
bytes.NewReader(content), int64(len(content)),
minio.PutObjectOptions{ContentType: contentType})
2024-12-02 12:12:52 +00:00
if err != nil {
return fmt.Errorf("error create test object: %w", err)
}
err = client.RemoveObject(ctx, minioDal.Bucket, objectName, minio.RemoveObjectOptions{})
if err != nil {
return fmt.Errorf("error remove test object: %w", err)
2024-12-02 12:12:52 +00:00
}
return nil
}