From 5b1d414ba23937e7fe1eba74547a7acf33e9c5a6 Mon Sep 17 00:00:00 2001
From: Joe Robinson <joe@mumsnet.com>
Date: Mon, 21 Jul 2014 11:51:34 +0100
Subject: Added some error checking

---
 user.go | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

(limited to 'user.go')

diff --git a/user.go b/user.go
index b686f74..f2a5afd 100644
--- a/user.go
+++ b/user.go
@@ -8,14 +8,20 @@ type User struct {
     isAdmin bool
 }
 
-func createUser(db *sqlite3.Conn, name string, isAdmin bool) {
+func createUser(db *sqlite3.Conn, name string, isAdmin bool) (User, error) {
+
+    user := getUserForName(db, name)
 
     //Check if a user with this name already exists
-    if (getUserForName(db, name).id == 0) {
+    if (user.id == 0) {
         args := sqlite3.NamedArgs{"$a": name, "$b": isAdmin}
         sql := "INSERT INTO users (name, admin) VALUES ($a, $b)"
 
-        db.Exec(sql, args)
+        err := db.Exec(sql, args)
+
+        return User{id:db.LastInsertId(), name:name, isAdmin:isAdmin}, err
+    } else {
+        return user, nil
     }
 
 }
-- 
cgit v1.2.3