faasd-in-rust/crates/provider/src/handlers/invoke_resolver.rs

use crate::consts::DEFAULT_FUNCTION_NAMESPACE;
use crate::handlers::function_get::get_function;
use actix_web::{Error, error::ErrorInternalServerError, error::ErrorServiceUnavailable};
use log;
use url::Url;

#[derive(Clone)]
pub struct InvokeResolver;

impl InvokeResolver {
    pub async fn resolve_function_url(function_name: &str) -> Result<Url, Error> {
        //根据函数名和containerd获取函数ip，
        //从函数名称中提取命名空间。如果函数名称中包含 .，则将其后的部分作为命名空间；否则使用默认命名空间

        let mut actual_function_name = function_name;
        let namespace =
            extract_namespace_from_function_or_default(function_name, DEFAULT_FUNCTION_NAMESPACE);
        if function_name.contains('.') {
            actual_function_name = function_name.trim_end_matches(&format!(".{}", namespace));
        }

        let function = match get_function(actual_function_name, &namespace).await {
            Ok(function) => function,
            Err(e) => {
                log::error!("Failed to get function:{}", e);
                return Err(ErrorServiceUnavailable("Failed to get function"));
            }
        };
        log::info!("Function:{:?}", function);

        let address = function.address.clone();
        let urlstr = format!("http://{}", address);
        match Url::parse(&urlstr) {
            Ok(url) => Ok(url),
            Err(e) => {
                log::error!("Failed to resolve url:{}", e);
                Err(ErrorInternalServerError("Failed to resolve URL"))
            }
        }
    }
}

fn extract_namespace_from_function_or_default(
    function_name: &str,
    default_namespace: &str,
) -> String {
    let mut namespace = default_namespace.to_string();
    if function_name.contains('.') {
        if let Some(index) = function_name.rfind('.') {
            namespace = function_name[index + 1..].to_string();
        }
    }
    namespace
}