Implement MutArray.Generic.unsafeSnoc similar 2 MutArray.unsafeSnoc

Author: adithyaov

core/src/Streamly/Internal/Data/MutArray/Generic.hs

@@ -410,6 +410,13 @@ reallocWith label sizer reqSize arr = do
 -- Snoc
 -------------------------------------------------------------------------------
 
+{-# INLINE snocNewEnd #-}
+snocNewEnd :: MonadIO m => Int -> MutArray a -> a -> m (MutArray a)
+snocNewEnd newEnd arr@MutArray{..} x = liftIO $ do
+    assert (newEnd <= arrBound) (return ())
+    putIndexUnsafeWith arrEnd arrContents# x
+    return $ arr {arrEnd = newEnd}
+
 -- XXX Not sure of the behavior of writeArray# if we specify an index which is
 -- out of bounds. This comment should be rewritten based on that.
 -- | Really really unsafe, appends the element into the first array, may
@@ -418,12 +425,8 @@ reallocWith label sizer reqSize arr = do
 --
 -- /Internal/
 {-# INLINE unsafeSnoc #-}
-unsafeSnoc, snocUnsafe :: MonadIO m => MutArray a -> a -> m (MutArray a)
-unsafeSnoc arr@MutArray {..} a = do
-    assert (arrEnd < arrBound) (return ())
-    let arr1 = arr {arrEnd = arrEnd + 1}
-    unsafePutIndex (length arr) arr1 a
-    return arr1
+snocUnsafe, unsafeSnoc :: MonadIO m => MutArray a -> a -> m (MutArray a)
+unsafeSnoc arr = snocNewEnd (arrEnd arr + 1) arr
 
 -- NOINLINE to move it out of the way and not pollute the instruction cache.
 {-# NOINLINE snocWithRealloc #-}