use ZPure and not State

Author: keisar

kvstore/src/main/scala/zio/kvstore/App.scala

@@ -8,7 +8,6 @@ import zio.raft.{Command, Index, MemberId, Raft, SnapshotStore, StateMachine}
 import zio.stream.{Stream, ZStream}
 import zio.zmq.ZContext
 import zio.{Chunk, UIO, ZIO, ZIOAppArgs, ZLayer}
-import zio.prelude.State
 
 import scodec.Codec
 import scodec.bits.BitVector
@@ -56,8 +55,8 @@ class KVStateMachine extends StateMachine[Any, Map[String, String], KVCommand]:
   override def apply(command: KVCommand)
     : ZPure[Any, Map[String, String], Map[String, String], Any, Nothing, command.Response] =
     (command match
-      case Set(k, v) => State.update((map: Map[String, String]) => map.updated(k, v))
-      case Get(k)    => State.get.map((map: Map[String, String]) => map.get(k).getOrElse(""))
+      case Set(k, v) => ZPure.update((map: Map[String, String]) => map.updated(k, v))
+      case Get(k)    => ZPure.get.map((map: Map[String, String]) => map.get(k).getOrElse(""))
     ).map(_.asInstanceOf[command.Response])
 
 class HttpServer(raft: Raft[Any, Map[String, String], KVCommand]):

raft/src/test/scala/zio/raft/TestStateMachine.scala

@@ -2,7 +2,6 @@ package zio.raft
 
 import zio.UIO
 import zio.stream.{Stream, ZStream}
-import zio.prelude.State
 import zio.prelude.fx.ZPure
 
 sealed trait TestCommands extends Command:
@@ -15,8 +14,8 @@ case class TestStateMachine(enableSnapshot: Boolean) extends StateMachine[Any, I
 
   def apply(command: TestCommands): ZPure[Any, Int, Int, Any, Nothing, command.Response] =
     command match
-      case Increase => State.modify(s => (s + 1, s + 1))
-      case Get      => State.get
+      case Increase => ZPure.modify(s => (s + 1, s + 1))
+      case Get      => ZPure.get
 
   override def restoreFromSnapshot(stream: Stream[Nothing, Byte]): UIO[Int] =
     stream.runCollect.map(b => new String(b.toArray).toInt)