diff --git a/src/main/scala/com/twitter/finagle/postgres/connection/ConnectionStateMachine.scala b/src/main/scala/com/twitter/finagle/postgres/connection/ConnectionStateMachine.scala index b5e4253f..3911712e 100644 --- a/src/main/scala/com/twitter/finagle/postgres/connection/ConnectionStateMachine.scala +++ b/src/main/scala/com/twitter/finagle/postgres/connection/ConnectionStateMachine.scala @@ -95,10 +95,8 @@ class ConnectionStateMachine(state: State = AuthenticationRequired, val id: Int) case (CommandComplete(Update(count)), SimpleQuery) => (None, EmitOnReadyForQuery(CommandCompleteResponse(count))) case (CommandComplete(Delete(count)), SimpleQuery) => (None, EmitOnReadyForQuery(CommandCompleteResponse(count))) case (CommandComplete(DiscardAll), SimpleQuery) => (None, EmitOnReadyForQuery(CommandCompleteResponse(1))) - case (CommandComplete(Begin), SimpleQuery) => (None, EmitOnReadyForQuery(CommandCompleteResponse(1))) - case (CommandComplete(Savepoint), SimpleQuery) => (None, EmitOnReadyForQuery(CommandCompleteResponse(1))) - case (CommandComplete(RollBack), SimpleQuery) => (None, EmitOnReadyForQuery(CommandCompleteResponse(1))) - case (CommandComplete(Commit), SimpleQuery) => (None, EmitOnReadyForQuery(CommandCompleteResponse(1))) + case (CommandComplete(Begin | Savepoint | Release | RollBack | Commit), SimpleQuery) => + (None, EmitOnReadyForQuery(CommandCompleteResponse(1))) case (CommandComplete(Do), SimpleQuery) => (None, EmitOnReadyForQuery(CommandCompleteResponse(1))) case (RowDescription(fields), SimpleQuery) => diff --git a/src/main/scala/com/twitter/finagle/postgres/messages/BackendMessageParser.scala b/src/main/scala/com/twitter/finagle/postgres/messages/BackendMessageParser.scala index ae14ad5d..61d89d5f 100644 --- a/src/main/scala/com/twitter/finagle/postgres/messages/BackendMessageParser.scala +++ b/src/main/scala/com/twitter/finagle/postgres/messages/BackendMessageParser.scala @@ -239,6 +239,7 @@ class BackendMessageParser { case "UPDATE" => Update(parts(1).toInt) case "BEGIN" => Begin case "SAVEPOINT" => Savepoint + case "RELEASE" => Release case "ROLLBACK" => RollBack case "COMMIT" => Commit case _ => throw new IllegalStateException("Unknown command complete response tag " + tag) diff --git a/src/main/scala/com/twitter/finagle/postgres/messages/BackendMessages.scala b/src/main/scala/com/twitter/finagle/postgres/messages/BackendMessages.scala index d14d3086..9a5ee2d1 100644 --- a/src/main/scala/com/twitter/finagle/postgres/messages/BackendMessages.scala +++ b/src/main/scala/com/twitter/finagle/postgres/messages/BackendMessages.scala @@ -77,6 +77,8 @@ case object Begin extends CommandCompleteStatus case object Savepoint extends CommandCompleteStatus +case object Release extends CommandCompleteStatus + case object RollBack extends CommandCompleteStatus case object Commit extends CommandCompleteStatus