add user crud interface in impl
This commit is contained in:
50
internal/user/gorm_store.go
Normal file
50
internal/user/gorm_store.go
Normal file
@@ -0,0 +1,50 @@
|
||||
package user
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type GormUserStore struct {
|
||||
db *gorm.DB
|
||||
}
|
||||
|
||||
func NewGormUserStore(db *gorm.DB) (*GormUserStore, error) {
|
||||
if db == nil {
|
||||
return nil, fmt.Errorf("db is nil")
|
||||
}
|
||||
return &GormUserStore{
|
||||
db: db,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (s *GormUserStore) Create(user *User) error {
|
||||
return s.db.Create(user).Error
|
||||
}
|
||||
|
||||
func (s *GormUserStore) GetByID(id int64) (*User, error) {
|
||||
var user User
|
||||
err := s.db.First(&user, id).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &user, nil
|
||||
}
|
||||
|
||||
func (s *GormUserStore) GetByUsername(username string) (*User, error) {
|
||||
var user User
|
||||
err := s.db.Where("username = ?", username).First(&user).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &user, nil
|
||||
}
|
||||
|
||||
func (s *GormUserStore) Update(user *User) error {
|
||||
return s.db.Save(user).Error
|
||||
}
|
||||
|
||||
func (s *GormUserStore) Delete(id int64) error {
|
||||
return s.db.Delete(&User{}, id).Error
|
||||
}
|
||||
11
internal/user/model.go
Normal file
11
internal/user/model.go
Normal file
@@ -0,0 +1,11 @@
|
||||
package user
|
||||
|
||||
import "gorm.io/gorm"
|
||||
|
||||
type User struct {
|
||||
ID int64 `gorm:"primaryKey"`
|
||||
Username string `gorm:"uniqueIndex;not null"`
|
||||
Email string `gorm:"uniqueIndex;not null"`
|
||||
Password string `gorm:"not null"`
|
||||
DeletedAt gorm.DeletedAt `gorm:"index"`
|
||||
}
|
||||
36
internal/user/service.go
Normal file
36
internal/user/service.go
Normal file
@@ -0,0 +1,36 @@
|
||||
package user
|
||||
|
||||
import "fmt"
|
||||
|
||||
type Service struct {
|
||||
store UserCRUD
|
||||
}
|
||||
|
||||
func NewService(store UserCRUD) (UserCRUD, error) {
|
||||
if store == nil {
|
||||
return nil, fmt.Errorf("store is nil")
|
||||
}
|
||||
return &Service{
|
||||
store: store,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (s *Service) Create(user *User) error {
|
||||
return s.store.Create(user)
|
||||
}
|
||||
|
||||
func (s *Service) GetByID(id int64) (*User, error) {
|
||||
return s.store.GetByID(id)
|
||||
}
|
||||
|
||||
func (s *Service) GetByUsername(username string) (*User, error) {
|
||||
return s.store.GetByUsername(username)
|
||||
}
|
||||
|
||||
func (s *Service) Update(user *User) error {
|
||||
return s.store.Update(user)
|
||||
}
|
||||
|
||||
func (s *Service) Delete(id int64) error {
|
||||
return s.store.Delete(id)
|
||||
}
|
||||
9
internal/user/store.go
Normal file
9
internal/user/store.go
Normal file
@@ -0,0 +1,9 @@
|
||||
package user
|
||||
|
||||
type UserCRUD interface {
|
||||
Create(user *User) error
|
||||
GetByID(id int64) (*User, error)
|
||||
GetByUsername(username string) (*User, error)
|
||||
Update(user *User) error
|
||||
Delete(id int64) error
|
||||
}
|
||||
84
internal/user/user_test.go
Normal file
84
internal/user/user_test.go
Normal file
@@ -0,0 +1,84 @@
|
||||
package user
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
|
||||
"gorm.io/driver/sqlite"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
func setupTestDB(t *testing.T) *gorm.DB {
|
||||
t.Helper()
|
||||
|
||||
dbPath := filepath.Join("testdata", "users.db")
|
||||
|
||||
_ = os.Remove(dbPath)
|
||||
|
||||
db, err := gorm.Open(sqlite.Open(dbPath), &gorm.Config{})
|
||||
if err != nil {
|
||||
t.Fatalf("failed to open db: %v", err)
|
||||
}
|
||||
|
||||
if err := db.AutoMigrate(&User{}); err != nil {
|
||||
t.Fatalf("failed to migrate: %v", err)
|
||||
}
|
||||
|
||||
return db
|
||||
}
|
||||
|
||||
func TestUsersCRUD(t *testing.T) {
|
||||
db := setupTestDB(t)
|
||||
|
||||
store, err := NewGormUserStore(db)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to create store: %v", err)
|
||||
}
|
||||
|
||||
service, err := NewService(store)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to create service: %v", err)
|
||||
}
|
||||
|
||||
user := &User{
|
||||
Username: "testuser",
|
||||
Email: "test@example.com",
|
||||
Password: "password123",
|
||||
}
|
||||
|
||||
if err := service.Create(user); err != nil {
|
||||
t.Fatalf("failed to create user: %v", err)
|
||||
}
|
||||
retrieved, err := service.GetByID(user.ID)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to get user by ID: %v", err)
|
||||
}
|
||||
if retrieved.Username != user.Username {
|
||||
t.Fatalf("expected username %s, got %s", user.Username, retrieved.Username)
|
||||
}
|
||||
|
||||
retrievedByUsername, err := service.GetByUsername(user.Username)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to get user by username: %v", err)
|
||||
}
|
||||
if retrievedByUsername.Email != user.Email {
|
||||
t.Fatalf("expected email %s, got %s", user.Email, retrievedByUsername.Email)
|
||||
}
|
||||
|
||||
user.Email = "newemail@example.com"
|
||||
if err := service.Update(user); err != nil {
|
||||
t.Fatalf("failed to update user: %v", err)
|
||||
}
|
||||
retrieved, err = service.GetByID(user.ID)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to get user by ID: %v", err)
|
||||
}
|
||||
if retrieved.Email != user.Email {
|
||||
t.Fatalf("expected email %s, got %s", user.Email, retrieved.Email)
|
||||
}
|
||||
err = service.Delete(user.ID)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to delete user: %v", err)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user