starfish-sdk 1.0.0

海星SDK
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83

use serde::{Deserialize, Serialize};

#[derive(Debug, Clone, Default, Serialize, Deserialize)]
pub struct LogConfig {
	#[serde(default = "default_log_level")]
	pub level: String,
	#[serde(default = "default_log_to_console")]
	pub console: bool,
	#[serde(default)]
	pub file: bool,
}

fn default_log_level() -> String {
	"info".into()
}

fn default_log_to_console() -> bool {
	true
}

#[cfg(feature = "web")]
#[derive(Debug, Clone, Default, Serialize, Deserialize)]
pub struct WebConfig {
	#[serde(default = "default_web_bind")]
	pub bind: String,
}
#[cfg(feature = "web")]
fn default_web_bind() -> String {
	"0.0.0.0:10000".into()
}

#[cfg(feature = "database")]
#[derive(Debug, Clone, Default, Serialize, Deserialize)]
pub struct DatabaseConfig {
	#[serde(default)]
	pub url: String,
	#[serde(default = "default_database_max_conn")]
	pub max_conn: u32,
	#[serde(default)]
	pub show_sql: bool,
}

#[cfg(feature = "database")]
fn default_database_max_conn() -> u32 {
	num_cpus::get() as u32
}

#[cfg(feature = "redis")]
#[derive(Debug, Clone, Default, Serialize, Deserialize)]
pub struct RedisConfig {
	/// redis://[<username>][:<password>@]<hostname>[:port][/<db>]
	#[serde(default)]
	pub url: String,
}

#[cfg(feature = "mongo")]
#[derive(Debug, Clone, Default, Serialize, Deserialize)]
pub struct MongoConfig {
	/// mongodb://user:pass@localhost:27017
	#[serde(default)]
	pub url: String,
}

#[cfg(feature = "influx")]
#[derive(Debug, Clone, Default, Serialize, Deserialize)]
pub struct InfluxConfig {
	#[serde(default)]
	pub url: String,
	#[serde(default)]
	pub org: String,
	#[serde(default)]
	pub token: String,
}

#[cfg(feature = "config")]
pub fn init<'de, T: Deserialize<'de>>(config_name: &str) -> anyhow::Result<T> {
	let settings = config::Config::builder()
		.add_source(config::File::with_name(config_name).required(false))
		.add_source(config::Environment::with_prefix("app").try_parsing(true).separator("__"))
		.build()?;
	let config: T = settings.try_deserialize()?;
	Ok(config)
}