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
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124

use std::fmt::{self, Display, Formatter};
use std::sync::Arc;

use futures_util::lock::Mutex;

use crate::extensions::{
    Extension, ExtensionContext, ExtensionFactory, NextParseQuery, NextResolve, ResolveInfo,
};
use crate::parser::types::{ExecutableDocument, OperationType, Selection};
use crate::{PathSegment, ServerError, ServerResult, Value, Variables};

/// Logger extension
#[cfg_attr(docsrs, doc(cfg(feature = "log")))]
pub struct Logger;

impl ExtensionFactory for Logger {
    fn create(&self) -> Arc<dyn Extension> {
        Arc::new(LoggerExtension {
            inner: Mutex::new(Inner {
                enabled: true,
                query: String::new(),
                variables: Default::default(),
            }),
        })
    }
}

struct Inner {
    enabled: bool,
    query: String,
    variables: Variables,
}

struct LoggerExtension {
    inner: Mutex<Inner>,
}

#[async_trait::async_trait]
impl Extension for LoggerExtension {
    async fn parse_query(
        &self,
        ctx: &ExtensionContext<'_>,
        query: &str,
        variables: &Variables,
        next: NextParseQuery<'_>,
    ) -> ServerResult<ExecutableDocument> {
        let mut inner = self.inner.lock().await;
        inner.query = query.replace(char::is_whitespace, "");
        inner.variables = variables.clone();
        let document = next.run(ctx, query, variables).await?;
        let is_schema = document
            .operations
            .iter()
            .filter(|(_, operation)| operation.node.ty == OperationType::Query)
            .any(|(_, operation)| operation.node.selection_set.node.items.iter().any(|selection| matches!(&selection.node, Selection::Field(field) if field.node.name.node == "__schema")));
        inner.enabled = !is_schema;
        Ok(document)
    }

    async fn resolve(
        &self,
        ctx: &ExtensionContext<'_>,
        info: ResolveInfo<'_>,
        next: NextResolve<'_>,
    ) -> ServerResult<Option<Value>> {
        let enabled = self.inner.lock().await.enabled;
        if enabled {
            let path = info.path_node.to_string();
            log::trace!(target: "async-graphql", "[ResolveStart] path: \"{}\"", path);
            let res = next.run(ctx, info).await;
            if let Err(err) = &res {
                let inner = self.inner.lock().await;
                log::error!(
                    target: "async-graphql",
                    "{}",
                    DisplayError { query:&inner.query,variables:&inner.variables, e: &err }
                );
            }
            log::trace!(target: "async-graphql", "[ResolveEnd] path: \"{}\"", path);
            res
        } else {
            next.run(ctx, info).await
        }
    }
}

struct DisplayError<'a> {
    query: &'a str,
    variables: &'a Variables,
    e: &'a ServerError,
}
impl<'a> Display for DisplayError<'a> {
    fn fmt(&self, f: &mut Formatter) -> fmt::Result {
        write!(f, "[Error] ")?;

        if !self.e.path.is_empty() {
            write!(f, "path: ")?;
            for (i, segment) in self.e.path.iter().enumerate() {
                if i != 0 {
                    write!(f, ".")?;
                }

                match segment {
                    PathSegment::Field(field) => write!(f, "{}", field),
                    PathSegment::Index(i) => write!(f, "{}", i),
                }?;
            }
            write!(f, ", ")?;
        }
        if !self.e.locations.is_empty() {
            write!(f, "pos: [")?;
            for (i, location) in self.e.locations.iter().enumerate() {
                if i != 0 {
                    write!(f, ", ")?;
                }
                write!(f, "{}:{}", location.line, location.column)?;
            }
            write!(f, "], ")?;
        }
        write!(f, r#"query: "{}", "#, self.query)?;
        write!(f, "variables: {}", self.variables)?;
        write!(f, "{}", self.e.message)
    }
}