1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

// Copyright (C) 2016 Benjamin Fry <benjaminfry@me.com>
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

//! Trait for client connections

use std::fmt::Debug;

use ::error::*;

/// Trait for client connections
pub trait ClientConnection: Sized+Debug {
  /// Sends a serialized message to via this connection, returning the serialized response.
  ///
  /// # Arguments
  ///
  /// * `bytes` - the serialized Message
  fn send(&mut self, bytes: Vec<u8>) -> ClientResult<Vec<u8>>;
  // TODO: split connect, send and read...
}

#[cfg(test)]
pub mod test {
  use std::fmt;
  use super::*;
  use ::op::Message;
  use ::authority::Catalog;
  use ::serialize::binary::{BinDecoder, BinEncoder, BinSerializable};
  use ::error::*;

  pub struct TestClientConnection<'a> {
    catalog: &'a Catalog
  }

  impl<'a> TestClientConnection<'a> {
    pub fn new(catalog: &'a Catalog) -> TestClientConnection<'a> {
      TestClientConnection { catalog: catalog }
    }
  }

  impl<'a> ClientConnection for TestClientConnection<'a> {
    fn send(&mut self, bytes: Vec<u8>) -> ClientResult<Vec<u8>> {
      let mut decoder = BinDecoder::new(&bytes);

      let message = try_rethrow!(ClientError::DecodeError, Message::read(&mut decoder));
      let response = self.catalog.handle_request(&message);

      let mut buf = Vec::with_capacity(512);
      {
        let mut encoder = BinEncoder::new(&mut buf);
        try_rethrow!(ClientError::EncodeError, response.emit(&mut encoder));
      }

      Ok(buf)
    }
  }

  impl<'a> fmt::Debug for TestClientConnection<'a> {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
      write!(f, "TestClientConnection catalog")
    }
  }
}