From 9cc646db7abeb49bd4b8f20ad42de79f64572317 Mon Sep 17 00:00:00 2001
From: Eliot Whalan <ewhal@pantsu.cat>
Date: Mon, 27 Jun 2016 07:46:58 +1000
Subject: [PATCH] Only open db once

---
 main.go | 23 ++++++++++++++---------
 1 file changed, 14 insertions(+), 9 deletions(-)

diff --git a/main.go b/main.go
index e56ba96..6de52b8 100644
--- a/main.go
+++ b/main.go
@@ -31,6 +31,10 @@ const (
 	DATABASE = USERNAME + ":" + PASS + "@/" + NAME + "?charset=utf8"
 )
 
+var (
+	db *db.SQL
+)
+
 type Response struct {
 	ID     string `json:"id"`
 	TITLE  string `json:"title"`
@@ -49,6 +53,15 @@ type Page struct {
 	Clone    string
 }
 
+func setup() {
+	var err error
+	db, err = sql.Open("mysql", DATABASE)
+	if err != nil {
+		log.Fatal(err.Error())
+	}
+
+}
+
 func check(err error) {
 	if err != nil {
 		fmt.Println(err)
@@ -57,8 +70,6 @@ func check(err error) {
 
 func generateName() string {
 	id := uniuri.NewLen(LENGTH)
-	db, err := sql.Open("mysql", DATABASE)
-	check(err)
 
 	query, err := db.Query("select id from pastebin where id=?", id)
 	if err != sql.ErrNoRows {
@@ -80,8 +91,6 @@ func hash(paste string) string {
 }
 
 func save(raw string, lang string, title string, expiry string) []string {
-	db, err := sql.Open("mysql", DATABASE)
-	check(err)
 
 	sha := hash(raw)
 	query, err := db.Query("select id, title, hash, data, delkey from pastebin where hash=?", sha)
@@ -159,9 +168,6 @@ func delHandler(w http.ResponseWriter, r *http.Request) {
 	paste := vars["pasteId"]
 	delkey := vars["delKey"]
 
-	db, err := sql.Open("mysql", DATABASE)
-	check(err)
-
 	stmt, err := db.Prepare("delete from pastebin where delkey=?")
 	check(err)
 
@@ -249,10 +255,9 @@ func highlight(s string, lang string) (string, error) {
 
 func getPaste(paste string, lang string) (string, string) {
 	param1 := html.EscapeString(paste)
-	db, err := sql.Open("mysql", DATABASE)
 	var title, s string
 	var expiry string
-	err = db.QueryRow("select title, data, expiry from pastebin where id=?", param1).Scan(&title, &s, &expiry)
+	err := db.QueryRow("select title, data, expiry from pastebin where id=?", param1).Scan(&title, &s, &expiry)
 	check(err)
 	if time.Now().Format("2006-01-02 15:04:05") > expiry {
 		stmt, err := db.Prepare("delete from pastebin where id=?")