Merge pull request #2 from starcoinorg/schema

Implement schema for HashValue
starcoinorg · Mar 7, 2022 · d871dfb · d871dfb
2 parents b4887ba + 2c95387
commit d871dfb
Show file tree

Hide file tree

Showing 3 changed files with 236 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/diem-crypto/Cargo.toml b/crates/diem-crypto/Cargo.toml
@@ -35,6 +35,7 @@ aes-gcm = "0.8.0"
 diem-crypto-derive = { path = "../diem-crypto-derive", version = "0.0.3" }
 bcs = "0.1.2"
 libc = "^0.2"
+schemars = {git = "https://github.com/starcoinorg/schemars", rev= "6972da92f4360e1779168bb3fe0274c521d324e2"}
 
 [dev-dependencies]
 bitvec = "0.19.4"

diff --git a/crates/diem-crypto/src/hash.rs b/crates/diem-crypto/src/hash.rs
@@ -115,6 +115,9 @@ use std::{
  fmt,
  str::FromStr,
 };
+use schemars::gen::SchemaGenerator;
+use schemars::JsonSchema;
+use schemars::schema::{InstanceType, Schema, SchemaObject};
 #[cfg(not(feature = "avx512f"))]
 use tiny_keccak::{Hasher, Sha3};
 
@@ -777,6 +780,21 @@ impl<T: ser::Serialize + ?Sized> TestOnlyHash for T {
  hasher.finish()
  }
 }
+impl JsonSchema for HashValue {
+
+ fn schema_name() -> String {
+ "HashValue".to_owned()
+ }
+
+ fn json_schema(_: &mut SchemaGenerator) -> Schema {
+ SchemaObject {
+ instance_type: Some(InstanceType::String.into()),
+ format: Some("HashValue".to_owned()),
+ ..Default::default()
+ }
+ .into()
+ }
+}
 
 #[cfg(test)]
 mod tests {