added adduser registeruser changepassword deluser changeuser
This commit is contained in:
+4
-3
@@ -3,16 +3,17 @@ package database
|
||||
import (
|
||||
"fmt"
|
||||
"form/model"
|
||||
"os"
|
||||
|
||||
"gorm.io/driver/sqlite"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
var db *gorm.DB
|
||||
var Db *gorm.DB
|
||||
|
||||
func Init() error {
|
||||
var err error
|
||||
db, err = gorm.Open(sqlite.Open("users.db"), &gorm.Config{})
|
||||
Db, err = gorm.Open(sqlite.Open(os.Getenv("USERS_DB")), &gorm.Config{})
|
||||
if err != nil {
|
||||
fmt.Println("failed to connect database")
|
||||
return err
|
||||
@@ -21,5 +22,5 @@ func Init() error {
|
||||
}
|
||||
|
||||
func Migrate() {
|
||||
db.AutoMigrate(&model.User{})
|
||||
Db.AutoMigrate(&model.User{})
|
||||
}
|
||||
|
||||
+6
-6
@@ -12,7 +12,7 @@ import (
|
||||
)
|
||||
|
||||
func CreateUser(user *model.User) error {
|
||||
e := db.Where(model.User{
|
||||
e := Db.Where(model.User{
|
||||
Username: user.Username,
|
||||
}).FirstOrCreate(&user).Error
|
||||
if e != nil {
|
||||
@@ -23,7 +23,7 @@ func CreateUser(user *model.User) error {
|
||||
|
||||
func GetUserById(id string) (*model.User, error) {
|
||||
var user model.User
|
||||
e := db.First(&user, "id = ?", id).Error
|
||||
e := Db.First(&user, "id = ?", id).Error
|
||||
if e != nil {
|
||||
if errors.Is(e, gorm.ErrRecordNotFound) {
|
||||
return nil, fmt.Errorf("user not found")
|
||||
@@ -35,7 +35,7 @@ func GetUserById(id string) (*model.User, error) {
|
||||
|
||||
func GetUserByUsername(username string) (*model.User, error) {
|
||||
var user model.User
|
||||
e := db.First(&user, "username = ?", username).Error
|
||||
e := Db.First(&user, "username = ?", username).Error
|
||||
if e != nil {
|
||||
if errors.Is(e, gorm.ErrRecordNotFound) {
|
||||
return nil, fmt.Errorf("user not found")
|
||||
@@ -47,14 +47,14 @@ func GetUserByUsername(username string) (*model.User, error) {
|
||||
|
||||
func DeleteUser(id string) error {
|
||||
var user model.User
|
||||
e := db.First(&user, "id = ?", id).Error
|
||||
e := Db.First(&user, "id = ?", id).Error
|
||||
if e != nil {
|
||||
if errors.Is(e, gorm.ErrRecordNotFound) {
|
||||
return fmt.Errorf("user not found")
|
||||
}
|
||||
return e
|
||||
}
|
||||
e = db.Delete(&user).Error
|
||||
e = Db.Unscoped().Delete(&user).Error
|
||||
if e != nil {
|
||||
if errors.Is(e, gorm.ErrRecordNotFound) {
|
||||
return fmt.Errorf("user not found")
|
||||
@@ -77,7 +77,7 @@ func GetUsersFromFile(filename string) []model.User {
|
||||
|
||||
func GetAllUsers() ([]model.User, error) {
|
||||
var users []model.User
|
||||
e := db.Find(&users).Error
|
||||
e := Db.Find(&users).Error
|
||||
if e != nil {
|
||||
return nil, e
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user