Method hook_predis_execute_command

src/plugin/plugin_predis.rs:181–267

(
        &self, class_name: &str, _function_name: &str,
    )

impl PredisPlugin {
    fn hook_predis_execute_command(
        &self, class_name: &str, _function_name: &str,
    ) -> (Box<BeforeExecuteHook>, Box<AfterExecuteHook>) {
        let class_name = class_name.to_owned();
        (
            Box::new(move |request_id, execute_data| {
                validate_num_args(execute_data, 1)?;

                let function_name = execute_data
                    .get_parameter(0)
                    .as_z_str()
                    .and_then(|s| s.to_str().ok())
                    .unwrap_or("")
                    .to_string();

                let cmd = function_name.to_uppercase();

                if !REDIS_ALL_COMMANDS.contains(&*cmd) {
                    return Ok(Box::new(()));
                }

                let this = get_this_mut(execute_data)?;
                let handle = this.handle();
                let connection = this.call("getConnection", [])?;

                let peer = Self::get_peer(connection)?;

                let op = if REDIS_READ_COMMANDS.contains(&*cmd) {
                    Some("read")
                } else if REDIS_WRITE_COMMANDS.contains(&*cmd) {
                    Some("write")
                } else {
                    None
                };

                let key = op
                    .and_then(|_| execute_data.get_parameter(1).as_z_arr())
                    .and_then(|params| params.get(0))
                    .and_then(|key| key.as_z_str())
                    .and_then(|s| s.to_str().ok());

                debug!(handle, cmd, key, op, "call redis command");

                let mut span =
                    RequestContext::try_with_global_ctx(request_id, |ctx| {
                        Ok(ctx.create_exit_span(
                            &format!("{}->{}", class_name, &function_name,),
                            &peer,
                        ))
                    })?;

                let span_object = span.span_object_mut();
                span_object.set_span_layer(SpanLayer::Cache);
                span_object.component_id = COMPONENT_PHP_PREDIS_ID;
                span_object.add_tag(TAG_CACHE_TYPE, "redis");
                span_object.add_tag(TAG_CACHE_CMD, cmd);
                if let Some(op) = op {
                    span_object.add_tag(TAG_CACHE_OP, op);