allow using hstore and hstore[] in macros

sqlx-postgres/src/type_checking.rs

@@ -209,6 +209,9 @@ impl_type_checking!(
         #[cfg(feature = "time")]
         Vec<sqlx::postgres::types::PgRange<sqlx::types::time::OffsetDateTime>> |
             &[sqlx::postgres::types::PgRange<sqlx::types::time::OffsetDateTime>],
+
+        sqlx::postgres::types::PgHstore,
+        Vec<sqlx::postgres::types::PgHstore> | &[sqlx::postgres::types::PgHstore],
     },
     ParamChecking::Strong,
     feature-types: info => info.__type_feature_gate(),

sqlx-postgres/src/types/hstore.rs

@@ -15,6 +15,8 @@ use crate::{
 use serde::{Deserialize, Serialize};
 use sqlx_core::bytes::Buf;
 
+use super::PgHasArrayType;
+
 /// Key-value support (`hstore`) for Postgres.
 ///
 /// SQLx currently maps `hstore` to a `BTreeMap<String, Option<String>>` but this may be expanded in
@@ -138,6 +140,12 @@ impl Type<Postgres> for PgHstore {
     }
 }
 
+impl PgHasArrayType for PgHstore {
+    fn array_type_info() -> PgTypeInfo {
+        PgTypeInfo::array_of("hstore")
+    }
+}
+
 impl<'r> Decode<'r, Postgres> for PgHstore {
     fn decode(value: PgValueRef<'r>) -> Result<Self, BoxDynError> {
         let mut buf = <&[u8] as Decode<Postgres>>::decode(value)?;

tests/postgres/macros.rs

@@ -20,6 +20,49 @@ async fn test_query() -> anyhow::Result<()> {
     Ok(())
 }
 
+#[sqlx_macros::test]
+async fn test_hstores() -> anyhow::Result<()> {
+    let mut conn = new::<Postgres>().await?;
+
+    let _ = sqlx::query!(
+        "INSERT INTO test_hstores(stores) VALUES ($1);",
+        &[
+            sqlx::postgres::types::PgHstore::from_iter([("key1".to_string(), "value".to_string())]),
+            sqlx::postgres::types::PgHstore::from_iter([("key2".to_string(), "value".to_string())]),
+        ]
+    )
+    .execute(&mut conn)
+    .await?;
+
+    let record = sqlx::query!("SELECT stores FROM test_hstores LIMIT 1;")
+        .fetch_one(&mut conn)
+        .await?;
+
+    assert_eq!(record.stores.unwrap()[1]["key2"], Some("value".to_string()));
+
+    Ok(())
+}
+
+#[sqlx_macros::test]
+async fn test_hstore() -> anyhow::Result<()> {
+    let mut conn = new::<Postgres>().await?;
+
+    let _ = sqlx::query!(
+        "INSERT INTO test_hstore(store) VALUES ($1);",
+        sqlx::postgres::types::PgHstore::from_iter([("key".to_string(), "value".to_string())])
+    )
+    .execute(&mut conn)
+    .await?;
+
+    let record = sqlx::query!("SELECT store FROM test_hstore LIMIT 1;")
+        .fetch_one(&mut conn)
+        .await?;
+
+    assert_eq!(record.store.unwrap()["key"], Some("value".to_string()));
+
+    Ok(())
+}
+
 #[sqlx_macros::test]
 async fn test_non_null() -> anyhow::Result<()> {
     let mut conn = new::<Postgres>().await?;

tests/postgres/setup.sql

@@ -7,6 +7,9 @@ CREATE EXTENSION IF NOT EXISTS cube;
 -- https://www.postgresql.org/docs/current/citext.html
 CREATE EXTENSION IF NOT EXISTS citext;
 
+-- https://www.postgresql.org/docs/current/hstore.html
+CREATE EXTENSION IF NOT EXISTS hstore;
+
 -- https://www.postgresql.org/docs/current/sql-createtype.html
 CREATE TYPE status AS ENUM ('new', 'open', 'closed');
 
@@ -58,3 +61,6 @@ CREATE TABLE test_citext (
 CREATE SCHEMA IF NOT EXISTS foo;
 
 CREATE TYPE foo."Foo" as ENUM ('Bar', 'Baz');
+
+CREATE TABLE test_hstore(id_test_hstore SERIAL, store HSTORE);
+CREATE TABLE test_hstores(id_test_hstores SERIAL, stores HSTORE[]);