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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
/*
*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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 WARRANTIE4bS OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*
*/
#include "SessionState.h"
#include "qpid/framing/reply_exceptions.h"
#include "qpid/framing/constants.h"
#include "qpid/framing/AMQMethodBody.h"
#include "qpid/log/Statement.h"
#include <algorithm>
#include <boost/bind.hpp>
#include <boost/none.hpp>
namespace qpid {
namespace framing {
SessionState::SessionState(uint32_t ack, const Uuid& uuid) :
state(ATTACHED),
id(uuid),
lastReceived(-1),
lastSent(-1),
ackInterval(ack),
sendAckAt(lastReceived+ackInterval),
solicitAckAt(lastSent+ackInterval),
ackSolicited(false)
{
assert(ackInterval > 0);
}
namespace {
bool isSessionCommand(const AMQFrame& f) {
return f.getMethod() && f.getMethod()->amqpClassId() == SESSION_CLASS_ID;
}
}
boost::optional<SequenceNumber> SessionState::received(const AMQFrame& f) {
if (isSessionCommand(f))
return boost::none;
if (state==RESUMING)
throw CommandInvalidException(
QPID_MSG("Invalid frame: Resuming session, expected session-ack"));
assert(state = ATTACHED);
assert(lastReceived<sendAckAt);
++lastReceived;
QPID_LOG(trace, "Recv # "<< lastReceived << " " << id);
if (lastReceived == sendAckAt)
return sendingAck();
else
return boost::none;
}
bool SessionState::sent(const AMQFrame& f) {
if (isSessionCommand(f))
return false;
unackedOut.push_back(f);
++lastSent;
QPID_LOG(trace, "Sent # "<< lastSent << " " << id);
return (state!=RESUMING) &&
(lastSent == solicitAckAt) &&
sendingSolicit();
}
SessionState::Replay SessionState::replay() {
Replay r(unackedOut.size());
std::copy(unackedOut.begin(), unackedOut.end(), r.begin());
return r;
}
void SessionState::receivedAck(SequenceNumber acked) {
if (state==RESUMING) state=ATTACHED;
assert(state==ATTACHED);
if (lastSent < acked)
throw InvalidArgumentException("Invalid sequence number in ack");
size_t keep = lastSent - acked;
if (keep < unackedOut.size())
unackedOut.erase(unackedOut.begin(), unackedOut.end()-keep);
solicitAckAt = std::max(solicitAckAt, SequenceNumber(acked+ackInterval));
}
SequenceNumber SessionState::sendingAck() {
sendAckAt = lastReceived+ackInterval;
return lastReceived;
}
bool SessionState::sendingSolicit() {
assert(state == ATTACHED);
if (ackSolicited)
return false;
solicitAckAt = lastSent + ackInterval;
return true;
}
SequenceNumber SessionState::resuming() {
state = RESUMING;
return sendingAck();
}
void SessionState::suspend() {
state = SUSPENDED;
}
}} // namespace qpid::framing
|
