From 8587f84e254f63b7f7241269c63c873548974c32 Mon Sep 17 00:00:00 2001
From: XieWei <wuxiaoshen@shu.edu.cn>
Date: Sun, 9 Jun 2019 01:31:38 +0800
Subject: [PATCH] feat: gorm example
Former-commit-id: 56495ea07f26efa1a98729046ad678ba65c3a92c
---
_examples/orm/gorm/main.go | 176 +++++++++++++++++++++++++++++++++++++
1 file changed, 176 insertions(+)
create mode 100644 _examples/orm/gorm/main.go
diff --git a/_examples/orm/gorm/main.go b/_examples/orm/gorm/main.go
new file mode 100644
index 00000000..3199b996
--- /dev/null
+++ b/_examples/orm/gorm/main.go
@@ -0,0 +1,176 @@
+package main
+
+import (
+ "net/http"
+ "os"
+ "time"
+
+ "github.com/jinzhu/gorm"
+ "github.com/kataras/iris"
+ _ "github.com/mattn/go-sqlite3"
+)
+
+type User struct {
+ gorm.Model
+ Salt string `gorm:"type:varchar(255)" json:"salt"`
+ Username string `gorm:"type:varchar(32)" json:"username"`
+ Password string `gorm:"type:varchar(200);column:password" json:"-"`
+ Languages string `gorm:"type:varchar(200);column:languages" json:"languages"`
+}
+
+func (u User) TableName() string {
+ return "gorm_user"
+}
+
+type UserSerializer struct {
+ ID uint `json:"id"`
+ CreatedAt time.Time `json:"created_at"`
+ UpdatedAt time.Time `json:"updated_at"`
+ Salt string `json:"salt"`
+ UserName string `json:"user_name"`
+ Password string `json:"-"`
+ Languages string `json:"languages"`
+}
+
+func (self User) Serializer() UserSerializer {
+ return UserSerializer{
+ ID: self.ID,
+ CreatedAt: self.CreatedAt.Truncate(time.Second),
+ UpdatedAt: self.UpdatedAt.Truncate(time.Second),
+ Salt: self.Salt,
+ Password: self.Password,
+ Languages: self.Languages,
+ UserName: self.Username,
+ }
+}
+
+func main() {
+ app := iris.Default()
+ db, err := gorm.Open("sqlite3", "test.db")
+ db.LogMode(true) // show SQL logger
+ if err != nil {
+ app.Logger().Fatalf("connect to sqlite3 failed")
+ return
+ }
+ iris.RegisterOnInterrupt(func() {
+ defer db.Close()
+ })
+
+ if os.Getenv("ENV") != "" {
+ db.DropTableIfExists(&User{}) // drop table
+ }
+ db.AutoMigrate(&User{}) // create table: // AutoMigrate run auto migration for given models, will only add missing fields, won't delete/change current data
+
+ app.Post("/post_user", func(context iris.Context) {
+ var user User
+ user = User{
+ Username: "gorm",
+ Salt: "hash---",
+ Password: "admin",
+ Languages: "gorm",
+ }
+ if err := db.FirstOrCreate(&user); err == nil {
+ app.Logger().Fatalf("created one record failed: %s", err.Error)
+ context.JSON(iris.Map{
+ "code": http.StatusBadRequest,
+ "error": err.Error,
+ })
+ return
+ }
+ context.JSON(
+ iris.Map{
+ "code": http.StatusOK,
+ "data": user.Serializer(),
+ })
+ })
+
+ app.Get("/get_user/{id:uint}", func(context iris.Context) {
+ var user User
+ id, _ := context.Params().GetUint("id")
+ app.Logger().Println(id)
+ if err := db.Where("id = ?", int(id)).First(&user).Error; err != nil {
+ app.Logger().Fatalf("find one record failed: %t", err == nil)
+ context.JSON(iris.Map{
+ "code": http.StatusBadRequest,
+ "error": err.Error,
+ })
+ return
+ }
+ context.JSON(iris.Map{
+ "code": http.StatusOK,
+ "data": user.Serializer(),
+ })
+ })
+
+ app.Delete("/delete_user/{id:uint}", func(context iris.Context) {
+ id, _ := context.Params().GetUint("id")
+ if id == 0 {
+ context.JSON(iris.Map{
+ "code": http.StatusOK,
+ "detail": "query param id should not be nil",
+ })
+ return
+ }
+ var user User
+ if err := db.Where("id = ?", id).First(&user).Error; err != nil {
+ app.Logger().Fatalf("record not found")
+ context.JSON(iris.Map{
+ "code": http.StatusOK,
+ "detail": err.Error,
+ })
+ return
+ }
+ db.Delete(&user)
+ context.JSON(iris.Map{
+ "code": http.StatusOK,
+ "data": user.Serializer(),
+ })
+ })
+
+ app.Patch("/patch_user/{id:uint}", func(context iris.Context) {
+ id, _ := context.Params().GetUint("id")
+ if id == 0 {
+ context.JSON(iris.Map{
+ "code": http.StatusOK,
+ "detail": "query param id should not be nil",
+ })
+ return
+ }
+ var user User
+ tx := db.Begin()
+ if err := tx.Where("id = ?", id).First(&user).Error; err != nil {
+ app.Logger().Fatalf("record not found")
+ context.JSON(iris.Map{
+ "code": http.StatusOK,
+ "detail": err.Error,
+ })
+ return
+ }
+
+ var body patchParam
+ context.ReadJSON(&body)
+ app.Logger().Println(body)
+ if err := tx.Model(&user).Updates(map[string]interface{}{"username": body.Data.UserName, "password": body.Data.Password}).Error; err != nil {
+ app.Logger().Fatalf("update record failed")
+ tx.Rollback()
+ context.JSON(iris.Map{
+ "code": http.StatusBadRequest,
+ "error": err.Error,
+ })
+ return
+ }
+ tx.Commit()
+ context.JSON(iris.Map{
+ "code": http.StatusOK,
+ "data": user.Serializer(),
+ })
+ })
+ app.Run(iris.Addr(":8080"), iris.WithoutServerError(iris.ErrServerClosed))
+}
+
+type patchParam struct {
+ Data struct {
+ UserName string `json:"user_name" form:"user_name"`
+ Password string `json:"password" form:"password"`
+ } `json:"data"`
+}