|
@@ -69,22 +69,27 @@ impl Parse for LogArgs {
|
|
|
}
|
|
|
|
|
|
pub(crate) fn log(args: LogArgs, level: Option<TokenStream>) -> Result<TokenStream> {
|
|
|
- let ctx = args.ctx;
|
|
|
- let target = match args.target {
|
|
|
+ let LogArgs {
|
|
|
+ ctx,
|
|
|
+ target,
|
|
|
+ level: level_expr,
|
|
|
+ format_string,
|
|
|
+ formatting_args,
|
|
|
+ } = args;
|
|
|
+ let target = match target {
|
|
|
Some(t) => quote! { #t },
|
|
|
None => quote! { module_path!() },
|
|
|
};
|
|
|
- let lvl: TokenStream = if let Some(l) = level {
|
|
|
- l
|
|
|
- } else if let Some(l) = args.level {
|
|
|
- quote! { #l }
|
|
|
- } else {
|
|
|
- return Err(Error::new(
|
|
|
- args.format_string.span(),
|
|
|
- "missing `level` argument: try passing an `aya_log_ebpf::Level` value",
|
|
|
- ));
|
|
|
+ let level = match level {
|
|
|
+ Some(l) => l,
|
|
|
+ None => {
|
|
|
+ let l = level_expr.ok_or(Error::new(
|
|
|
+ format_string.span(),
|
|
|
+ "missing `level` argument: try passing an `aya_log_ebpf::Level` value",
|
|
|
+ ))?;
|
|
|
+ quote! { #l }
|
|
|
+ }
|
|
|
};
|
|
|
- let format_string = args.format_string;
|
|
|
|
|
|
let format_string_val = format_string.value();
|
|
|
let fragments = parse(&format_string_val).map_err(|e| {
|
|
@@ -101,7 +106,7 @@ pub(crate) fn log(args: LogArgs, level: Option<TokenStream>) -> Result<TokenStre
|
|
|
match fragment {
|
|
|
Fragment::Literal(s) => values.push(quote!(#s)),
|
|
|
Fragment::Parameter(p) => {
|
|
|
- let arg = match args.formatting_args {
|
|
|
+ let arg = match formatting_args {
|
|
|
Some(ref args) => args[arg_i].clone(),
|
|
|
None => return Err(Error::new(format_string.span(), "no arguments provided")),
|
|
|
};
|
|
@@ -153,7 +158,7 @@ pub(crate) fn log(args: LogArgs, level: Option<TokenStream>) -> Result<TokenStre
|
|
|
let #size = ::aya_log_ebpf::write_record_header(
|
|
|
#buf,
|
|
|
#target,
|
|
|
- #lvl,
|
|
|
+ #level,
|
|
|
module_path!(),
|
|
|
file!(),
|
|
|
line!(),
|