All files / libs/tools/plugin/src/executors/support-weather gcpLogs.ts

0% Statements 0/132
0% Branches 0/1
0% Functions 0/1
0% Lines 0/132

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133                                                                                                                                                                                                                                                                         
import { Logging } from '@google-cloud/logging';
import { uniqWith } from 'lodash-es';

import { dayjs } from '@amalia/ext/dayjs';

const deduplicateString = (input?: string): string | undefined => {
  if (!input) return undefined;

  // Use regex to deduplicate consecutive repeated substrings
  const regex = /\b(?<chunk>.+?)\1+/gu;
  return input.replaceAll(regex, '$1');
};

const GCP_LOG_URL = 'https://console.cloud.google.com/logs/query;query=';

const getStartTimestamp = () => dayjs.utc().startOf('day').subtract(2, 'hours').toISOString();

type RefreshmentError = {
  companyName: string | undefined;
  connectorName: string | undefined;
  objectName: string | undefined;
  message: string | undefined;
};

export const getRefreshmentsLogs = async (): Promise<{ errors: RefreshmentError[]; unknownErrors: string[] }> => {
  const logging = new Logging({ projectId: 'amalia-prod' });
  const logs = await logging.getEntries({
    filter: `resource.type="cloud_run_revision" AND resource.labels.service_name="data-refreshments" AND logName="projects/amalia-prod/logs/run.googleapis.com%2Fstderr" AND timestamp >= "${getStartTimestamp()}"`,
    pageSize: 1000,
  });

  const unknownErrorsUrls: string[] = [];

  const logsFormatted = logs[0]
    .map((log) => {
      // classic logs
      if (log.metadata.jsonPayload) {
        return {
          companyName: deduplicateString(
            log.metadata.jsonPayload.fields?.['company']?.structValue?.fields?.['name']?.stringValue as string,
          ),
          connectorName: deduplicateString(
            log.metadata.jsonPayload.fields?.['connector']?.structValue?.fields?.['type']?.stringValue as string,
          ),
          objectName: deduplicateString(
            log.metadata.jsonPayload.fields?.['dataRefreshment']?.structValue?.fields?.['object']
              ?.stringValue as string,
          ),
          message: (log.metadata.jsonPayload.fields?.['message']?.stringValue as string).replaceAll('\n', ' '),
        };
      }
      // fallback on textPayload logs
      if (log.metadata.textPayload) {
        unknownErrorsUrls.push(
          `${GCP_LOG_URL}insertId=${log.metadata.insertId};duration=P1D - ${log.metadata.textPayload}`,
        );
      }
      return null;
    })
    .filter((log) => log !== null);

  return {
    unknownErrors: unknownErrorsUrls,
    errors: uniqWith(
      logsFormatted,
      (a, b) => a.companyName === b.companyName && a.objectName === b.objectName && a.connectorName === b.connectorName,
    ),
  };
};

type ComputationError = {
  companyName: string | undefined;
  planName: string | undefined;
  periodName: string | undefined;
  message: string | undefined;
};

export const getComputationLogs = async (): Promise<{ errors: ComputationError[]; unknownErrors: string[] }> => {
  const logging = new Logging({ projectId: 'amalia-prod' });
  const logs = await logging.getEntries({
    filter: `resource.type="cloud_run_revision" AND resource.labels.service_name="calculation-engine" AND logName="projects/amalia-prod/logs/run.googleapis.com%2Fstderr" AND timestamp >= "${getStartTimestamp()}"`,
    pageSize: 1000,
  });

  const unknownErrorsUrls: string[] = [];

  const logsFormatted = logs[0]
    .filter((log) => !(log.data as { message?: string }).message?.includes('user not assigned anymore'))
    .map((log) => {
      // classic logs
      if (log.metadata.jsonPayload) {
        return {
          companyName: deduplicateString(
            log.metadata.jsonPayload.fields?.['company']?.structValue?.fields?.['name']?.stringValue as string,
          ),
          message: (log.metadata.jsonPayload.fields?.['message']?.stringValue as string).replaceAll('\n', ' '),
          planName: log.metadata.jsonPayload.fields?.['currentBatch']?.structValue?.fields?.['planName']
            ?.stringValue as string,
          periodName: log.metadata.jsonPayload.fields?.['currentStep']?.structValue?.fields?.['periodName']
            ?.stringValue as string,
        };
      }
      // fallback on textPayload logs
      if (log.metadata.textPayload) {
        unknownErrorsUrls.push(
          `${GCP_LOG_URL}insertId=${log.metadata.insertId};duration=P1D - ${log.metadata.textPayload}`,
        );
      }
      return null;
    })
    .filter((log) => log !== null);

  return {
    unknownErrors: unknownErrorsUrls,
    errors: uniqWith(
      logsFormatted,
      (a, b) => a.companyName === b.companyName && a.periodName === b.periodName && a.planName === b.planName,
    ),
  };
};

export const buildRefreshmentsErrorMessage = (error: RefreshmentError) =>
  [
    error.connectorName ? `\`${error.connectorName}\`` : '',
    error.objectName ? `\`${error.objectName}\`` : '',
    error.message,
  ]
    .filter(Boolean)
    .join(' — ');

export const buildComputationErrorMessage = (error: ComputationError) =>
  [error.planName ? `\`${error.planName}\`` : '', error.periodName, error.message].filter(Boolean).join(' — ');