Refactor: Improve error handling for address conversion in state query responses

Author: lowhung

modules/address_state/src/address_state.rs

@@ -210,9 +210,16 @@ impl AddressState {
                     AddressStateQuery::GetAddressUTxOs { address } => {
                         match state.get_address_utxos(address).await {
                             Ok(Some(utxos)) => AddressStateQueryResponse::AddressUTxOs(utxos),
-                            Ok(None) => AddressStateQueryResponse::Error(QueryError::not_found(
-                                "Address not found",
-                            )),
+                            Ok(None) => match address.to_string() {
+                                Ok(addr_str) => AddressStateQueryResponse::Error(
+                                    QueryError::not_found(format!("Address {}", addr_str)),
+                                ),
+                                Err(e) => {
+                                    AddressStateQueryResponse::Error(QueryError::internal_error(
+                                        format!("Could not convert address to string: {}", e),
+                                    ))
+                                }
+                            },
                             Err(e) => AddressStateQueryResponse::Error(QueryError::internal_error(
                                 e.to_string(),
                             )),
@@ -221,9 +228,16 @@ impl AddressState {
                     AddressStateQuery::GetAddressTransactions { address } => {
                         match state.get_address_transactions(address).await {
                             Ok(Some(txs)) => AddressStateQueryResponse::AddressTransactions(txs),
-                            Ok(None) => AddressStateQueryResponse::Error(QueryError::not_found(
-                                "Address not found",
-                            )),
+                            Ok(None) => match address.to_string() {
+                                Ok(addr_str) => AddressStateQueryResponse::Error(
+                                    QueryError::not_found(format!("Address {}", addr_str)),
+                                ),
+                                Err(e) => {
+                                    AddressStateQueryResponse::Error(QueryError::internal_error(
+                                        format!("Could not convert address to string: {}", e),
+                                    ))
+                                }
+                            },
                             Err(e) => AddressStateQueryResponse::Error(QueryError::internal_error(
                                 e.to_string(),
                             )),