minor fix Fs2UnaryServerCallHandler state handling

typelevel · Feb 6, 2022 · f3be05f · f3be05f
1 parent 1cc473c
commit f3be05f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/runtime/src/main/scala/fs2/grpc/internal/server/Fs2UnaryServerCallHandler.scala b/runtime/src/main/scala/fs2/grpc/internal/server/Fs2UnaryServerCallHandler.scala
@@ -66,7 +66,7 @@ object Fs2UnaryServerCallHandler {
 
       private def earlyClose(current: State[Request], status: Status): SyncIO[Unit] = {
         if (current.cancel.isEmpty) {
-          state.set(State(Some(SyncIO.unit), None)) >> call.close(status, new Metadata())
+          state.set(current.copy(cancel = Some(SyncIO.unit))) >> call.close(status, new Metadata())
         } else {
           SyncIO.unit
         }