feat(examples): Expand rust-outbound-pg example

Uncomment write fn and map it to the `/write` endpoint.
Also, add basic README.

Signed-off-by: Konstantin Shabanov <mail@etehtsea.me>

Author: etehtsea

examples/rust-outbound-pg/README.md

@@ -0,0 +1,49 @@
+# Spin Outbound PostgreSQL example
+
+This example shows how to access a PostgreSQL database from Spin component.
+
+## Spin up
+
+From example root:
+
+```
+createdb spin_dev
+psql -d spin_dev -f db/testdata.sql
+RUST_LOG=spin=trace spin build --up
+```
+
+Curl the read route:
+
+```
+$ curl -i localhost:3000/read
+HTTP/1.1 200 OK
+content-length: 501
+date: Sun, 25 Sep 2022 15:45:02 GMT
+
+Found 2 article(s) as follows:
+article: Article {
+    id: 1,
+    title: "My Life as a Goat",
+    content: "I went to Nepal to live as a goat, and it was much better than being a butler.",
+    authorname: "E. Blackadder",
+}
+article: Article {
+    id: 2,
+    title: "Magnificent Octopus",
+    content: "Once upon a time there was a lovely little sausage.",
+    authorname: "S. Baldrick",
+}
+
+(Column info: id:DbDataType::Int32, title:DbDataType::Str, content:DbDataType::Str, authorname:DbDataType::Str)
+```
+
+Curl the write route:
+
+```
+$ curl -i localhost:3000/write
+HTTP/1.1 200 OK
+content-length: 9
+date: Sun, 25 Sep 2022 15:46:22 GMT
+
+Count: 3
+```

examples/rust-outbound-pg/db/testdata.sql

@@ -1,9 +1,18 @@
 CREATE TABLE articletest (
-   id integer not null,
-   title varchar(40) not null,
-   content varchar(8000) not null,
-   authorname varchar(40) not null
+   id integer GENERATED BY DEFAULT AS IDENTITY PRIMARY KEY,
+   title varchar(40) NOT NULL,
+   content text NOT NULL,
+   authorname varchar(40) NOT NULL
 );
 
-INSERT INTO articletest VALUES (1, 'My Life as a Goat', 'I went to Nepal to live as a goat, and it was much better than being a butler.', 'E. Blackadder');
-INSERT INTO articletest VALUES (2, 'Magnificent Octopus', 'Once upon a time there was a lovely little sausage.', 'S. Baldrick');
+INSERT INTO articletest (title, content, authorname) VALUES
+(
+   'My Life as a Goat',
+   'I went to Nepal to live as a goat, and it was much better than being a butler.',
+   'E. Blackadder'
+),
+(
+   'Magnificent Octopus',
+   'Once upon a time there was a lovely little sausage.',
+   'S. Baldrick'
+);

examples/rust-outbound-pg/spin.toml

@@ -5,12 +5,10 @@ trigger = { type = "http", base = "/" }
 version = "0.1.0"
 
 [[component]]
-environment = { DB_URL = "host=localhost user=postgres password=123 dbname=postgres" }
+environment = { DB_URL = "host=localhost user=postgres dbname=spin_dev" }
 id = "outbound-pg"
 source = "target/wasm32-wasi/release/rust_outbound_pg.wasm"
 [component.trigger]
-#route = "/write"
-route = "/read"
+route = "/..."
 [component.build]
 command = "cargo build --target wasm32-wasi --release"
-

examples/rust-outbound-pg/src/lib.rs

@@ -18,10 +18,20 @@ struct Article {
 }
 
 #[http_component]
+fn process(req: Request) -> Result<Response> {
+    match req.uri().path() {
+        "/read" => read(req),
+        "/write" => write(req),
+        _ => Ok(http::Response::builder()
+            .status(404)
+            .body(Some("Not found".into()))?),
+    }
+}
+
 fn read(_req: Request) -> Result<Response> {
     let address = std::env::var(DB_URL_ENV)?;
 
-    let sql = "select id, title, content, authorname from articletest";
+    let sql = "SELECT id, title, content, authorname FROM articletest";
     let rowset = pg::query(&address, sql, &[])
         .map_err(|e| anyhow!("Error executing Postgres query: {:?}", e))?;
 
@@ -64,18 +74,27 @@ fn read(_req: Request) -> Result<Response> {
         .status(200)
         .body(Some(response.into()))?)
 }
-/*
+
 fn write(_req: Request) -> Result<Response> {
     let address = std::env::var(DB_URL_ENV)?;
 
-    let sql = "insert into articletest values ('aaa', 'bbb', 'ccc')";
-    let nrow_executed = pg::execute(&address, sql, &vec![]).map_err(|_| anyhow!("Error execute pg command"))?;
+    let sql = "INSERT INTO articletest (title, content, authorname) VALUES ('aaa', 'bbb', 'ccc')";
+    let nrow_executed =
+        pg::execute(&address, sql, &[]).map_err(|_| anyhow!("Error execute pg command"))?;
 
     println!("nrow_executed: {}", nrow_executed);
 
-    Ok(http::Response::builder().status(200).body(None)?)
+    let sql = "SELECT COUNT(id) FROM articletest";
+    let rowset = pg::query(&address, sql, &[])
+        .map_err(|e| anyhow!("Error executing Postgres query: {:?}", e))?;
+    let row = &rowset.rows[0];
+    let count = as_bigint(&row[0])?;
+    let response = format!("Count: {}\n", count);
+
+    Ok(http::Response::builder()
+        .status(200)
+        .body(Some(response.into()))?)
 }
-*/
 
 fn as_owned_string(value: &pg::DbValue) -> anyhow::Result<String> {
     match value {
@@ -94,6 +113,16 @@ fn as_int(value: &pg::DbValue) -> anyhow::Result<i32> {
     }
 }
 
+fn as_bigint(value: &pg::DbValue) -> anyhow::Result<i64> {
+    match value {
+        pg::DbValue::Int64(n) => Ok(*n),
+        _ => Err(anyhow!(
+            "Expected integer from database but got {:?}",
+            value
+        )),
+    }
+}
+
 fn format_col(column: &pg::Column) -> String {
     format!("{}:{:?}", column.name, column.data_type)
 }