堆栈内存溢出
  繁体   English   中英

如何在Revel Controller中访问Gorm？

[英]How to access Gorm in Revel Controller?

 原文  2014-08-05 15:14:37  0   4    mysql/ go/ go-gorm/ revel

问题描述

首先让我说这些是我在Go玩弄的第几天。

我正在尝试使用像Gorm这样的Revel框架：

app/controllers/gorm.go 

package controllers

import (
    "fmt"
    "go-testapp/app/models"

    _ "github.com/go-sql-driver/mysql"
    "github.com/jinzhu/gorm"
    "github.com/revel/revel"
)

var DB gorm.DB

func InitDB() {
    var err error
    DB, err = gorm.Open("mysql", "root:@/go-testapp?charset=utf8&parseTime=True")
    if err != nil {
        panic(err)
    }
    DB.LogMode(true)
    DB.AutoMigrate(models.User{})
}

type GormController struct {
    *revel.Controller
    DB *gorm.DB
}

app/controller/app.go 

package controllers

import (
    "fmt"
    "go-bingo/app/models"

    _ "github.com/go-sql-driver/mysql"
    "github.com/revel/revel"
)

type App struct {
    GormController
}

func (c App) Index() revel.Result {
    user := models.User{Name: "Jinzhu", Age: 18}

    fmt.Println(c.DB)
    c.DB.NewRecord(user)

    c.DB.Create(&user)

    return c.RenderJson(user)
}

运行后导致：

runtime error: invalid memory address or nil pointer dereference第19行的runtime error: invalid memory address or nil pointer dereference c.DB.NewRecord(user)

它成功创建了具有自动迁移的数据表，但我不知道如何在控制器中使用Gorm。

正确方向的任何暗示？

4 个解决方案

解决方案1
 9 已采纳  2014-09-22 14:19:31

重要的提示

它只是Revel 原始示例的GORP的替代品。 它带来了一些起源的陷阱。 此答案可用作原始答案的替代品。 但它并没有解决陷阱。

请看看这个unswer和@MaxGabriel的答案的评论 ，以解决陷阱。

我建议使用@ MaxGabriel的解决方案来保护您的应用程序免受某些慢速* DDoS攻击。 并减少（在某些情况下）DB压力。

原始答案

@rauyran权限 ，你必须在init函数内调用InitDB （进入controllers包）。

这里有完整的例子（太多）：

树 

/app
    /controllers
      app.go
      gorm.go
      init.go
    /models
      user.go
[...]

user.go 

// models/user.go
package models

import  "time" // if you need/want

type User struct {          // example user fields
    Id                    int64
    Name                  string
    EncryptedPassword     []byte
    Password              string      `sql:"-"`
    CreatedAt             time.Time
    UpdatedAt             time.Time
    DeletedAt             time.Time     // for soft delete
}

gorm.go 

//controllers/gorm.go
package controllers

import (
    "github.com/jinzhu/gorm"
    _ "github.com/lib/pq" // my example for postgres
    // short name for revel
    r "github.com/revel/revel"
    // YOUR APP NAME
    "yourappname/app/models"
    "database/sql"
)

// type: revel controller with `*gorm.DB`
// c.Txn will keep `Gdb *gorm.DB`
type GormController struct {
    *r.Controller
    Txn *gorm.DB
}

// it can be used for jobs
var Gdb *gorm.DB

// init db
func InitDB() {
    var err error
    // open db
    Gdb, err = gorm.Open("postgres", "user=uname dbname=udbname sslmode=disable password=supersecret")
    if err != nil {
        r.ERROR.Println("FATAL", err)
        panic( err )
    }
    Gdb.AutoMigrate(&models.User{})
    // unique index if need
    //Gdb.Model(&models.User{}).AddUniqueIndex("idx_user_name", "name")
}


// transactions

// This method fills the c.Txn before each transaction
func (c *GormController) Begin() r.Result {
    txn := Gdb.Begin()
    if txn.Error != nil {
        panic(txn.Error)
    }
    c.Txn = txn
    return nil
}

// This method clears the c.Txn after each transaction
func (c *GormController) Commit() r.Result {
    if c.Txn == nil {
        return nil
    }
    c.Txn.Commit()
    if err := c.Txn.Error; err != nil && err != sql.ErrTxDone {
        panic(err)
    }
    c.Txn = nil
    return nil
}

// This method clears the c.Txn after each transaction, too
func (c *GormController) Rollback() r.Result {
    if c.Txn == nil {
        return nil
    }
    c.Txn.Rollback()
    if err := c.Txn.Error; err != nil && err != sql.ErrTxDone {
        panic(err)
    }
    c.Txn = nil
    return nil
}

app.go 

package controllers

import(
    "github.com/revel/revel"
    "yourappname/app/models"
)

type App struct {
    GormController
}

func (c App) Index() revel.Result {
    user := models.User{Name: "Jinzhup"}
    c.Txn.NewRecord(user)
    c.Txn.Create(&user)
    return c.RenderJSON(user)
}

init.go 

package controllers
import "github.com/revel/revel"

func init() {
    revel.OnAppStart(InitDB) // invoke InitDB function before
    revel.InterceptMethod((*GormController).Begin, revel.BEFORE)
    revel.InterceptMethod((*GormController).Commit, revel.AFTER)
    revel.InterceptMethod((*GormController).Rollback, revel.FINALLY)
}

