crud/service.go at master · chiwenchen/crud · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
package main

import "errors"

// UserService is for user table interaction
type UserService interface {
	FetchUser(id string) (*User, error)
	CreateUser(username string, regionID int) (*User, error)
	UpdateUser(id string, username string) (*User, error)
	DeleteUser(id string) (*User, error)
}

type userService struct{}

func (userService) FetchUser(id string) (*User, error) {
	if id == "" {
		return &User{}, errEmpty
	}

	var user User

	db, _ := InitDB()

	if err := db.Where("id = ?", id).First(&user).Error; err != nil {
		return &User{}, err
	}

	return &user, nil
}
func (userService) CreateUser(username string, regionID int) (*User, error) {

	user := User{Username: username, RegionID: regionID}

	db, _ := InitDB()

	db.Create(&user)

	return &user, errors.New("")
}

func (userService) UpdateUser(id string, username string) (*User, error) {
	db, _ := InitDB()

	var user User

	if err := db.Where("id = ?", id).First(&user).Error; err != nil {
		return &User{}, err
	}

	user.Username = username
	db.Save(&user)

	return &user, nil
}

func (userService) DeleteUser(id string) (*User, error) {
	db, _ := InitDB()

	var user User

	if err := db.Where("id = ?", id).First(&user).Error; err != nil {
		return &User{}, err
	}

	db.Delete(&user)

	return &user, nil
}