Only open db once

This commit is contained in:
Eliot Whalan 2016-06-27 07:46:58 +10:00
parent 219a870162
commit 9cc646db7a
1 changed files with 14 additions and 9 deletions

23
main.go
View File

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