From cdb93db57ea5b422e10df09ffd0743139e1a2c6f Mon Sep 17 00:00:00 2001
From: Semyon Krestyaninov <krestyaninov@perx.ru>
Date: Tue, 2 Apr 2024 08:59:10 +0000
Subject: [PATCH] =?UTF-8?q?=D0=94=D0=BE=D0=B1=D0=B0=D0=B2=D0=BB=D0=B5?=
 =?UTF-8?q?=D0=BD=D0=BE=20=D0=BB=D0=BE=D0=B3=D0=B8=D1=80=D0=BE=D0=B2=D0=B0?=
 =?UTF-8?q?=D0=BD=D0=B8=D0=B5=20=D1=81=D0=BE=D0=B1=D1=8B=D1=82=D0=B8=D0=B9?=
 =?UTF-8?q?=20=D1=81=D0=B5=D1=80=D0=B2=D0=B8=D1=81=D0=B0=20=D0=BE=D0=BA?=
 =?UTF-8?q?=D1=80=D1=83=D0=B6=D0=B5=D0=BD=D0=B8=D0=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pkg/environments/events.go                    |   8 +
 .../middleware/logging_middleware.go          | 166 ++++++++++++++++++
 pkg/environments/middleware/middleware.go     |   2 +-
 3 files changed, 175 insertions(+), 1 deletion(-)
 create mode 100644 pkg/environments/events.go
 create mode 100644 pkg/environments/middleware/logging_middleware.go

