fixed server message handle

WatcherAgent/src/docker/serverclientcomm.rs

@@ -5,7 +5,7 @@
 use crate::models::ServerMessage;
 
 use super::container::{restart_container, update_docker_image};
-use bollard::query_parameters::{CreateImageOptions, RestartContainerOptions};
+//use bollard::query_parameters::{CreateImageOptions, RestartContainerOptions};
 use bollard::Docker;
 use std::error::Error;
 
@@ -40,7 +40,7 @@ pub async fn handle_server_message(
             if let Some(image_name) = msg.data.get("image").and_then(|v| v.as_str()) {
                 println!("Received restart command for image: {}", image_name);
                 // Call your update_docker_image function here
-                update_docker_image(docker, image_name).await?;
+                restart_container(docker, image_name).await?;
                 Ok(())
             } else {
                 Err("Missing image name in update message".into())