feat: update methods & improved demo

store/address.go

@@ -57,3 +57,13 @@ func ListAddressesByUserID(userID string) ([]*Address, error) {
 	})
 	return addresses, err
 }
+
+func UpdateAddress(a *Address) error {
+	data, err := json.Marshal(a)
+	if err != nil {
+		return err
+	}
+	return GetDB().Update(func(txn *badger.Txn) error {
+		return txn.Set([]byte(addressPrefix+a.ID), data)
+	})
+}

store/db.go

@@ -8,14 +8,13 @@ import (
 )
 
 var (
-	db     *badger.DB
-	once   sync.Once
-	dbPath = "./data"
+	db   *badger.DB
+	once sync.Once
 )
 
 func InitDB() {
 	once.Do(func() {
-		opts := badger.DefaultOptions(dbPath).WithLogger(nil)
+		opts := badger.DefaultOptions("").WithInMemory(true).WithLogger(nil)
 		var err error
 		db, err = badger.Open(opts)
 		if err != nil {

store/user.go

@@ -25,7 +25,6 @@ var (
 )
 
 func CreateUser(name, email, password string) (*User, error) {
-	// Enforce uniqueness
 	if _, err := findUserIDByEmail(email); err == nil {
 		return nil, ErrEmailExists
 	}
@@ -113,3 +112,13 @@ func ListUsers() ([]*User, error) {
 	})
 	return users, err
 }
+
+func UpdateUser(u *User) error {
+	data, err := json.Marshal(u)
+	if err != nil {
+		return err
+	}
+	return GetDB().Update(func(txn *badger.Txn) error {
+		return txn.Set([]byte(userPrefix+u.ID), data)
+	})
+}