Send error

problem_06/src/connection.rs

@@ -8,6 +8,7 @@ use tokio::{
 
 use crate::frame::{self, ClientFrames, ServerFrames};
 
+#[derive(PartialEq)]
 pub(crate) enum ConnectionType {
 	Camera,
 	Dispatcher,

problem_06/src/server.rs

@@ -180,6 +180,9 @@ impl Handler {
 	) -> crate::Result<()> {
 		match frame {
 			ClientFrames::Plate { plate, timestamp } => {
+				if self.connection_type.is_some()
+					&& self.connection_type == Some(ConnectionType::Camera)
+				{
 					info!("Receive new plate: {plate} at {timestamp}");
 					issue_possible_ticket(
 						db,
@@ -190,6 +193,14 @@ impl Handler {
 						CameraId(self.connection.get_address()),
 					)
 					.await;
+				} else {
+					let _ = send_message
+						.send(ServerFrames::Error {
+							msg: "Not connected as camera".to_string(),
+						})
+						.await;
+					return Err("Already connected".into());
+				}
 			}
 			ClientFrames::WantHeartbeat { interval } => {
 				if interval > 0 {
@@ -204,7 +215,7 @@ impl Handler {
 				if self.connection_type.is_some() {
 					let _ = send_message
 						.send(ServerFrames::Error {
-							msg: "Already connected as camera".to_string(),
+							msg: "Already connected as a connection type".to_string(),
 						})
 						.await;
 					return Err("Already connected".into());
@@ -224,7 +235,7 @@ impl Handler {
 				if self.connection_type.is_some() {
 					let _ = send_message
 						.send(ServerFrames::Error {
-							msg: "Already connected as dispatcher".to_string(),
+							msg: "Already connected as a connection type".to_string(),
 						})
 						.await;
 					return Err("Already connected".into());