Esempio n. 1
0
func (mc *Client) sendAck(pkt *transport.MCRequest) (int, error) {
	res := transport.MCResponse{
		Opcode: pkt.Opcode,
		Opaque: pkt.Opaque,
		Status: transport.SUCCESS,
	}
	return res.Transmit(mc.conn)
}
Esempio n. 2
0
func BenchmarkTransmitResNull(b *testing.B) {
	res := transport.MCResponse{
		Opcode: transport.SET,
		Cas:    938424885,
		Opaque: 7242,
		Status: 824,
		Key:    []byte("somekey"),
		Body:   []byte("somevalue"),
	}

	b.SetBytes(int64(res.Size()))

	for i := 0; i < b.N; i++ {
		_, err := transmitResponse(ioutil.Discard, &res)
		if err != nil {
			b.Fatalf("Error transmitting request: %v", err)
		}
	}
}
Esempio n. 3
0
func TestTransmitRes(t *testing.T) {
	b := &bytes.Buffer{}
	buf := bufio.NewWriter(b)

	res := transport.MCResponse{
		Opcode: transport.SET,
		Cas:    938424885,
		Opaque: 7242,
		Status: 0x338,
		Key:    []byte("somekey"),
		Body:   []byte("somevalue"),
	}

	_, err := transmitResponse(buf, &res)
	if err != nil {
		t.Fatalf("Error transmitting request: %v", err)
	}

	buf.Flush()

	expected := []byte{
		transport.RES_MAGIC, byte(transport.SET),
		0x0, 0x7, // length of key
		0x0,       // extra length
		0x0,       // reserved
		0x3, 0x38, // Status
		0x0, 0x0, 0x0, 0x10, // Length of value
		0x0, 0x0, 0x1c, 0x4a, // opaque
		0x0, 0x0, 0x0, 0x0, 0x37, 0xef, 0x3a, 0x35, // CAS
		's', 'o', 'm', 'e', 'k', 'e', 'y',
		's', 'o', 'm', 'e', 'v', 'a', 'l', 'u', 'e'}

	if len(b.Bytes()) != res.Size() {
		t.Fatalf("Expected %v bytes, got %v", res.Size(),
			len(b.Bytes()))
	}

	if !reflect.DeepEqual(b.Bytes(), expected) {
		t.Fatalf("Expected:\n%#v\n  -- got -- \n%#v",
			expected, b.Bytes())
	}
}
Esempio n. 4
0
func BenchmarkReceive(b *testing.B) {
	res := transport.MCResponse{
		Opcode: transport.SET,
		Cas:    938424885,
		Opaque: 7242,
		Status: 824,
		Key:    []byte("somekey"),
		Body:   []byte("somevalue"),
	}

	datum := res.Bytes()
	datum[0] = transport.REQ_MAGIC
	b.SetBytes(int64(len(datum)))

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		_, err := ReadPacket(bytes.NewReader(datum))
		if err != nil {
			b.Fatalf("Failed to read: %v", err)
		}
	}
}
Esempio n. 5
0
func BenchmarkTransmitRes(b *testing.B) {
	bout := &bytes.Buffer{}

	res := transport.MCResponse{
		Opcode: transport.SET,
		Cas:    938424885,
		Opaque: 7242,
		Status: 824,
		Key:    []byte("somekey"),
		Body:   []byte("somevalue"),
	}

	b.SetBytes(int64(res.Size()))

	for i := 0; i < b.N; i++ {
		bout.Reset()
		buf := bufio.NewWriterSize(bout, res.Size()*2)
		_, err := transmitResponse(buf, &res)
		if err != nil {
			b.Fatalf("Error transmitting request: %v", err)
		}
	}
}
Esempio n. 6
0
func transmitResponse(o io.Writer, res *transport.MCResponse) (int, error) {
	return res.Transmit(o)
}
Esempio n. 7
0
func notFound(req *transport.MCRequest, s *storage) *transport.MCResponse {
	var response transport.MCResponse
	response.Status = transport.UNKNOWN_COMMAND
	return &response
}