diff --git a/pkg/environments/events.go b/pkg/environments/events.go
new file mode 100644
index 00000000..21b79f7f
--- /dev/null
+++ b/pkg/environments/events.go
@@ -0,0 +1,8 @@
+package environments
+
+const (
+	EventCreate  = "environment.create"
+	EventUpdate  = "environment.update"
+	EventDelete  = "environment.delete"
+	EventMigrate = "environment.migrate"
+)
diff --git a/pkg/environments/middleware/logging_middleware.go b/pkg/environments/middleware/logging_middleware.go
new file mode 100644
index 00000000..df753096
--- /dev/null
+++ b/pkg/environments/middleware/logging_middleware.go
@@ -0,0 +1,166 @@
+package middleware
+
+import (
+	"context"
+	"git.perx.ru/perxis/perxis-go/id"
+	"git.perx.ru/perxis/perxis-go/pkg/environments"
+
+	logzap "git.perx.ru/perxis/perxis-go/zap"
+	"go.uber.org/zap"
+)
+
+type loggingMiddleware struct {
+	logger *zap.Logger
+	next   environments.Environments
+}
+
+func LoggingMiddleware(logger *zap.Logger) Middleware {
+	return func(next environments.Environments) environments.Environments {
+		return &loggingMiddleware{
+			next:   next,
+			logger: logger.With(logzap.Component("Environments")),
+		}
+	}
+}
+
+func (m *loggingMiddleware) Create(ctx context.Context, env *environments.Environment) (created *environments.Environment, err error) {
+	logger := m.logger.With(
+		logzap.CallerFromContext(ctx, ""),
+		logzap.Event(environments.EventCreate),
+		logzap.Object(env),
+	)
+
+	created, err = m.next.Create(ctx, env)
+	if err != nil {
+		logger.Error("Failed to create", zap.Error(err), logzap.Channels(logzap.Userlog, logzap.Syslog))
+		return
+	}
+
+	logger.Info("Environment created", logzap.Channels(logzap.Userlog))
+
+	return created, err
+}
+
+func (m *loggingMiddleware) Delete(ctx context.Context, spaceId string, envId string) (err error) {
+	logger := m.logger.With(
+		logzap.CallerFromContext(ctx, ""),
+		logzap.Event(environments.EventDelete),
+		logzap.Object(id.NewEnvironmentId(spaceId, envId)),
+	)
+
+	err = m.next.Delete(ctx, spaceId, envId)
+	if err != nil {
+		logger.Error("Failed to delete", zap.Error(err), logzap.Channels(logzap.Userlog, logzap.Syslog))
+		return
+	}
+
+	logger.Info("Environment deleted", logzap.Channels(logzap.Userlog))
+
+	return err
+}
+
+func (m *loggingMiddleware) Get(ctx context.Context, spaceId string, envId string) (env *environments.Environment, err error) {
+	logger := m.logger.With(
+		logzap.CallerFromContext(ctx, ""),
+		logzap.Object(id.NewEnvironmentId(spaceId, envId)),
+	)
+
+	env, err = m.next.Get(ctx, spaceId, envId)
+	if err != nil {
+		logger.Error("Failed to get", zap.Error(err))
+		return
+	}
+
+	return env, err
+}
+
+func (m *loggingMiddleware) List(ctx context.Context, spaceId string) (envs []*environments.Environment, err error) {
+	logger := m.logger.With(
+		logzap.CallerFromContext(ctx, ""),
+	)
+
+	envs, err = m.next.List(ctx, spaceId)
+	if err != nil {
+		logger.Error("Failed to list", zap.Error(err))
+		return
+	}
+
+	return envs, err
+}
+
+func (m *loggingMiddleware) Migrate(ctx context.Context, spaceId string, envId string, options ...*environments.MigrateOptions) (err error) {
+	logger := m.logger.With(
+		logzap.CallerFromContext(ctx, ""),
+		logzap.Event(environments.EventMigrate),
+		logzap.Object(id.NewEnvironmentId(spaceId, envId)),
+	)
+
+	err = m.next.Migrate(ctx, spaceId, envId, options...)
+	if err != nil {
+		logger.Error("Failed to migrate", zap.Error(err), logzap.Channels(logzap.Userlog, logzap.Syslog))
+		return
+	}
+
+	opt := environments.MergeMigrateOptions(options...)
+
+	// В случае асинхронной миграции, информация о ней будет отправлена в сервис логирования внутри самого сервиса.
+	if !opt.Async {
+		logger.Info("Environment migrated", logzap.Channels(logzap.Userlog))
+	}
+
+	return err
+}
+
+func (m *loggingMiddleware) RemoveAlias(ctx context.Context, spaceId string, envId string, alias string) (err error) {
+	logger := m.logger.With(
+		logzap.CallerFromContext(ctx, ""),
+		logzap.Event(environments.EventUpdate),
+		logzap.Object(id.NewEnvironmentId(spaceId, envId)),
+	)
+
+	err = m.next.RemoveAlias(ctx, spaceId, envId, alias)
+	if err != nil {
+		logger.Error("Failed to remove alias", zap.Error(err), logzap.Channels(logzap.Userlog, logzap.Syslog))
+		return
+	}
+
+	logger.Info("Environment alias removed", logzap.Channels(logzap.Userlog))
+
+	return err
+}
+
+func (m *loggingMiddleware) SetAlias(ctx context.Context, spaceId string, envId string, alias string) (err error) {
+	logger := m.logger.With(
+		logzap.CallerFromContext(ctx, ""),
+		logzap.Event(environments.EventUpdate),
+		logzap.Object(id.NewEnvironmentId(spaceId, envId)),
+	)
+
+	err = m.next.SetAlias(ctx, spaceId, envId, alias)
+	if err != nil {
+		logger.Error("Failed to set alias", zap.Error(err), logzap.Channels(logzap.Userlog, logzap.Syslog))
+		return
+	}
+
+	logger.Info("Environment alias was set", logzap.Channels(logzap.Userlog))
+
+	return err
+}
+
+func (m *loggingMiddleware) Update(ctx context.Context, env *environments.Environment) (err error) {
+	logger := m.logger.With(
+		logzap.CallerFromContext(ctx, ""),
+		logzap.Event(environments.EventUpdate),
+		logzap.Object(env),
+	)
+
+	err = m.next.Update(ctx, env)
+	if err != nil {
+		logger.Error("Failed to update", zap.Error(err), logzap.Channels(logzap.Userlog, logzap.Syslog))
+		return
+	}
+
+	logger.Info("Environment updated", logzap.Channels(logzap.Userlog))
+
+	return err
+}
diff --git a/pkg/environments/middleware/middleware.go b/pkg/environments/middleware/middleware.go
index 7c887417..d8e28169 100644
--- a/pkg/environments/middleware/middleware.go
+++ b/pkg/environments/middleware/middleware.go
@@ -21,7 +21,7 @@ func WithLog(s environments.Environments, logger *zap.Logger, log_access bool) e
 	if log_access {
 		s = AccessLoggingMiddleware(logger)(s)
 	}
-	s = ErrorLoggingMiddleware(logger)(s)
+	s = LoggingMiddleware(logger)(s)
 
 	s = RecoveringMiddleware(logger)(s)
 	return s
-- 
GitLab