Fix race condition

zqz · May 29, 2016 · b966c04 · b966c04
1 parent aaae197
commit b966c04
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,5 +1,6 @@
 files/
 assets/
+assets
 zqz
 *.coverprofile
 /vendor/*
diff --git a/models/connection.go b/models/connection.go
@@ -4,6 +4,7 @@ import (
 	"database/sql"
 	"fmt"
 	"os"
+	"sync"
 
 	"github.com/lib/pq"
 )
@@ -15,8 +16,12 @@ type Executor interface {
 }
 
 var database *sql.DB
+var mu sync.Mutex
 
 func GetDB() (*sql.DB, error) {
+	mu.Lock()
+	defer mu.Unlock()
+
 	var err error
 	if database == nil {
 		database, err = Connection()