65 lines
1.9 KiB
TypeScript
65 lines
1.9 KiB
TypeScript
|
import { CoreV1Api, KubeConfig, Log } from '@kubernetes/client-node';
|
||
|
|
import { Writable } from 'stream';
|
||
|
|
import * as core from '@actions/core';
|
||
|
|
|
||
|
|
class KubernetesLogging {
|
||
|
|
static async streamLogs(
|
||
|
|
kubeConfig: KubeConfig,
|
||
|
|
kubeClient: CoreV1Api,
|
||
|
|
jobName: string,
|
||
|
|
podName: string,
|
||
|
|
containerName: string,
|
||
|
|
namespace: string,
|
||
|
|
logCallback: any,
|
||
|
|
) {
|
||
|
|
core.info(`Streaming logs from pod: ${podName} container: ${containerName} namespace: ${namespace}`);
|
||
|
|
const stream = new Writable();
|
||
|
|
let didStreamAnyLogs: boolean = false;
|
||
|
|
stream._write = (chunk, encoding, next) => {
|
||
|
|
didStreamAnyLogs = true;
|
||
|
|
logCallback(chunk.toString());
|
||
|
|
next();
|
||
|
|
};
|
||
|
|
const logOptions = {
|
||
|
|
follow: true,
|
||
|
|
pretty: true,
|
||
|
|
previous: false,
|
||
|
|
};
|
||
|
|
try {
|
||
|
|
const resultError = await new Promise(async (resolve) =>
|
||
|
|
new Log(kubeConfig).log(namespace, podName, containerName, stream, resolve, logOptions),
|
||
|
|
);
|
||
|
|
if (resultError) {
|
||
|
|
throw resultError;
|
||
|
|
}
|
||
|
|
if (!didStreamAnyLogs) {
|
||
|
|
throw new Error(
|
||
|
|
JSON.stringify(
|
||
|
|
{
|
||
|
|
message: 'Failed to stream any logs, listing namespace events, check for an error with the container',
|
||
|
|
events: (await kubeClient.listNamespacedEvent(namespace)).body.items
|
||
|
|
.filter((x) => {
|
||
|
|
return x.involvedObject.name === podName || x.involvedObject.name === jobName;
|
||
|
|
})
|
||
|
|
.map((x) => {
|
||
|
|
return {
|
||
|
|
type: x.involvedObject.kind,
|
||
|
|
name: x.involvedObject.name,
|
||
|
|
message: x.message,
|
||
|
|
};
|
||
|
|
}),
|
||
|
|
},
|
||
|
|
undefined,
|
||
|
|
4,
|
||
|
|
),
|
||
|
|
);
|
||
|
|
}
|
||
|
|
} catch (error) {
|
||
|
|
throw error;
|
||
|
|
}
|
||
|
|
core.info('end of log stream');
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
export default KubernetesLogging;
|