正如您所看到的，这就像为GORM修改了Revel的预订。

对我来说很好。 结果： 

{
  "Id": 5,
  "Name": "Jinzhup",
  "EncryptedPassword": null,
  "Password": "",
  "CreatedAt": "2014-09-22T17:55:14.828661062+04:00",
  "UpdatedAt": "2014-09-22T17:55:14.828661062+04:00",
  "DeletedAt": "0001-01-01T00:00:00Z"
}

解决方案2
 2  2014-08-05 15:25:58

您的错误是由于您尚未初始化c.DB数据库变量引起的，它仍然是零。

在您的controllers / init.go文件中，确保您正在调用revel.OnAppStart（InitDB）。 它应该看起来像这样： 

package controllers

import "github.com/revel/revel"

func init() {
    revel.OnAppStart(InitDB)
    // maybe some other init things
}

解决方案3
 2  2017-10-30 18:34:13

根据他的建议，这个答案来自@ IvanBlack的回答。 他的版本是Revel示例代码的直接翻译，但是我们发现了这个答案修复的原始代码的一些问题。 它的主要变化是：

  1. 整个HTTP请求不再由数据库事务包装。 包装整个HTTP请求会使事务打开的时间远远超过必要的时间，这可能会导致许多问题：

    • 您的事务将保留数据库上的锁，并且许多持有锁的事务可能导致死锁
    • 使用了更多的数据库资源
    • 如果您的HTTP请求主要不受SQL数据库访问限制，则会放大这些问题。 例如，如果您的HTTP请求在发出外部HTTP请求或访问其他数据库时阻塞了30秒，那么这些服务的减速可能会影响您的SQL数据库。

  2. 由于在HTTP请求结束时不再自动检查事务是否存在错误，因此只要进行数据库插入，就会检查错误。 这也更正确：想象一下，在将User结构插入数据库之后，然后将User.Id存储在另一个数据库中，如Redis。 如果数据库插入工作正常，但是如果它失败并且您没有立即检查错误，则可以将默认的int64值0插入Redis（之后仅回滚SQL事务）。

树 

/app
    /controllers
      app.go
      gorm.go
      init.go
    /models
      user.go
[...]

user.go 

// models/user.go
package models

import  "time" // if you need/want

type User struct {          // example user fields
    Id                    int64
    Name                  string
    EncryptedPassword     []byte
    Password              string      `sql:"-"`
    CreatedAt             time.Time
    UpdatedAt             time.Time
    DeletedAt             time.Time     // for soft delete
}

gorm.go 

//controllers/gorm.go
package controllers

import (
    "github.com/jinzhu/gorm"
    _ "github.com/lib/pq" // my example for postgres
    // short name for revel
    r "github.com/revel/revel"
)

// type: revel controller with `*gorm.DB`
type GormController struct {
    *r.Controller
    DB *gorm.DB
}

// it can be used for jobs
var Gdb *gorm.DB

// init db
func InitDB() {
    var err error
    // open db
    Gdb, err = gorm.Open("postgres", "user=USERNAME dbname=DBNAME sslmode=disable")
    Gdb.LogMode(true) // Print SQL statements
    if err != nil {
        r.ERROR.Println("FATAL", err)
        panic(err)
    }
}

func (c *GormController) SetDB() r.Result {
    c.DB = Gdb
    return nil
}

init.go 

package controllers
import "github.com/revel/revel"

func init() {
    revel.OnAppStart(InitDB) // invoke InitDB function before
    revel.InterceptMethod((*GormController).SetDB, revel.BEFORE)
}

app.go 

package controllers

import(
    "github.com/revel/revel"
    "yourappname/app/models"
)

type App struct {
    GormController
}

func (c App) Index() revel.Result {
    user := models.User{Name: "Jinzhup"} // Note: In practice you should initialize all struct fields
    if err := c.DB.Create(&user).Error; err != nil {
        panic(err)
    }
    return c.RenderJSON(user)
}

结果： 

{
  "Id": 5,
  "Name": "Jinzhup",
  "EncryptedPassword": null,
  "Password": "",
  "CreatedAt": "2014-09-22T17:55:14.828661062+04:00",
  "UpdatedAt": "2014-09-22T17:55:14.828661062+04:00",
  "DeletedAt": "0001-01-01T00:00:00Z"
}

解决方案4
 -1  2017-10-18 20:41:26

或者你需要传递指向AutoMigrate的指针？ 

 DB.AutoMigrate(&models.User{})

暂无
暂无

声明:本站的技术帖子网页，遵循CC BY-SA 4.0协议，如果您需要转载，请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

相关问题 使用MySQL数据库狂欢没有gorm 如何在gorm中链接连接？ golang gorm 访问底层mysql查询 如何避免 GORM 中的竞争条件 如何将多个更新 GORM MySql 查询合并为 1？ 如何在Gorm中留下连接子查询计数 GoLang 如何使用 GORM 加载嵌套对象 如何为gorm组织多个外键 如何在gorm model中传递动态表名 如何在golang中使用GORM在数据库之间切换？
相关标签
 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM