Commit 2dfb18c

committed

Handle new added problems properly

1 parent f43e67d commit 2dfb18cCopy full SHA for 2dfb18c

File tree

+24

-21

lines changed

+24

-21

lines changed

Lines changed: 4 additions & 14 deletions

Original file line number	Diff line number	Diff line change
`@@ -41,13 +41,7 @@ impl Cache {`
`41`	`41`
`42`	`42`	`/// ref to download probems`
`43`	`43`	`pub fn update(self) -> Result<(), Error> {`
`44`		`- let c = conn((&self.0.conf.storage.cache()?).to_owned());`
`45`		`- let ps = self.download_problems()?;`
`46`		`- for i in ps.into_iter() {`
`47`		`- let target = problems.filter(id.eq(i.id));`
`48`		`- diesel::update(target).set(i.to_owned()).execute(&c)?;`
`49`		`- }`
`50`		`-`
	`44`	`+ self.download_problems()?;`
`51`	`45`	`Ok(())`
`52`	`46`	`}`
`53`	`47`
`@@ -68,13 +62,9 @@ impl Cache {`
`68`	`62`	`parser::problem(&mut ps, json)?;`
`69`	`63`	`}`
`70`	`64`
`71`		`- let count = self.get_problems()?.len();`
`72`		`- if count == 0 {`
`73`		`- ps.sort_by(\|a, b\| b.id.partial_cmp(&a.id).unwrap_or(std::cmp::Ordering::Equal));`
`74`		`- diesel::insert_into(problems)`
`75`		`- .values(&ps)`
`76`		`- .execute(&self.conn()?)?;`
`77`		`- }`
	`65`	`+ diesel::replace_into(problems)`
	`66`	`+ .values(&ps)`
	`67`	`+ .execute(&self.conn()?)?;`
`78`	`68`
`79`	`69`	`Ok(ps)`
`80`	`70`	`}`

Lines changed: 9 additions & 6 deletions

Original file line number	Diff line number	Diff line change
`@@ -100,13 +100,16 @@ impl Command for PickCommand {`
`100`	`100`	`crate::helper::filter(&mut problems, query.to_string());`
`101`	`101`	`}`
`102`	`102`
`103`		`- if let Some(id) = m.value_of("id") {`
`104`		`- problems.retain(\|x\| x.fid.to_string() == id);`
`105`		`- }`
`106`		`-`
`107`		`- let problem = &problems[rand::thread_rng().gen_range(0, problems.len())];`
	`103`	`+ let fid = m`
	`104`	`+ .value_of("id")`
	`105`	`+ .and_then(\|id\| id.parse::<i32>().ok())`
	`106`	`+ .unwrap_or_else(\|\| {`
	`107`	`+ // Pick random without specify id`
	`108`	`+ let problem = &problems[rand::thread_rng().gen_range(0, problems.len())];`
	`109`	`+ problem.fid`
	`110`	`+ });`
`108`	`111`
`109`		`- let r = cache.get_question(problem.fid);`
	`112`	`+ let r = cache.get_question(fid);`
`110`	`113`	`if r.is_err() {`
`111`	`114`	`let e = r.err()?;`
`112`	`115`	`eprintln!("{:?}", &e);`

Lines changed: 11 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,5 @@`
`1`	`1`	`//! Errors in leetcode-cli`
	`2`	`+use crate::cmds::{Command, DataCommand};`
`2`	`3`	`use colored::Colorize;`
`3`	`4`	`use std::fmt;`
`4`	`5`
`@@ -14,6 +15,7 @@ pub enum Error {`
`14`	`15`	`ScriptError(String),`
`15`	`16`	`CookieError,`
`16`	`17`	`DecryptError,`
	`18`	`+ SilentError,`
`17`	`19`	`}`
`18`	`20`
`19`	`21`	`impl std::fmt::Debug for Error {`
`@@ -40,6 +42,7 @@ impl std::fmt::Debug for Error {`
`40`	`42`	`Error::MatchError => write!(f, "{} Nothing matches", e),`
`41`	`43`	`Error::DecryptError => write!(f, "{} openssl decrypt failed", e),`
`42`	`44`	`Error::ScriptError(s) => write!(f, "{} {}", e, s),`
	`45`	`+ Error::SilentError => write!(f, ""),`
`43`	`46`	`}`
`44`	`47`	`}`
`45`	`48`	`}`
`@@ -61,7 +64,14 @@ impl std::convert::From<std::num::ParseIntError> for Error {`
`61`	`64`	`// sql`
`62`	`65`	`impl std::convert::From<diesel::result::Error> for Error {`
`63`	`66`	`fn from(err: diesel::result::Error) -> Self {`
`64`		`- Error::CacheError(err.to_string())`
	`67`	`+ match err {`
	`68`	`+ diesel::result::Error::NotFound => {`
	`69`	`+ println!("NotFound, you may update cache, and try it again\r\n");`
	`70`	`+ DataCommand::usage().print_help().unwrap_or(());`
	`71`	`+ Error::SilentError`
	`72`	`+ }`
	`73`	`+ _ => Error::CacheError(err.to_string()),`
	`74`	`+ }`
`65`	`75`	`}`
`66`	`76`	`}`
`67`	`77`

Comments

(0)