Refresh selected db after query

lewissteele · Dec 14, 2024 · c355c88 · c355c88
1 parent 8085b43
commit c355c88
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/internal/db/user.go b/internal/db/user.go
@@ -80,6 +80,8 @@ func Query(q string) ([]map[string]interface{}, error) {
 	var results []map[string]interface{}
 	err := Conn.Raw(q).Scan(&results).Error
 
+	go updateSelected()
+
 	return results, err
 }
 
@@ -90,6 +92,11 @@ func Port(d Driver) string {
 	return "3306"
 }
 
+func updateSelected() {
+	UserDB.Database = Selected()
+	LocalDB.Save(UserDB)
+}
+
 func dialector(u model.Database) gorm.Dialector {
 	var dialector gorm.Dialector