*: update, add genesis.rs

Signed-off-by: Gyuho Lee <[email protected]>

Author: gyuho

Cargo.toml

@@ -19,8 +19,14 @@ clap = { version = "3.1.8", features = ["cargo", "derive"] }
 env_logger = "0.9.0"
 log = "0.4.16"
 prost = "0.10.0"
-tokio = { version = "1.17.0", features = ["rt-multi-thread"] }
+prost-types = "0.10.0"
+serde = { version = "1.0.136", features = ["derive"] }
+serde_json = "1.0.79"
+serde_yaml = "0.8.23"
+tokio = { version = "1.17.0", features = ["fs", "rt-multi-thread"] }
+tokio-stream = { version = "0.1.8", features = ["net"] }
 tonic = "0.7.1"
+tonic-health = "0.6.0"
 
 [build-dependencies]
 # ref. https://github.com/hyperium/tonic/tree/master/tonic-build

build.rs

@@ -3,15 +3,6 @@ fn main() {
     tonic_build::configure()
         .build_server(true)
         .build_client(false)
-        .compile(
-            &[
-                // e.g.,
-                // git submodule add https://github.com/prometheus/client_model
-                // git submodule update --remote
-                "client_model/io/prometheus/client/metrics.proto",
-                "proto/vm.proto",
-            ],
-            &["client_model", "proto"],
-        )
+        .compile(&["proto/metrics.proto", "proto/vm.proto"], &["proto"])
         .unwrap();
 }

client_model

@@ -0,0 +1,94 @@
+// copied from https://github.com/prometheus/client_model/tree/master/io/prometheus/client
+
+// Copyright 2013 Prometheus Team
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+syntax = "proto2";
+
+package metrics;
+option java_package = "io.prometheus.client";
+option go_package = "github.com/prometheus/client_model/go;io_prometheus_client";
+
+import "google/protobuf/timestamp.proto";
+
+message LabelPair {
+  optional string name  = 1;
+  optional string value = 2;
+}
+
+enum MetricType {
+  COUNTER    = 0;
+  GAUGE      = 1;
+  SUMMARY    = 2;
+  UNTYPED    = 3;
+  HISTOGRAM  = 4;
+}
+
+message Gauge {
+  optional double value = 1;
+}
+
+message Counter {
+  optional double   value    = 1;
+  optional Exemplar exemplar = 2;
+}
+
+message Quantile {
+  optional double quantile = 1;
+  optional double value    = 2;
+}
+
+message Summary {
+  optional uint64   sample_count = 1;
+  optional double   sample_sum   = 2;
+  repeated Quantile quantile     = 3;
+}
+
+message Untyped {
+  optional double value = 1;
+}
+
+message Histogram {
+  optional uint64 sample_count = 1;
+  optional double sample_sum   = 2;
+  repeated Bucket bucket       = 3; // Ordered in increasing order of upper_bound, +Inf bucket is optional.
+}
+
+message Bucket {
+  optional uint64   cumulative_count = 1; // Cumulative in increasing order.
+  optional double   upper_bound      = 2; // Inclusive.
+  optional Exemplar exemplar         = 3;
+}
+
+message Exemplar {
+  repeated LabelPair                 label     = 1;
+  optional double                    value     = 2;
+  optional google.protobuf.Timestamp timestamp = 3; // OpenMetrics-style.
+}
+
+message Metric {
+  repeated LabelPair label        = 1;
+  optional Gauge     gauge        = 2;
+  optional Counter   counter      = 3;
+  optional Summary   summary      = 4;
+  optional Untyped   untyped      = 5;
+  optional Histogram histogram    = 7;
+  optional int64     timestamp_ms = 6;
+}
+
+message MetricFamily {
+  optional string     name   = 1;
+  optional string     help   = 2;
+  optional MetricType type   = 3;
+  repeated Metric     metric = 4;
+}

proto/metrics.proto

@@ -5,9 +5,14 @@ syntax = "proto3";
 package vm;
 
 import "google/protobuf/empty.proto";
-import "io/prometheus/client/metrics.proto";
 
