wip

Author: keroro520

docker/.env.example

@@ -2,8 +2,12 @@
 
 # Used in https://github.com/timescale/timescaledb-docker
 POSTGRES_DB=ckb
-POSTGRES_USER=ckb
-POSTGRES_PASSWORD=ckb
+POSTGRES_USER=postgres
+POSTGRES_PASSWORD=postgres
+
+# Used in ckb-analyzer
+POSTGRES_HOST=host.docker.internal
+# POSTGRES_HOST=127.0.0.1
 
 # Used when render Grafana configuration file
 GRAFANA_DOMAIN=127.0.0.1

docker/Dockerfile

@@ -3,7 +3,7 @@ FROM ubuntu:20.04
 RUN apt update \
     && apt install -y wget
 
-RUN wget https://github.com/cryptape/ckb-analyzer/releases/download/v0.1.2/ckb-analyzer-linux-x86_64.tar.gz \
+RUN wget https://github.com/cryptape/ckb-analyzer/releases/download/rc-v0.1.3/ckb-analyzer-linux-x86_64.tar.gz \
     && tar xzf ckb-analyzer-linux-x86_64.tar.gz \
     && mv target/release/ckb-analyzer /bin/ckb-analyzer \
     && chmod +x /bin/ckb-analyzer \

docker/collector.yaml

@@ -41,12 +41,10 @@ services:
       - ckb-mainnet
     command: 
       - "/bin/ckb-analyzer"
-      - "--node.rpc"
+      - "--ckb-rpc-url"
       - "http://ckb-mainnet:8114"
-      - "--node.subscription"
-      - "127.0.0.1:18114"
-      - "--topics"
-      - "ChainCrawler"
+      - "--ckb-subscription-addr"
+      - "ckb-mainnet:18114"
     extra_hosts:
       # use `host.docker.internal` as host DNS name
       - "host.docker.internal:host-gateway"
@@ -59,12 +57,10 @@ services:
       - ckb-testnet
     command: 
       - "/bin/ckb-analyzer"
-      - "--node.rpc"
+      - "--ckb-rpc-url"
       - "http://ckb-testnet:8114"
-      - "--node.subscription"
-      - "127.0.0.1:18114"
-      - "--topics"
-      - "ChainCrawler"
+      - "--ckb-subscription-addr"
+      - "ckb-testnet:18114"
     extra_hosts:
       # use `host.docker.internal` as host DNS name
       - "host.docker.internal:host-gateway"

src/main.rs

@@ -5,7 +5,7 @@ use crate::topic::{
 };
 use crate::util::crossbeam_channel_to_tokio_channel;
 use ckb_testkit::{connector::SharedState, ConnectorBuilder, Node};
-use clap::{crate_version, value_t_or_exit, values_t_or_exit, App, Arg};
+use clap::{crate_version, values_t_or_exit, App, Arg};
 use std::env;
 use std::net::SocketAddr;
 use std::path::PathBuf;
@@ -25,10 +25,39 @@ async fn main() {
     log::info!("CKBAnalyzer starting");
 
     let matches = clap_app().get_matches();
-    let rpc_url = value_t_or_exit!(matches, "node.rpc", String);
-    let subscription_addr = value_t_or_exit!(matches, "node.subscription", SocketAddr);
+    let rpc_url = {
+        let raw = match matches.value_of("ckb-rpc-url") {
+            Some(raw) => raw.to_string(),
+            None => match env::var_os("CKB_RPC_URL") {
+                Some(raw) => raw.to_string_lossy().to_string(),
+                None => {
+                    panic!("Miss CKB Rpc url via neither --ckb-rpc-url nor environment variable \"CKB_RPC_URL\"");
+                }
+            },
+        };
+        let _ = url::Url::parse(&raw)
+            .map_err(|err| panic!("Invalid CKB RPC url, url: \"{}\", error: {:?}", raw, err));
+        raw
+    };
+    let subscription_addr = {
+        let raw = match matches.value_of("ckb-subscription-addr") {
+            Some(raw) => raw.to_string(),
+            None => match env::var_os("CKB_SUBSCRIPTION_ADDR") {
+                Some(raw) => raw.to_string_lossy().to_string(),
+                None => {
+                    panic!("Miss CKB subscription addr via neither --ckb-subscription-addr nor environment variable \"CKB_SUBSCRIPTION_ADDR\"");
+                }
+            },
+        };
+        raw.parse::<SocketAddr>().unwrap_or_else(|err| {
+            panic!(
+                "Invalid CKB subscription addr, addr: \"{}\", error: {:?}",
+                raw, err
+            )
+        })
+    };
     let topics = values_t_or_exit!(matches, "topics", String);
-    log::info!("CKB Node RPC: \"{}\"", rpc_url);
+    log::info!("CKB CKB RPC: \"{}\"", rpc_url);
     log::info!("CKB Node Subscription: \"{}\"", subscription_addr);
     log::info!("Topics: {:?}", topics);
 
@@ -317,28 +346,18 @@ pub fn clap_app() -> App<'static, 'static> {
                 }),
         )
         .arg(
-            Arg::with_name("node.rpc")
-                .long("node.rpc")
+            Arg::with_name("ckb-rpc-url")
+                .long("ckb-rpc-url")
                 .value_name("URL")
-                .required(true)
-                .takes_value(true)
-                .validator(|s| {
-                    url::Url::parse(&s)
-                        .map(|_| ())
-                        .map_err(|err| err.to_string())
-                }),
+                .required(false)
+                .takes_value(true),
         )
         .arg(
-            Arg::with_name("node.subscription")
-                .long("node.subscription")
-                .value_name("HOSTPORT")
-                .required(true)
-                .takes_value(true)
-                .validator(|s| {
-                    s.parse::<SocketAddr>()
-                        .map(|_| ())
-                        .map_err(|err| err.to_string())
-                }),
+            Arg::with_name("ckb-subscription-addr")
+                .long("ckb-subscription-addr")
+                .value_name("HOST:PORT")
+                .required(false)
+                .takes_value(true),
         )
         .arg(
             Arg::with_name("topics")