Add instance for primitive's ByteArray

Author: phadej

CHANGELOG.md

@@ -1,3 +1,7 @@
+## 1.0.3
+
+- Add instance for `primitive`s `ByteArray`.
+
 ## 1.0.2
 
 - Support `aeson-2.0.0.0`

binary-instances.cabal

@@ -43,6 +43,7 @@ library
  , binary-orphans >=1.0.3 && <1.1
  , case-insensitive >=1.2.0.4 && <1.2.2
  , hashable >=1.2.3.3 && <1.5
+ , primitive >=0.7.2.0 && <0.8
  , scientific >=0.3.3.8 && <0.4
  , tagged >=0.7.3 && <0.8.7
  , text >=1.2.0.6 && <1.3 || >=2.0 && <2.1
@@ -57,6 +58,7 @@ library
  Data.Binary.Instances.Aeson
  Data.Binary.Instances.CaseInsensitive
  Data.Binary.Instances.Hashable
+ Data.Binary.Instances.Primitive
  Data.Binary.Instances.Scientific
  Data.Binary.Instances.Tagged
  Data.Binary.Instances.Text
@@ -78,8 +80,9 @@ test-suite binary-instances-test
  , bytestring
  , case-insensitive
  , hashable
+ , primitive
  , QuickCheck >=2.13.1 && <2.15
- , quickcheck-instances >=0.3.25 && <0.4
+ , quickcheck-instances >=0.3.28 && <0.4
  , scientific
  , tagged
  , tasty >=0.10.1.2 && <1.5

src/Data/Binary/Instances.hs

@@ -28,6 +28,7 @@ import Data.Binary.Orphans ()
 import Data.Binary.Instances.Aeson ()
 import Data.Binary.Instances.CaseInsensitive ()
 import Data.Binary.Instances.Hashable ()
+import Data.Binary.Instances.Primitive ()
 import Data.Binary.Instances.Scientific ()
 import Data.Binary.Instances.Tagged ()
 import Data.Binary.Instances.Text ()

src/Data/Binary/Instances/Primitive.hs

@@ -0,0 +1,27 @@
+{-# LANGUAGE CPP #-}
+{-# OPTIONS_GHC -fno-warn-orphans #-}
+module Data.Binary.Instances.Primitive where
+
+import Data.Binary.Orphans ()
+
+import Control.Monad (replicateM)
+import Data.Binary (Binary, Get, Put, get, put)
+import Data.Word (Word8)
+
+import qualified Data.Primitive as Prim
+
+-- | @since 1.0.3
+instance Binary Prim.ByteArray where
+ put ba = put maxI >> go 0
+ where
+ maxI :: Int
+ maxI = Prim.sizeofByteArray ba
+
+ go :: Int -> Put
+ go i | i < maxI = put (Prim.indexByteArray ba i :: Word8) >> go (i + 1)
+ | otherwise = return ()
+
+ get = do
+ len <- get
+ xs <- replicateM len get
+ return (Prim.byteArrayFromListN len (xs :: [Word8]))

test/Tests.hs

@@ -23,6 +23,7 @@ import Data.Vector (Vector)
 
 import qualified Data.Vector.Unboxed as VU
 
+import qualified Data.Primitive as Prim
 import qualified Data.Time.Calendar.Compat as Time
 import qualified Data.Time.Calendar.Month.Compat as Time
 import qualified Data.Time.Calendar.Quarter.Compat as Time
@@ -70,6 +71,7 @@ tests = testGroup "Roundtrip"
  -- vector
  , roundtripProperty (undefined :: Vector Char)
  , roundtripProperty (undefined :: VU.Vector Char)
+ , roundtripProperty (undefined :: Prim.ByteArray)
  ]
 
 roundtrip :: (Eq a, Show a, Binary a) => a -> a -> Property