-option go_package = "github.com/ava-labs/avalanchego/proto/pb/vm";
+// THIS DOES NOT WORK:
+// git submodule add https://github.com/prometheus/client_model
+// git submodule update --remote
+//
+// original path converts to "super::io::prometheus::client::MetricFamily"
+// which is invalid in Rust
+import "metrics.proto";
 
 service VM {
   rpc Initialize(InitializeRequest) returns (InitializeResponse);
@@ -248,5 +253,5 @@ message GetBlockIDAtHeightResponse {
 }
 
 message GatherResponse {
-  repeated io.prometheus.client.MetricFamily metric_families = 1;
+  repeated metrics.MetricFamily metric_families = 1;
 }

proto/vm.proto

@@ -1,6 +1,10 @@
+use std::io;
+
 use clap::{crate_version, Arg, Command};
 use log::info;
 
+use mini_kvvm_rs::genesis;
+
 pub const APP_NAME: &str = "mini-kvvm-rs";
 
 fn main() {
@@ -19,6 +23,7 @@ fn main() {
                 .allow_invalid_utf8(false)
                 .default_value("info"),
         )
+        .subcommands(vec![command_genesis()])
         .get_matches();
 
     let log_level = matches.value_of("LOG_LEVEL").unwrap_or("info").to_string();
@@ -28,5 +33,58 @@ fn main() {
         env_logger::Env::default().filter_or(env_logger::DEFAULT_FILTER_ENV, log_level),
     );
 
+    if let Some(("genesis", sub_matches)) = matches.subcommand() {
+        let author = sub_matches.value_of("AUTHOR").unwrap_or("");
+        let msg = sub_matches.value_of("WELCOME_MESSAGE").unwrap_or("");
+        let p = sub_matches.value_of("GENESIS_FILE_PATH").unwrap_or("");
+        execute_genesis(author, msg, p).unwrap();
+        return;
+    }
+
     info!("starting mini-kvvm-rs");
+    // TODO
+}
+
+pub fn command_genesis() -> Command<'static> {
+    Command::new("genesis")
+        .about("Generates the genesis file")
+        .arg(
+            Arg::new("AUTHOR")
+                .long("author")
+                .short('a')
+                .help("author of the genesis")
+                .required(true)
+                .takes_value(true)
+                .allow_invalid_utf8(false)
+                .default_value("subnet creator"),
+        )
+        .arg(
+            Arg::new("WELCOME_MESSAGE")
+                .long("welcome-message")
+                .short('m')
+                .help("message field in genesis")
+                .required(true)
+                .takes_value(true)
+                .allow_invalid_utf8(false)
+                .default_value("hi"),
+        )
+        .arg(
+            Arg::new("GENESIS_FILE_PATH")
+                .long("genesis-file-path")
+                .short('p')
+                .help("file path to save genesis file")
+                .required(true)
+                .takes_value(true)
+                .allow_invalid_utf8(false),
+        )
+}
+
+pub fn execute_genesis(author: &str, msg: &str, p: &str) -> io::Result<()> {
+    let g = genesis::Genesis {
+        author: String::from(author),
+        welcome_message: String::from(msg),
+    };
+    g.sync(p)?;
+
+    Ok(())
 }

src/bin/mini-kvvm-rs/main.rs

@@ -1,3 +1,75 @@
+use std::{
+    fmt,
+    fs::{self, File},
+    io::{self, Error, ErrorKind, Write},
+    path::Path,
+};
+
+use log::info;
+use serde::{Deserialize, Serialize};
+
+#[derive(Debug, Serialize, Deserialize, Eq, PartialEq, Clone)]
 pub struct Genesis {
-    pub data: String,
+    pub author: String,
+    pub welcome_message: String,
+}
+
+impl Default for Genesis {
+    fn default() -> Self {
+        Self::default()
+    }
+}
+
+impl Genesis {
+    pub fn default() -> Self {
+        Self {
+            author: String::from("subnet creator"),
+            welcome_message: String::from("Hello from Rust VM!"),
+        }
+    }
+
+    pub fn from_json<S>(d: S) -> io::Result<Self>
+    where
+        S: AsRef<[u8]>,
+    {
+        let resp: Self = match serde_json::from_slice(d.as_ref()) {
+            Ok(p) => p,
+            Err(e) => {
+                return Err(Error::new(
+                    ErrorKind::Other,
+                    format!("failed to decode {}", e),
+                ));
+            }
+        };
+        Ok(resp)
+    }
+
+    pub fn sync(&self, file_path: &str) -> io::Result<()> {
+        info!("syncing genesis to '{}'", file_path);
+        let path = Path::new(file_path);
+        let parent_dir = path.parent().unwrap();
+        fs::create_dir_all(parent_dir)?;
+
+        let ret = serde_json::to_vec(&self);
+        let d = match ret {
+            Ok(d) => d,
+            Err(e) => {
+                return Err(Error::new(
+                    ErrorKind::Other,
+                    format!("failed to serialize genesis info to YAML {}", e),
+                ));
+            }
+        };
+        let mut f = File::create(&file_path)?;
+        f.write_all(&d)?;
+
+        Ok(())
+    }
+}
+
+impl fmt::Display for Genesis {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        let s = serde_yaml::to_string(&self).unwrap();
+        write!(f, "{}", s)
+    }
 }

src/genesis.rs

@@ -1 +1,10 @@
 pub mod genesis;
+pub mod plugin;
+
+pub mod vm {
+    tonic::include_proto!("vm");
+}
+
+pub mod metrics {
+    tonic::include_proto!("metrics");
+}