Skip to content

Commit

Permalink
adapter-k8s, remove old adapters package
Browse files Browse the repository at this point in the history
  • Loading branch information
patuwwy committed Jun 29, 2023
1 parent c348536 commit 4abd144
Show file tree
Hide file tree
Showing 13 changed files with 812 additions and 0 deletions.
7 changes: 7 additions & 0 deletions packages/adapter-k8s/.eslintrc.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
module.exports = {
ignorePatterns: [".eslintrc.js"],
parserOptions:{
project: "./tsconfig.json",
tsconfigRootDir: __dirname
}
};
13 changes: 13 additions & 0 deletions packages/adapter-k8s/.nycrc.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
{
"all": true,
"include": [
"dist/**/*"
],
"exclude": [
"**/*.spec.ts"
],
"reporter": [
"lcovonly",
"text"
]
}
61 changes: 61 additions & 0 deletions packages/adapter-k8s/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,61 @@
# Scramjet Transform Hub Adapters

This module holds two types of adapters utilized by Scramjet Transform Hub: Instance Adapter and Sequence Adapter. These Adapters allows for running the Sequence identification and Instance execution in two basic modes: as a non containerized standalone processes or in a Docker container.

The adapter provides two main exports:

* [DockerSequenceAdapter](https://github.com/scramjetorg/transform-hub/tree/HEAD/packages/adapters/src/docker-sequence-adapter.ts) - An adapter for preparing Sequence to be run in Docker container.
* [DockerInstanceAdapter](https://github.com/scramjetorg/transform-hub/tree/HEAD/packages/adapters/src/docker-instance-adapter.ts) - An adapter for running Instance by Runner executed in Docker container.

## Docs

See the code documentation here: [scramjetorg/transform-hub/docs/adapters/modules.md](https://github.com/scramjetorg/transform-hub/tree/HEAD/docs/adapters/modules.md)

## Scramjet Transform Hub

This package is part of [Scramjet Transform Hub](https://www.npmjs.org/package/@scramjet/sth).

Scramjet Transform Hub is a deployment and execution platform. Once installed on a server, it will allow you to start your programs and keep them running on a remote machine. You will be able to start programs in the background or connect to them and see their output directly on your terminal. You will be able to pipe your local data to the program, as if it was running from your terminal. You can start your server in AWS, Google Cloud or Azure, start it on your local machine, install it on a Raspberry Pi or wherever else you'd like.

## Use cases

There's no limit what you can use it for. You want a stock checker? A chat bot? Maybe you'd like to automate your home? Retrieve sensor data? Maybe you have a lot of data and want to transfer and wrangle it? You have a database of cities and you'd like to enrich your data? You do machine learning and you want to train your set while the data is fetched in real time? Hey, you want to use it for something else and ask us if that's a good use? Ask us [via email](mailto:get@scramjet.org) or hop on our [Scramjet Discord](https://scr.je/join-community-mg1)!

## Some important links

* Scramjet, the company behind [Transform Hub](https://scramjet.org)
* The [Scramjet Framework - functional reactive stream processing framework](https://framework.scramjet.org)
* The [Transform Hub repo on github](https://github.com/scramjetorg/transform-hub)
* You can see the [Scramjet Transform Hub API docs here](https://github.com/scramjetorg/transform-hub/tree/HEAD/docs/api-client/README.md)
* You can see the [CLI documentation here](https://github.com/scramjetorg/transform-hub/tree/HEAD/packages/cli/README.md), but `si help` should also be quite effective.
* Don't forget to ⭐ this repo if you like it, `subscribe` to releases and keep visiting us for new versions and updates.
* You can [open an issue - file a bug report or a feature request here](https://github.com/scramjetorg/transform-hub/issues/new/choose)

## License and contributions

This module is licensed under AGPL-3.0 license.

The Scramjet Transform Hub project is dual-licensed under the AGPL-3.0 and MIT licenses. Parts of the project that are linked with your programs are MIT licensed, the rest is AGPL.

## Contributions

We accept valid contributions and we will be publishing a more specific project roadmap so contributors can propose features and also help us implement them. We kindly ask you that contributed commits are Signed-Off `git commit --sign-off`.

We provide support for contributors via test cases. If you expect a certain type of workflow to be officially supported, please specify and implement a test case in `Gherkin` format in `bdd` directory and include it in your pull request. More info about our BDD test you will find [here](https://github.com/scramjetorg/transform-hub/tree/HEAD/bdd/README.md).

### Help wanted 👩‍🎓🧑👱‍♀️

The project need's your help! There's lots of work to do and we have a lot of plans. If you want to help and be part of the Scramjet team, please reach out to us, [on discord](https://scr.je/join-community-mg1) or email us: [opensource@scramjet.org](mailto:opensource@scramjet.org).

### Donation 💸

Do you like this project? It helped you to reduce time spent on delivering your solution? You are welcome to buy us a coffee ☕ Thanks a lot! 😉

[You can sponsor us on github](https://github.com/sponsors/scramjetorg)

* There's also a Paypal donation link if you prefer that:

[![paypal](https://www.paypalobjects.com/en_US/i/btn/btn_donateCC_LG.gif)](https://www.paypal.com/cgi-bin/webscr?cmd=_s-xclick&hosted_button_id=7F7V65C43EBMW)



61 changes: 61 additions & 0 deletions packages/adapter-k8s/package.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,61 @@
{
"name": "@scramjet/adapter-k8s",
"version": "0.34.4",
"description": "This package is part of Scramjet Transform Hub. This module holds the docker adapters utilized by Scramjet Transform Hub",
"main": "./src/index.ts",
"scripts": {
"start": "ts-node ./src/index",
"build": "../../scripts/build-all.js --config-name=tsconfig.build.json --copy-dist",
"build:docs": "typedoc",
"clean": "rm -rf ./dist .bic_cache",
"cloc": "cloc src --fullpath --include-lang TypeScript --not-match-d \"(node_modules|test|dist|bdd)\" --by-percent cm",
"test": "nyc ava",
"prepack": "node ../../scripts/publish.js"
},
"author": "Scramjet <open-source@scramjet.org>",
"license": "AGPL-3.0",
"dependencies": {
"@scramjet/model": "^0.34.4",
"@scramjet/obj-logger": "^0.34.4",
"@scramjet/pre-runner": "^0.34.4",
"@scramjet/python-runner": "^0.34.4",
"@scramjet/runner": "^0.34.4",
"@scramjet/sth-config": "^0.34.4",
"@scramjet/symbols": "^0.34.4",
"@scramjet/utility": "^0.34.4",
"@scramjet/adapters-utils": "^0.34.4",
"@kubernetes/client-node": "^0.17.1",
"scramjet": "^4.36.9",
"shell-escape": "^0.2.0",
"systeminformation": "^5.12.7",
"ts.data.json": "^2.2.0"
},
"devDependencies": {
"@scramjet/types": "^0.34.4",
"@types/js-yaml": "4.0.5",
"@types/node": "15.12.5",
"@types/request": "2.48.8",
"@types/shell-escape": "^0.2.1",
"@types/ws": "8.5.3",
"ava": "^3.15.0",
"ts-node": "^10.9.1",
"typedoc": "^0.23.17",
"typedoc-plugin-markdown": "^3.13.6",
"typescript": "~4.7.4"
},
"ava": {
"extensions": [
"ts"
],
"files": [
"**/*.spec.ts"
],
"require": [
"ts-node/register"
]
},
"repository": {
"type": "git",
"url": "https://github.com/scramjetorg/transform-hub.git"
}
}
11 changes: 11 additions & 0 deletions packages/adapter-k8s/src/index.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
/**
* Adapter module must provide SequenceAdapter, InstanceAdapter classes, init method and name field.
*/
export { KubernetesSequenceAdapter as SequenceAdapter } from "./kubernetes-sequence-adapter";
export { KubernetesInstanceAdapter as InstanceAdapter } from "./kubernetes-instance-adapter";

export const init = (..._args: any[]) => {
return true;
};

export const name = "kubernetes";
212 changes: 212 additions & 0 deletions packages/adapter-k8s/src/kubernetes-client-adapter.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,212 @@
import { IObjectLogger } from "@scramjet/types";
import * as k8s from "@kubernetes/client-node";
import { ObjLogger } from "@scramjet/obj-logger";
import { defer } from "@scramjet/utility";
import { Writable, Readable } from "stream";
import http from "http";
import { HttpError } from "@kubernetes/client-node";

const POD_STATUS_CHECK_INTERVAL_MS = 500;
const POD_STATUS_FAIL_LIMIT = 10;

class KubernetesClientAdapter {
logger: IObjectLogger;
name = "KubernetesClientAdapter";

private _configPath: string;
private _config?: k8s.KubeConfig;
private _namespace: string;

constructor(configPath: string = "", namespace: string = "default") {
this.logger = new ObjLogger(this.name);
this._configPath = configPath;
this._namespace = namespace;
}

private get config(): k8s.KubeConfig {
if (!this._config) {
throw new Error("Kubernetes API client not initialized");
}

return this._config;
}

public init() {
const kc = new k8s.KubeConfig();

try {
if (this._configPath && this._configPath.length) {
kc.loadFromFile(this._configPath);
} else {
kc.loadFromCluster();
}

this._config = kc;
} catch (err: any) {
this.logger.error("Unable to load kubeconfig", err);
}
}

async createPod(metadata: k8s.V1ObjectMeta, spec: k8s.V1PodSpec, retries: number = 0) {
const kubeApi = this.config.makeApiClient(k8s.CoreV1Api);

const result = await this.runWithRetries(retries, "Create Pod", () =>
kubeApi.createNamespacedPod(this._namespace, {
apiVersion: "v1",
kind: "Pod",
metadata,
spec
})
);

return result as {
response: http.IncomingMessage;
body: k8s.V1Pod;
};
}

async deletePod(podName: string, retries: number = 0) {
const kubeApi = this.config.makeApiClient(k8s.CoreV1Api);

const result = await this.runWithRetries(retries, "Delete Pod", () =>
kubeApi.deleteNamespacedPod(podName, this._namespace, undefined, undefined, 0)
);

return result as {
response: http.IncomingMessage;
body: k8s.V1Pod;
};
}

async exec(podName: string, containerName: string, command: string | string[],
stdout: Writable | null, stderr: Writable | null, stdin: Readable | null, retries: number = 0
) {
const exec = new k8s.Exec(this.config);

await this.runWithRetries(retries, "Exec", () =>
exec.exec(this._namespace, podName, containerName, command, stdout, stderr, stdin, false,
(...args) => this.logger.debug("exec status", ...args))
);
}

async waitForPodStatus(podName: string, expectedStatuses: string[]): Promise<{ status: string, code?: number }> {
const kubeApi = this.config.makeApiClient(k8s.CoreV1Api);

let failCount = 0;

// eslint-disable-next-line no-constant-condition
while (true) {
try {
const response = await kubeApi.readNamespacedPodStatus(podName, this._namespace);
const status = response.body.status?.phase || "";

const container = (response.body.status?.containerStatuses || []).find(c => c.name === podName);

if (expectedStatuses.includes(status)) {
return {
status,
code: container?.state?.terminated?.exitCode
};
}
} catch (err: any) {
if (err instanceof HttpError) {
this.logger.error(`Status for "${podName}" pod responded with error`, err?.body?.message);

if (err.statusCode === 404) {
this.logger.error("You have deleted this pod already! Try to increase runnerExitDelay in CSIController.");
}
} else {
this.logger.error(`Failed to get pod status: ${podName}.`, err);
}

failCount++;

if (failCount > POD_STATUS_FAIL_LIMIT) {
throw new Error("Reached the limit of failed pod status requests");
}
}

await defer(POD_STATUS_CHECK_INTERVAL_MS);
}
}

async getPodLog(podName: string): Promise<string[]> {
const kubeApi = this.config.makeApiClient(k8s.CoreV1Api);
const response = await kubeApi.readNamespacedPodLog(
podName, this._namespace,
undefined, false, undefined, undefined,
undefined, false, undefined,
100, true
);

return [response.body];
}

async getPodTerminatedContainerReason(podName: string): Promise<string | undefined> {
const kubeApi = this.config.makeApiClient(k8s.CoreV1Api);
const response = await kubeApi.readNamespacedPod(podName, this._namespace);

return response.body.status?.containerStatuses?.[0].state?.terminated?.reason;
}

async isPodsLimitReached(quotaName: string) {
const kubeApi = this.config.makeApiClient(k8s.CoreV1Api);

try {
const getQuotaPromise =
await kubeApi.readNamespacedResourceQuota(quotaName, this._namespace);

const responseBody = getQuotaPromise.body;

if (responseBody) {
const used = parseInt(responseBody.status?.used?.pods || "", 10) || 0;
const hard = parseInt(responseBody.status?.hard?.pods || "", 10) || Infinity;

this.logger.info("Pods limit quota", used, hard);

return used >= hard;
}
} catch (e) {
this.logger.warn("Can't get quota object. ");
}

return false;
}

private async runWithRetries(retries: number, name: string, callback: any) {
let tries = 0;
let sleepMs = 1000;
let success = false;
let result: any = null;

this.logger.debug(`Starting: ${name}...`);

while (!success && tries <= retries) {
tries++;

try {
result = await callback();

success = true;
} catch (err: any) {
if (err instanceof HttpError) {
this.logger.error(`Running "${name}" responded with error`, err?.body?.message);
} else {
this.logger.error(`Failed to run: ${name}.`, err);
}

await defer(sleepMs);

sleepMs *= 2;
}
}

if (!success) {
throw new Error(`Failed to run: ${name} after ${tries} retries.`);
}

return result;
}
}

export { KubernetesClientAdapter };
19 changes: 19 additions & 0 deletions packages/adapter-k8s/src/kubernetes-config-decoder.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
import { K8SAdapterConfiguration } from "@scramjet/types";
import { JsonDecoder } from "ts.data.json";

export const adapterConfigDecoder = JsonDecoder.object<K8SAdapterConfiguration>({
authConfigPath: JsonDecoder.optional(JsonDecoder.string),
namespace: JsonDecoder.string,
quotaName: JsonDecoder.optional(JsonDecoder.string),
sthPodHost: JsonDecoder.string,
runnerImages: JsonDecoder.object({
python3: JsonDecoder.string,
node: JsonDecoder.string
}, "K8SImagesDecoder"),
sequencesRoot: JsonDecoder.string,
timeout: JsonDecoder.optional(JsonDecoder.string),
runnerResourcesRequestsMemory: JsonDecoder.optional(JsonDecoder.string),
runnerResourcesRequestsCpu: JsonDecoder.optional(JsonDecoder.string),
runnerResourcesLimitsMemory: JsonDecoder.optional(JsonDecoder.string),
runnerResourcesLimitsCpu: JsonDecoder.optional(JsonDecoder.string)
}, "K8SAdapterConfiguration");
Loading

0 comments on commit 4abd144

Please sign in to comment.