Skip to content

Commit

Permalink
[CHANGE] removed service starting/request etc from echo-service
Browse files Browse the repository at this point in the history 
[EXAMPLE] added sample service monitoring
  • Loading branch information
@aricart
aricart committed May 19, 2023
1 parent fe31d5d commit b02abeb
Show file tree
Hide file tree
Showing 2 changed files with 192 additions and 228 deletions.
238 changes: 10 additions & 228 deletions examples/services/echo-service.ts
View file
Expand Up @@ -14,160 +14,17 @@
*/

import { cli, Command, Flags } from "https://deno.land/x/cobra@v0.0.9/mod.ts";
import {
connect,
headers,
millis,
NatsConnection,
StringCodec,
} from "../../src/mod.ts";
import { collect } from "../../nats-base-client/util.ts";
import { connect, headers } from "../../src/mod.ts";

const root = cli({
use: "echo-service [--server server:port] (service | monitor | request)",
});

root.addFlag({
name: "server",
type: "string",
usage: "NATS server host:port",
default: "demo.nats.io",
persistent: true,
});

const adm = root.addCommand({
use: "monitor",
short: "monitor [ping|info|status] [--name n] [--id id]",
});

adm.addFlag({
short: "n",
name: "name",
type: "string",
usage: "service name to filter on",
default: "",
persistent: true,
});
adm.addFlag({
name: "id",
type: "string",
usage: "service id to filter on",
default: "",
persistent: true,
});

adm.addCommand({
use: "ping [--name] [--id]",
short: "pings services",
run: async (
cmd: Command,
_args: string[],
flags: Flags,
): Promise<number> => {
const nc = await createConnection(flags);
const opts = options(flags);
const mc = nc.services.client();
const infos = (await collect(await mc.ping(opts.name, opts.id))).sort(
(a, b) => {
const A = `${a.name} ${a.version}`;
const B = `${b.name} ${b.version}`;
return B.localeCompare(A);
},
);
if (infos.length) {
console.table(infos);
} else {
cmd.stdout("no services found");
}
await nc.close();
return 0;
},
});

adm.addCommand({
use: "stats [--name] [--id]",
short: "get service stats",
run: async (
cmd: Command,
_args: string[],
flags: Flags,
): Promise<number> => {
const nc = await createConnection(flags);
const mc = nc.services.client();
const opts = options(flags);
// stats are grouped by endpoint, if this is a multi-endpoint map this so that each
// endpoint looks like a service
const stats: { num_requests: number }[] = [];
(await collect(await mc.stats(opts.name, opts.id))).forEach(
(s) => {
const { name, id, version } = s;
s.endpoints?.forEach((ne) => {
const line = Object.assign(ne, {
name,
endpoint: ne.name,
id,
version,
});
line.processing_time = millis(line.processing_time);
line.average_processing_time = millis(line.average_processing_time);
stats.push(line);
});
},
);

stats.sort((a, b) => {
return b.num_requests - a.num_requests;
});

if (stats.length) {
console.table(stats);
} else {
cmd.stdout("no services found");
}

await nc.close();
return 0;
},
});

adm.addCommand({
use: "info [--name] [--id]",
short: "get service info",
run: async (
cmd: Command,
_args: string[],
flags: Flags,
): Promise<number> => {
const nc = await createConnection(flags);
const mc = nc.services.client();
const opts = options(flags);

const infos = (await collect(await mc.info(opts.name, opts.id))).sort(
(a, b) => {
const A = `${a.name} ${a.version}`;
const B = `${b.name} ${b.version}`;
return B.localeCompare(A);
},
);
if (infos.length) {
console.table(infos);
} else {
cmd.stdout("no services found");
}
await nc.close();
return 0;
},
});

root.addCommand({
short: "start an instance of the echo service",
use: "start",
use: "echo-service [--server server:port]",
run: async (
_cmd: Command,
_args: string[],
flags: Flags,
): Promise<number> => {
const nc = await createConnection(flags);
const servers = flags.values<string>("server");
const nc = await connect({ servers });
const srv = await nc.services.add({
name: "echo",
version: "0.0.1",
Expand Down Expand Up @@ -204,88 +61,13 @@ root.addCommand({
},
});

const request = root.addCommand({
short: "make a request a service",
use: "request",
run: async (
_cmd: Command,
_args: string[],
flags: Flags,
): Promise<number> => {
const nc = await createConnection(flags);
const subj = flags.value<string>("subject");
console.log(`Sending request on "${subj}"`);
const payload = flags.value<string>("payload");
const json = flags.value<string>("json");
const timeout = flags.value<number>("timeout");
const start = Date.now();
const r = await nc.request(subj, StringCodec().encode(payload), {
timeout,
});
const time = Date.now() - start;
console.log(`Received with rtt ${time}ms`);

r.headers?.keys().forEach((k) => {
console.log(`${k}: ${r.headers?.get(k)}`);
});
console.log();
if (json) {
console.dir(r.json());
} else {
console.log(r.string());
}
console.log();

await nc.close();

return 0;
},
});

request.addFlag({
name: "subject",
type: "string",
usage: "subject to send the request on",
default: "echo",
});

request.addFlag({
name: "payload",
root.addFlag({
name: "server",
short: "s",
type: "string",
usage: "payload to send",
default: "hello service",
});

request.addFlag({
name: "timeout",
type: "number",
usage: "timeout in milliseconds",
default: 2000,
});

request.addFlag({
name: "json",
type: "boolean",
usage: "interpret response as JSON",
usage: "NATS server host:port",
default: "demo.nats.io",
persistent: true,
});

function createConnection(flags: Flags): Promise<NatsConnection> {
const servers = [flags.value<string>("server")];
return connect({ servers });
}

function options(
flags: Flags,
): { name?: string; id?: string } {
let name: string | undefined = flags.value<string>("name");
if (name === "") {
name = undefined;
}
let id: string | undefined = flags.value<string>("id");
if (id === "") {
id = undefined;
}
return { name, id };
}

Deno.exit(await root.execute(Deno.args));

0 comments on commit b02abeb

Please sign in to comment.