added adduser registeruser changepassword deluser changeuser

This commit is contained in:
wt
2025-05-27 22:07:50 +07:00
parent 55dbc36b92
commit c65d6d5be5
8 changed files with 211 additions and 24 deletions
+4 -3
View File
@@ -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
View File
@@ -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
}