/
main.go
113 lines (70 loc) · 1.59 KB
/
main.go
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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
package main
import (
"fmt"
"net/http"
"github.com/gin-gonic/gin"
"github.com/jinzhu/gorm"
_ "github.com/mattn/go-sqlite3"
)
var db *gorm.DB
type User struct {
ID int
FirstName string `gorm:"column:FirstName"`
LastName string
Age int
}
func init() {
var err error
db, err = gorm.Open("sqlite3", "./users.db")
db.LogMode(true)
if err != nil {
panic("Couldn't create db")
}
if !db.HasTable(&User{}) {
db.CreateTable(new(User))
}
}
func main() {
router := gin.New()
router.Use(func(c *gin.Context) {
c.Set("Random", "This Is my String")
})
router.GET("/test", func(c *gin.Context) {
c.String(http.StatusOK, "Testing...")
})
router.GET("/hello", sayHello)
router.GET("/hello/:name", sayHelloToUser)
router.POST("/user", addUser)
router.GET("/user", listAllUsers)
router.GET("/user/:id", getSingleUser)
router.Run(":9000")
}
func getSingleUser(c *gin.Context) {
id := c.Param("id")
user := User{}
db.Find(&user, id)
c.JSON(http.StatusOK, user)
}
func listAllUsers(c *gin.Context) {
users := []User{}
db.Find(&users, "FirstName = 'Matt2'")
c.JSON(http.StatusOK, users)
}
func addUser(c *gin.Context) {
user := User{}
if c.BindJSON(&user) != nil {
c.String(http.StatusInternalServerError, "Unable to parse JSON")
return
}
db.Create(&user)
c.String(http.StatusOK, "Inserted record")
}
func sayHelloToUser(c *gin.Context) {
name := c.Param("name")
c.String(http.StatusOK, fmt.Sprintf("Hello %s!", name))
}
func sayHello(c *gin.Context) {
string, _ := c.Get("Random")
fmt.Println(string)
c.String(http.StatusOK, "Hello!")
}