@punks/backend-entity-manager 0.0.300 → 0.0.301

package/dist/cjs/types/platforms/nest/plugins/jobs/aws-batch/manager/index.d.ts

@@ -6,6 +6,7 @@ export declare class AwsBatchService {
     submitJob(input: SubmitBatchJobInput): Promise<void>;
     ensureJobDefinition(definition: AwsJobDefinition): Promise<import("@aws-sdk/client-batch").JobDefinition>;
     private unregisterJobDefinition;
+    private ensureJobLogDriver;
     private registerJobDefinition;
     private getLatestJobDefinition;
     private getActiveJobDefinitions;

package/dist/cjs/types/platforms/nest/plugins/jobs/aws-batch/models/index.d.ts

@@ -7,6 +7,7 @@ export type AwsBatchInfrastructureParams = {
     dockerImage: string;
     computePlatformType?: AwsJobComputePlatformType;
     assignPublicIp?: boolean;
+    logGroupName?: string;
 };
 export type AwsBatchInvocationParams = {
     startCommand: string[];

package/dist/cjs/types/platforms/nest/plugins/jobs/aws-batch/utils/logs.d.ts

@@ -0,0 +1,2 @@
+import { AwsBatchSettings } from "../settings";
+export declare const ensureJobLogGroup: (logGroupName: string, awsSettings: AwsBatchSettings) => Promise<void>;

package/dist/esm/index.js

@@ -18,6 +18,7 @@ import { getSignedUrl } from '@aws-sdk/s3-request-presigner';
 import { SendEmailCommand, SESClient } from '@aws-sdk/client-ses';
 import { MailService } from '@sendgrid/mail';
 import { BatchClient, CancelJobCommand, ListJobsCommand, SubmitJobCommand, DeregisterJobDefinitionCommand, RegisterJobDefinitionCommand, JobDefinitionType, PlatformCapability, ResourceType, AssignPublicIp, DescribeJobDefinitionsCommand, CreateJobQueueCommand, DescribeJobQueuesCommand, JobStatus as JobStatus$1 } from '@aws-sdk/client-batch';
+import { CloudWatchLogsClient, DescribeLogGroupsCommand, CreateLogGroupCommand } from '@aws-sdk/client-cloudwatch-logs';
 import require$$1$1 from 'http';
 import require$$2 from 'https';
 import require$$0$1 from 'url';
@@ -39973,6 +39974,28 @@ const awsBatchSettings = new AppInMemorySettings();
 
 const JobProvider = (provider) => SetMetadata(JOB_PROVIDER, provider);
 
+const ensureJobLogGroup = async (logGroupName, awsSettings) => {
+    const client = new CloudWatchLogsClient({
+        ...(awsSettings.awsSecretAccessKey && awsSettings.awsSecretAccessKey
+            ? {
+                accessKeyId: awsSettings.awsSecretAccessKey,
+                secretAccessKey: awsSettings.awsSecretAccessKey,
+            }
+            : {}),
+    });
+    const describeCommand = new DescribeLogGroupsCommand({
+        logGroupNamePrefix: logGroupName,
+    });
+    const describeResponse = await client.send(describeCommand);
+    const existingLogGroup = describeResponse.logGroups?.find((group) => group.logGroupName === logGroupName);
+    if (!existingLogGroup) {
+        const createCommand = new CreateLogGroupCommand({
+            logGroupName,
+        });
+        await client.send(createCommand);
+    }
+};
+
 var AwsBatchService_1;
 const jobDefinitionName = (jobUid) => `job-${jobUid}`;
 const jobInstanceName = (jobUid, instanceId) => `job-${jobUid}-${instanceId}`;
@@ -40066,7 +40089,16 @@ let AwsBatchService = AwsBatchService_1 = class AwsBatchService {
             jobDefinition: jobDefinitionArn,
         }));
     }
+    async ensureJobLogDriver(jobName, definition) { }
     async registerJobDefinition(jobName, definition) {
+        const logGroup = definition.infrastructureParams.logGroupName
+            ? {
+                name: `/aws/batch/job-${definition.infrastructureParams.logGroupName}`,
+            }
+            : undefined;
+        if (logGroup) {
+            await ensureJobLogGroup(logGroup.name, this.awsSettings);
+        }
         await this.client().send(new RegisterJobDefinitionCommand({
             jobDefinitionName: jobName,
             type: JobDefinitionType.Container,
@@ -40100,6 +40132,16 @@ let AwsBatchService = AwsBatchService_1 = class AwsBatchService {
                         assignPublicIp: AssignPublicIp.ENABLED,
                     }
                     : undefined,
+                logConfiguration: logGroup
+                    ? {
+                        logDriver: "awslogs",
+                        options: {
+                            "awslogs-group": logGroup.name,
+                            "awslogs-region": await this.client().config.region(),
+                            "awslogs-stream-prefix": "ecs",
+                        },
+                    }
+                    : undefined,
             },
         }));
     }