From f87db1e084435dd7fd0cfeffdd11635f51ed76d0 Mon Sep 17 00:00:00 2001 From: Juan Vanecek Date: Fri, 30 Aug 2024 02:56:14 -0300 Subject: [PATCH] Rename ensure to assert in messages that only checks something and raise error --- .../AmqpChannelHandler.class.st | 16 ++++---- .../AmqpConnection.class.st | 40 +++++++++---------- 2 files changed, 28 insertions(+), 28 deletions(-) diff --git a/source/Ansible-Protocol-Core/AmqpChannelHandler.class.st b/source/Ansible-Protocol-Core/AmqpChannelHandler.class.st index 8e895066..3682c310 100644 --- a/source/Ansible-Protocol-Core/AmqpChannelHandler.class.st +++ b/source/Ansible-Protocol-Core/AmqpChannelHandler.class.st @@ -16,6 +16,13 @@ Class { #package : 'Ansible-Protocol-Core' } +{ #category : 'handling' } +AmqpChannelHandler >> assertChannelOpen [ + + closeReason notNil + ifTrue: [ AmqpDisconnectedError signal: 'Channel closed' ] +] + { #category : 'accessing' } AmqpChannelHandler >> channelNumber [ @@ -67,13 +74,6 @@ AmqpChannelHandler >> dispatchCommand [ self resetState ] -{ #category : 'handling' } -AmqpChannelHandler >> ensureOpen [ - - closeReason notNil - ifTrue: [ AmqpDisconnectedError signal: 'Channel closed' ] -] - { #category : 'handling' } AmqpChannelHandler >> handleBodyFrame: frame [ @@ -167,7 +167,7 @@ AmqpChannelHandler >> waitForReply: acceptableReplies [ [ i > inbound size ] whileTrue: [ | currentConnection | - self ensureOpen. + self assertChannelOpen. currentConnection := connection mainLoopCycle. currentConnection whenConnected: [ ] diff --git a/source/Ansible-Protocol-Core/AmqpConnection.class.st b/source/Ansible-Protocol-Core/AmqpConnection.class.st index b08b36b0..2b04b3c3 100644 --- a/source/Ansible-Protocol-Core/AmqpConnection.class.st +++ b/source/Ansible-Protocol-Core/AmqpConnection.class.st @@ -38,6 +38,22 @@ AmqpConnection class >> to: aHostname extraProperties: aClientPropertyCollection ] +{ #category : 'connection-handling' } +AmqpConnection >> assertChannelOpen: channelNumber [ + + self assertSocketConnected. + ^ ( channels at: channelNumber ) assertChannelOpen +] + +{ #category : 'connection-handling' } +AmqpConnection >> assertSocketConnected [ + + ( socket isConnected and: [ socket isOtherEndClosed not ] ) ifFalse: [ + AmqpDisconnectedError signal: 'Connection closed' ]. + + ^ true +] + { #category : 'connection-handling' } AmqpConnection >> becomeCloseAfter: aBlock [ @@ -106,7 +122,7 @@ AmqpConnection >> createChannel [ | handler | - self ensureOpen. + self assertSocketConnected. handler := AmqpChannelHandler new connection: self. ^ AmqpChannel using: handler ] @@ -117,22 +133,6 @@ AmqpConnection >> credentials [ ^ credentials ] -{ #category : 'connection-handling' } -AmqpConnection >> ensureChannel: channelNumber [ - - self ensureOpen. - ^ ( channels at: channelNumber ) ensureOpen -] - -{ #category : 'connection-handling' } -AmqpConnection >> ensureOpen [ - - ( socket isConnected and: [ socket isOtherEndClosed not ] ) ifFalse: [ - AmqpDisconnectedError signal: 'Connection closed' ]. - - ^ true -] - { #category : 'connection-handling' } AmqpConnection >> handleConnectionClose: cmd [ @@ -347,7 +347,7 @@ AmqpConnection >> rpc: aRequestMethod onChannel: aChannelNumber ifConnectionClos ^ [ | ch | - ch := self ensureChannel: aChannelNumber. + ch := self assertChannelOpen: aChannelNumber. self sendMethod: aRequestMethod onChannel: aChannelNumber. ch waitForReply: aRequestMethod acceptableResponseClasses ] @@ -395,7 +395,7 @@ AmqpConnection >> sendMethod: aMethod properties: aProperties body: aByteArray o | pos | - self ensureChannel: channelNumber. + self assertChannelOpen: channelNumber. self sendFrame: ( AmqpMethodFrame new @@ -483,7 +483,7 @@ AmqpConnection >> virtualHost [ { #category : 'connection-handling' } AmqpConnection >> waitForEvent [ - [ self ensureOpen ] whileTrue: [ self mainLoopCycle ] + [ self assertSocketConnected ] whileTrue: [ self mainLoopCycle ] ] { #category : 'connection-handling' }