feat: add more instruments

server/src/api.rs

@@ -260,6 +260,7 @@ pub struct PipelineStatus {
     pub available_servers: u64,
 }
 
+#[tracing::instrument(skip(pool))]
 pub async fn pipeline_status(pool: DbPool) -> anyhow::Result<Vec<PipelineStatus>> {
     let mut conn = pool
         .get()
@@ -313,6 +314,7 @@ pub async fn pipeline_status(pool: DbPool) -> anyhow::Result<Vec<PipelineStatus>
     Ok(res)
 }
 
+#[tracing::instrument(skip(pool))]
 pub async fn worker_status(pool: DbPool) -> anyhow::Result<Vec<Worker>> {
     let mut conn = pool
         .get()

server/src/bot.rs

@@ -10,14 +10,13 @@ use buildit_utils::github::{get_archs, OpenPRError, OpenPRRequest};
 use chrono::Local;
 use diesel::{Connection, ExpressionMethods, OptionalExtension, QueryDsl, RunQueryDsl};
 use serde::Deserialize;
-use tracing::warn;
-
 use std::borrow::Cow;
 use teloxide::{
     prelude::*,
     types::{ChatAction, ParseMode},
     utils::command::BotCommands,
 };
+use tracing::{info_span, warn, Instrument};
 
 #[derive(BotCommands, Clone, Debug)]
 #[command(
@@ -66,6 +65,7 @@ fn handle_archs_args(archs: Vec<&str>) -> Vec<&str> {
     archs
 }
 
+#[tracing::instrument(skip(pool))]
 async fn status(pool: DbPool) -> anyhow::Result<String> {
     let mut res = String::from("__*Queue Status*__\n\n");
 
@@ -676,6 +676,7 @@ pub async fn answer(bot: Bot, msg: Message, cmd: Command, pool: DbPool) -> Respo
     Ok(())
 }
 
+#[tracing::instrument(skip(bot, msg))]
 async fn bot_send_message_handle_length(
     bot: &Bot,
     msg: &Message,

server/src/github.rs

@@ -14,6 +14,7 @@ pub struct GithubToken {
     pub token_type: String,
 }
 
+#[tracing::instrument(skip(msg, arguments))]
 pub async fn login_github(
     msg: &Message,
     arguments: String,
@@ -67,6 +68,7 @@ pub fn get_packages_from_pr(pr: &PullRequest) -> Vec<String> {
 }
 
 /// Create octocrab instance authenticated as github installation
+#[tracing::instrument]
 pub async fn get_crab_github_installation() -> anyhow::Result<Option<Octocrab>> {
     if let Some(id) = ARGS
         .github_app_id

server/src/schema.rs

@@ -66,9 +66,4 @@ diesel::table! {
 diesel::joinable!(jobs -> pipelines (pipeline_id));
 diesel::joinable!(pipelines -> users (creator_user_id));
 
-diesel::allow_tables_to_appear_in_same_query!(
-    jobs,
-    pipelines,
-    users,
-    workers,
-);
+diesel::allow_tables_to_appear_in_same_query!(jobs, pipelines, users, workers,);