@@ -20,13 +20,14 @@ import scala.util.control.NonFatal
2020import scala .concurrent .ExecutionContext
2121import scala .concurrent .Promise
2222import scala .concurrent .Await
23+ import ch .epfl .scala .debugadapter .Logger
2324
24- class TestingDebugClient (socket : Socket , debug : String => Unit )(implicit
25+ class TestingDebugClient (socket : Socket , logger : Logger )(implicit
2526 ec : ExecutionContext
2627) extends AbstractDebugClient (
2728 socket.getInputStream,
2829 socket.getOutputStream,
29- debug
30+ logger
3031 ) {
3132 override def close (): Unit = {
3233 super .close()
@@ -239,12 +240,12 @@ object TestingDebugClient {
239240 def connect (
240241 uri : URI ,
241242 timeout : Duration = 4 .seconds,
242- debug : String => Unit = _ => ()
243+ logger : Logger = NoopLogger
243244 )(implicit ec : ExecutionContext ): TestingDebugClient = {
244245 val socket = new Socket ()
245246 val address = new InetSocketAddress (uri.getHost, uri.getPort)
246247 socket.connect(address, timeout.toMillis.intValue)
247- val client = new TestingDebugClient (socket, debug )
248+ val client = new TestingDebugClient (socket, logger )
248249 val listening = new java.lang.Thread {
249250 override def run (): Unit = client.run()
250251 }
@@ -256,7 +257,7 @@ object TestingDebugClient {
256257class AbstractDebugClient (
257258 input : InputStream ,
258259 output : OutputStream ,
259- debug : String => Unit
260+ logger : Logger
260261) {
261262 private final val BufferSize = 4096
262263 private final val TwoCRLF = " \r\n\r\n "
@@ -339,7 +340,7 @@ class AbstractDebugClient(
339340 .flatten
340341 rawMessages.foreach { raw =>
341342 try {
342- debug(s " Received $raw" )
343+ logger. debug(s " Received $raw" )
343344 val message = JsonUtils .fromJson(raw, classOf [Messages .ProtocolMessage ])
344345
345346 if (message.`type`.equals(" response" )) {
@@ -387,7 +388,7 @@ class AbstractDebugClient(
387388 try {
388389 this .writer.write(data)
389390 this .writer.flush()
390- debug(s " Sent $jsonMessage" )
391+ logger. debug(s " Sent $jsonMessage" )
391392 } catch {
392393 case NonFatal (e) =>
393394 System .err.println(s " Write data to io exception: ${e.getMessage}" )
0 commit comments