aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/atom.rs11
-rw-r--r--src/main.rs1
-rw-r--r--src/query/issues.rs10
3 files changed, 13 insertions, 9 deletions
diff --git a/src/atom.rs b/src/atom.rs
index 2894ccb..a3d7e10 100644
--- a/src/atom.rs
+++ b/src/atom.rs
@@ -45,15 +45,15 @@ fn entity_escape(from: &str) -> String {
}
async fn query_issues_for_label<'conn>(conn: &'conn mut Conn,
- repo_id: i64, label: &str, only_open: bool) -> impl Stream<Item=sqlx::Result<Issue>> + 'conn {
+ repo_id: i64, label: &str, state_mask: i64) -> impl Stream<Item=sqlx::Result<Issue>> + 'conn {
sqlx::query_as::<_, Issue>(r#"
SELECT issues.number, state, title, body, user_login, html_url, updated_at FROM issues
INNER JOIN is_labeled ON is_labeled.issue=issues.number
WHERE is_labeled.label=(SELECT id FROM labels WHERE repo=? AND name=?)
- AND (?=0 OR issues.state=?)
+ AND issues.state & ? != 0
ORDER BY issues.number DESC
"#).bind(repo_id).bind(label)
- .bind(only_open).bind(query::issues::IssueState::OPEN.to_integer())
+ .bind(state_mask)
.fetch(conn)
}
@@ -124,6 +124,9 @@ pub async fn generate(mut conn: &mut Conn, (ref owner, ref name): (String, Strin
let repo_id = repo_id(&mut conn, owner, name).await?;
+ let mut state_mask = !0;
+ if only_open { state_mask &= !query::issues::IssueState::CLOSED.to_integer(); }
+
for label in labels {
info!("atom for {:?}", label);
@@ -148,7 +151,7 @@ pub async fn generate(mut conn: &mut Conn, (ref owner, ref name): (String, Strin
.map_err(anyhow::Error::msg)?
]);
- let issues: Vec<Issue> = query_issues_for_label(&mut conn, repo_id, &label, only_open).await
+ let issues: Vec<Issue> = query_issues_for_label(&mut conn, repo_id, &label, state_mask).await
.filter_map(|res| async { res.ok() })
.collect().await;
diff --git a/src/main.rs b/src/main.rs
index e544da1..7b352bd 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -61,6 +61,7 @@ async fn init_db(conn: &mut Conn) {
updated_at integer,
PRIMARY KEY (repo, number)
);
+ CREATE INDEX IF NOT EXISTS issues_state ON issues (repo, number, state);
CREATE TABLE IF NOT EXISTS labels(
id integer PRIMARY KEY,
diff --git a/src/query/issues.rs b/src/query/issues.rs
index 525ebe5..7221858 100644
--- a/src/query/issues.rs
+++ b/src/query/issues.rs
@@ -25,17 +25,17 @@ pub use issues_query::IssueState;
impl IssueState {
pub fn from_integer(i: i64) -> Option<Self> {
match i {
- 0 => Some(Self::OPEN),
- 1 => Some(Self::CLOSED),
+ 1 => Some(Self::OPEN),
+ 2 => Some(Self::CLOSED),
_ => None
}
}
pub fn to_integer(&self) -> i64 {
match self {
- Self::OPEN => 0,
- Self::CLOSED => 1,
- Self::Other(_) => 2
+ Self::OPEN => 1,
+ Self::CLOSED => 2,
+ Self::Other(_) => 0
}
}