From caef31f48f60b08331642246542a8d68291879cc Mon Sep 17 00:00:00 2001 From: SoXX Date: Tue, 27 Aug 2024 14:17:31 +0200 Subject: [PATCH] feat: Add validation check for user account before starting task --- pkg/plug/grpc.go | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/pkg/plug/grpc.go b/pkg/plug/grpc.go index ae434f1..4de7628 100644 --- a/pkg/plug/grpc.go +++ b/pkg/plug/grpc.go @@ -2,6 +2,7 @@ package plug import ( "context" + "errors" "git.anthrove.art/Anthrove/otter-space-sdk/v3/pkg/database" "git.anthrove.art/Anthrove/otter-space-sdk/v3/pkg/models" @@ -65,6 +66,18 @@ func (s *server) TaskStart(ctx context.Context, creation *gRPC.PlugTaskCreation) s.ctx[id] = cancel span.AddEvent("Created new context for task", trace.WithAttributes(attribute.String("task_id", id))) + if !userSource.AccountValidate { + err = errors.New("user is not validated") + + log.WithContext(taskCtx).WithError(err).WithField("task_id", id).Error("Task execution failed") + span.RecordError(err) + span.SetStatus(codes.Error, err.Error()) + + plugTaskState.TaskState = gRPC.PlugTaskState_STOPPED + + return &plugTaskState, err + } + log.WithContext(taskCtx).WithFields(log.Fields{ "task_id": id, "user_source_id": creation.UserSourceId,