Endpoint: support CLI parameters

flashbake · Jul 1, 2022 · 4e977c8 · 4e977c8
1 parent 3b3a80b
commit 4e977c8
Show file tree

Hide file tree

Showing 7 changed files with 495 additions and 185 deletions.
diff --git a/charts/flashbake-endpoint/templates/deployment.yaml b/charts/flashbake-endpoint/templates/deployment.yaml
@@ -39,9 +39,16 @@ spec:
               protocol: TCP
           resources:
             {{- toYaml .Values.resources | nindent 12 }}
-          env:
-          - name: TEZOS_RPC_URL
-            value: {{ .Values.tezos_rpc_url }}
+          command:
+          - node
+          - app.js
+          - start
+          - --relay_listener_port
+          - "{{ .Values.relay_listener_port }}"
+          - --tezos_rpc_url
+          - {{ .Values.tezos_rpc_url }}
+          - --baker_listener_port
+          - "{{ .Values.baker_listener_port }}"
       {{- with .Values.nodeSelector }}
       nodeSelector:
         {{- toYaml . | nindent 8 }}

diff --git a/charts/flashbake-endpoint/values.yaml b/charts/flashbake-endpoint/values.yaml
@@ -1,6 +1,15 @@
 # URL to talk to Tezos RPC API
 tezos_rpc_url: http://localhost:8732
 
+# port where endpoint listens to relay connections
+relay_listener_port: 11732
+
+# port where endpoint listens to connections from the
+# baker. --operations-pool argument of the baker should
+# target this endpoint.
+baker_listener_port: 12732
+
+
 replicaCount: 1
 
 image:

diff --git a/flashbake-endpoint-values-0.yaml b/flashbake-endpoint-values-0.yaml
@@ -1,4 +1,7 @@
 tezos_rpc_url: http://flashbake-baker-0:8732/
+relay_listener_port: 11732
+baker_listener_port: 12732
+
 image:
   repository: localhost/flashbake
   pullPolicy: IfNotPresent

diff --git a/flashbake-endpoint-values-1.yaml b/flashbake-endpoint-values-1.yaml
@@ -1,4 +1,7 @@
 tezos_rpc_url: http://flashbake-baker-1:8732/
+relay_listener_port: 11732
+baker_listener_port: 12732
+
 image:
   repository: localhost/flashbake
   pullPolicy: IfNotPresent

diff --git a/flashbake/app.ts b/flashbake/app.ts
@@ -1,7 +1,7 @@
 import express from 'express';
 import { Mempool, InMemoryMempool } from "@flashbake/relay";
 import { HttpBakerEndpoint } from '@flashbake/baker-endpoint';
-
+import yargs, { Argv } from "yargs";
 
 function startBakerEndpoint(relayListenerPort: number, bakerListenerPort: number, rpcApiUrl: string): HttpBakerEndpoint {
   const relayFacingApp = express();
@@ -19,13 +19,25 @@ function startBakerEndpoint(relayListenerPort: number, bakerListenerPort: number
   return baker;
 }
 
-function main() {
-  const relayListenerPort = 11732;
-  const bakerListenerPort = 12732;
-  const rpcApiUrl = process.env["TEZOS_RPC_URL"] || '';
+async function main() {
+  let argv = await yargs
+    .command('start', "Start flashbake-endpoint.", (yargs: Argv) => {
+      return yargs.option('relay_listener_port', {
+        describe: "Relay listener port",
+        type: "number",
+        demandOption: true,
+      }).option('tezos_rpc_url', {
+        describe: "Tezos node RPC API URL",
+        type: "string",
+        demandOption: true,
+      }).option('baker_listener_port', {
+        describe: "Baker listener port",
+        type: "number",
+        demandOption: true,
+      })
+    }).argv;
 
-  console.debug(`Using RPC API URL ${rpcApiUrl}`);
-  startBakerEndpoint(relayListenerPort, bakerListenerPort, rpcApiUrl);
+  startBakerEndpoint(argv.relay_listener_port, argv.baker_listener_port, argv.tezos_rpc_url);
 }
 
 main();