gorm demo

Author: jesse-XD

gorm/db/db.go

@@ -0,0 +1,25 @@
+package db
+
+import (
+	"github.com/jinzhu/gorm"
+	"sync"
+)
+
+
+var (
+	db *gorm.DB
+	once sync.Once
+)
+
+// 连接数据库
+func DB() *gorm.DB{
+	once.Do(func() {
+		d, err := gorm.Open("mysql", "root:root@/test?charset=utf8&parseTime=True&loc=Local")
+		if err != nil {
+			panic(err)
+		}
+		d.LogMode(true)
+		db = d
+	})
+	return db
+}

gorm/models/school.go

@@ -0,0 +1,27 @@
+package models
+
+import (
+	"github.com/jinzhu/gorm"
+)
+
+type User struct {
+	gorm.Model
+	Name        string
+	UserAddress []*UserAddress `gorm:"ForeignKey:user_id"`
+}
+
+func (User) TableName() string {
+	return "user"
+}
+
+type UserAddress struct {
+	gorm.Model
+	UserId   int64
+	Contacts string
+	Tel      string
+	Address  string
+}
+
+func (UserAddress) TableName() string {
+	return "user_address"
+}

gorm/models/user.go

@@ -0,0 +1,83 @@
+package gorm
+
+import (
+	"demo/gorm/db"
+	"demo/gorm/models"
+	"errors"
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/jinzhu/gorm"
+)
+
+type User struct {
+}
+
+// 添加
+func (u *User) Add(user *models.User) (errs []error) {
+	if errs = db.DB().Create(user).GetErrors(); len(errs) > 0 {
+		return
+	}
+	return
+}
+
+// GetById 根据 id 获取数据
+func (u *User) GetById(id int64, user *models.User) bool {
+	if db.DB().First(&user, id).RecordNotFound() {
+		return false
+	}
+
+	return true
+}
+
+// Update 修改数据
+func (u *User) Update(user *models.User) (errs []error) {
+	// 先判断是否存在
+	var old models.User
+	if !u.GetById(int64(user.ID), &old) {
+		errs = []error{errors.New("记录不存在")}
+		return
+	}
+	// 修改
+	if errs = db.DB().Model(&old).Save(user).GetErrors(); len(errs) > 0 {
+		return
+	}
+
+	return
+}
+
+// Delete 删除 force：false|true 是否物理删除
+func (u *User) Delete(id int64, force bool) (errs []error) {
+	// 先判断是否存在
+	var old models.User
+	if !u.GetById(id, &old) {
+		errs = []error{errors.New("记录不存在")}
+		return
+	}
+	d := db.DB()
+	if force {
+		// 物理删除
+		d = d.Unscoped()
+	}
+	if errs = d.Delete(&old).GetErrors(); len(errs) > 0 {
+		return
+	}
+
+	return
+}
+
+// List 列表
+func (u *User) List(param map[string]string) (userList []*models.User) {
+	d := db.DB()
+	if val, ok := param["name"]; ok && val != "" {
+		d = d.Where("name like ?", "%" + val + "%")
+	}
+	d = d.Preload("UserAddress", func(db *gorm.DB)*gorm.DB {
+		if val, ok := param["contacts"]; ok && val != "" {
+			db = db.Where("contacts like ?", "%" + val + "%")
+		}
+		if val, ok := param["tel"]; ok && val != "" {
+			db = db.Where("tel = ?", val)
+		}
+		return db
+	}).Find(&userList)
+	return
+}

gorm/school.go

@@ -0,0 +1,51 @@
+package gorm
+
+import (
+	"demo/gorm/models"
+	"encoding/json"
+	"testing"
+)
+
+func TestUser_Add(t *testing.T) {
+	var user models.User
+	user.Name = "Tom"
+	var addressList []*models.UserAddress
+	addressList = append(addressList, &models.UserAddress{Contacts: "Jesse", Tel: "15123123121", Address: "广州123"})
+	addressList = append(addressList, &models.UserAddress{Contacts: "Jesse2", Tel: "18123123121", Address: "深圳123"})
+	user.UserAddress = addressList
+	errs := (*User)(nil).Add(&user)
+	if len(errs) > 0 {
+		t.Fatal(errs)
+	}
+}
+
+func TestUser_Update(t *testing.T) {
+	var user models.User
+	user.ID = 2
+	user.Name = "Jack"
+	//var addressList []*models.UserAddress
+	//addressList = append(addressList, &models.UserAddress{Model: gorm.Model{ID: 1},Contacts: "Jesse11", Tel: "15123123121", Address: "广州123"})
+	//addressList = append(addressList, &models.UserAddress{Contacts: "Jesse2", Tel: "18123123121", Address: "深圳123"})
+	//user.UserAddress = addressList
+	errs := (*User)(nil).Update(&user)
+	if len(errs) > 0 {
+		t.Fatal(errs)
+	}
+}
+
+func TestUser_Delete(t *testing.T) {
+	id := 3
+	errs := (*User)(nil).Delete(int64(id), true)
+	if len(errs) > 0 {
+		t.Fatal(errs)
+	}
+}
+
+func TestUser_List(t *testing.T) {
+	param := make(map[string]string)
+	param["name"] = "to"
+	param["contacts"] = "Jesse2"
+	list := (*User)(nil).List(param)
+	s, _ := json.Marshal(list)
+	t.Log(string(s))
+}