Move bookstore to root dir

Author: leftrightleft

main.go

@@ -0,0 +1,67 @@
+package main
+
+import (
+	"database/sql"
+	"fmt"
+	"log"
+	"net/http"
+	"os"
+
+	"github.com/octodemo/advanced-security-go/models"
+
+	_ "github.com/mattn/go-sqlite3"
+)
+
+func main() {
+	var err error
+
+	os.Remove("./bookstore.db")
+
+	models.DB, err = sql.Open("sqlite3", "./bookstore.db")
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer models.DB.Close()
+
+	sqlStmt := `
+	CREATE TABLE books (
+		name varchar(255) NOT NULL,
+		author varchar(255) NOT NULL,
+		read varchar(255) NOT NULL
+	);
+
+	INSERT INTO books (name, author, read) VALUES
+	("The Hobbit", "JRR Tolkien", "True"),
+	("The Fellowship of the Ring", "JRR Tolkien", "True"),
+	("The Eye of the World", "Robert Jordan", "False"),
+	("A Game of Thrones", "George R. R. Martin", "True"),
+	("The Way of Kings", "Brandon Sanderson", "False");
+	`
+	_, err = models.DB.Exec(sqlStmt)
+	if err != nil {
+		log.Printf("%q: %s\n", err, sqlStmt)
+		return
+	}
+
+	_, err = models.DB.Begin()
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	http.HandleFunc("/books", booksIndex)
+	http.ListenAndServe(":3000", nil)
+}
+
+// booksIndex sends a HTTP response listing all books.
+func booksIndex(w http.ResponseWriter, r *http.Request) {
+	bks, err := models.AllBooks()
+	if err != nil {
+		log.Println(err)
+		http.Error(w, http.StatusText(500), 500)
+		return
+	}
+
+	for _, bk := range bks {
+		fmt.Fprintf(w, "%s, %s, %s\n", bk.Title, bk.Author, bk.Read)
+	}
+}

models/models.go

@@ -0,0 +1,42 @@
+package models
+
+import (
+	"database/sql"
+)
+
+// Create an exported global variable to hold the database connection pool.
+var DB *sql.DB
+
+type Book struct {
+	Title  string
+	Author string
+	Read   string
+}
+
+// AllBooks returns a slice of all books in the books table.
+func AllBooks() ([]Book, error) {
+	// Note that we are calling Query() on the global variable.
+	rows, err := DB.Query("SELECT * FROM books")
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	var bks []Book
+
+	for rows.Next() {
+		var bk Book
+
+		err := rows.Scan(&bk.Title, &bk.Author, &bk.Read)
+		if err != nil {
+			return nil, err
+		}
+
+		bks = append(bks, bk)
+	}
+	if err = rows.Err(); err != nil {
+		return nil, err
+	}
+
+	return bks, nil
+}