Refine the context and Try to fix the nil issue.
This commit is contained in:
@ -1,6 +1,7 @@
|
||||
package rest
|
||||
|
||||
import (
|
||||
"github.com/eyebluecn/tank/code/core"
|
||||
"github.com/nu7hatch/gouuid"
|
||||
"time"
|
||||
)
|
||||
@ -14,7 +15,7 @@ func (this *UploadTokenDao) FindByUuid(uuid string) *UploadToken {
|
||||
|
||||
// Read
|
||||
var uploadToken = &UploadToken{}
|
||||
db := CONTEXT.GetDB().Where(&UploadToken{Base: Base{Uuid: uuid}}).First(uploadToken)
|
||||
db := core.CONTEXT.GetDB().Where(&UploadToken{Base: Base{Uuid: uuid}}).First(uploadToken)
|
||||
if db.Error != nil {
|
||||
return nil
|
||||
}
|
||||
@ -32,7 +33,7 @@ func (this *UploadTokenDao) Create(uploadToken *UploadToken) *UploadToken {
|
||||
uploadToken.CreateTime = time.Now()
|
||||
uploadToken.UpdateTime = time.Now()
|
||||
uploadToken.Sort = time.Now().UnixNano() / 1e6
|
||||
db := CONTEXT.GetDB().Create(uploadToken)
|
||||
db := core.CONTEXT.GetDB().Create(uploadToken)
|
||||
this.PanicError(db.Error)
|
||||
|
||||
return uploadToken
|
||||
@ -42,7 +43,7 @@ func (this *UploadTokenDao) Create(uploadToken *UploadToken) *UploadToken {
|
||||
func (this *UploadTokenDao) Save(uploadToken *UploadToken) *UploadToken {
|
||||
|
||||
uploadToken.UpdateTime = time.Now()
|
||||
db := CONTEXT.GetDB().Save(uploadToken)
|
||||
db := core.CONTEXT.GetDB().Save(uploadToken)
|
||||
this.PanicError(db.Error)
|
||||
|
||||
return uploadToken
|
||||
|
Reference in New Issue
Block a user