github.com/keysonZZZ/kmg@v0.0.0-20151121023212-05317bfd7d39/beegoBundle/BeegoExtension.go (about)

     1  package beegoBundle
     2  
     3  import (
     4  	"github.com/astaxie/beego/orm"
     5  	"github.com/bronze1man/kmg/kmgContext"
     6  	"github.com/bronze1man/kmg/kmgSql"
     7  
     8  	"time"
     9  )
    10  
    11  type tBeegoOrmKey struct{}
    12  
    13  var beegoOrmKey tBeegoOrmKey = tBeegoOrmKey{}
    14  
    15  func InitOrm() {
    16  	orm.RegisterDataBase("default", "mysql", kmgSql.GetDefaultDbConfig().GetDsn())
    17  	orm.SetDataBaseTZ("default", time.UTC)
    18  }
    19  
    20  func ContextGetOrm(c kmgContext.Context) orm.Ormer {
    21  	o, ok := c.Value(beegoOrmKey).(orm.Ormer)
    22  	if !ok {
    23  		o = orm.NewOrm()
    24  		c.SetValue(beegoOrmKey, o)
    25  	}
    26  	return o
    27  }