actions/install-action
1
0
Fork 0
mirror of https://github.com/taiki-e/install-action.git synced 2026-05-10 22:50:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

codegen: Use LazyLock again

Browse Source
This commit is contained in:
Taiki Endo
2024-08-12 05:02:13 +09:00
parent c995477ff1
commit 66b16da49c
1 changed files with 21 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

61
tools/codegen/src/main.rs
View File

@@ -7,7 +7,7 @@ use std::{
ffi::OsStr, ffi::OsStr,
io::Read, io::Read,
path::Path, path::Path,
sync::{OnceLock, RwLock}, sync::{LazyLock, RwLock},
time::Duration, time::Duration,
}; };
@@ -649,43 +649,38 @@ struct GitHubTokens {
other: RwLock<Option<String>>, other: RwLock<Option<String>>,
} }
impl GitHubTokens { impl GitHubTokens {
// TODO: Use std::sync::LazyLock once 1.80 is released fn get(&self, url: &str) -> Option<String> {
fn get_github_tokens() -> &'static GitHubTokens { if url.starts_with("https://raw.githubusercontent.com/") {
static GITHUB_TOKENS: OnceLock<GitHubTokens> = OnceLock::new(); self.raw.read().unwrap().clone()
GITHUB_TOKENS.get_or_init(|| { } else if url.starts_with("https://api.github.com/") {
self.api.read().unwrap().clone()
} else if url.starts_with("https://github.com/") {
self.other.read().unwrap().clone()
} else {
None
}
}
fn clear(&self, url: &str) {
if url.starts_with("https://raw.githubusercontent.com/") {
*self.raw.write().unwrap() = None;
} else if url.starts_with("https://api.github.com/") {
*self.api.write().unwrap() = None;
} else if url.starts_with("https://github.com/") {
*self.other.write().unwrap() = None;
}
}
}
static GITHUB_TOKENS: LazyLock<GitHubTokens> = LazyLock::new(|| {
let token = env::var("GITHUB_TOKEN").ok().filter(|v| !v.is_empty()); let token = env::var("GITHUB_TOKEN").ok().filter(|v| !v.is_empty());
GitHubTokens { GitHubTokens {
raw: RwLock::new(token.clone()), raw: RwLock::new(token.clone()),
api: RwLock::new(token.clone()), api: RwLock::new(token.clone()),
other: RwLock::new(token), other: RwLock::new(token),
} }
}) });
}
fn get(url: &str) -> Option<String> {
if url.starts_with("https://raw.githubusercontent.com/") {
Self::get_github_tokens().raw.read().unwrap().clone()
} else if url.starts_with("https://api.github.com/") {
Self::get_github_tokens().api.read().unwrap().clone()
} else if url.starts_with("https://github.com/") {
Self::get_github_tokens().other.read().unwrap().clone()
} else {
None
}
}
fn clear(url: &str) {
if url.starts_with("https://raw.githubusercontent.com/") {
*Self::get_github_tokens().raw.write().unwrap() = None;
} else if url.starts_with("https://api.github.com/") {
*Self::get_github_tokens().api.write().unwrap() = None;
} else if url.starts_with("https://github.com/") {
*Self::get_github_tokens().other.write().unwrap() = None;
}
}
}
fn download(url: &str) -> Result<ureq::Response> { fn download(url: &str) -> Result<ureq::Response> {
let mut token = GitHubTokens::get(url); let mut token = GITHUB_TOKENS.get(url);
let mut retry = 0; let mut retry = 0;
let mut retry_time = 0; let mut retry_time = 0;
let mut max_retry = 6; let mut max_retry = 6;
@@ -707,7 +702,7 @@ fn download(url: &str) -> Result<ureq::Response> {
retry_time = 0; retry_time = 0;
token = None; token = None;
// rate limit // rate limit
GitHubTokens::clear(url); GITHUB_TOKENS.clear(url);
} }
retry += 1; retry += 1;
if retry > max_retry { if retry > max_retry {
@@ -721,7 +716,7 @@ fn download(url: &str) -> Result<ureq::Response> {
fn github_head(url: &str) -> Result<()> { fn github_head(url: &str) -> Result<()> {
eprintln!("fetching head of {url} .."); eprintln!("fetching head of {url} ..");
let mut token = GitHubTokens::get(url); let mut token = GITHUB_TOKENS.get(url);
let mut retry = 0; let mut retry = 0;
let mut retry_time = 0; let mut retry_time = 0;
let mut max_retry = 2; let mut max_retry = 2;
@@ -744,7 +739,7 @@ fn github_head(url: &str) -> Result<()> {
if token.is_some() && retry == max_retry / 2 { if token.is_some() && retry == max_retry / 2 {
retry_time = 0; retry_time = 0;
token = None; token = None;
GitHubTokens::clear(url); GITHUB_TOKENS.clear(url);
} }
retry += 1; retry += 1;
if retry > max_retry { if retry > max_retry {