Use AfterLoad instead of AfterSet on Structs (#2628)
* use AfterLoad instead of AfterSet on Structs * fix the comments on AfterLoad * fix the comments on action AfterLoad
This commit is contained in:
parent
1ad902d529
commit
a8717e5e3a
35 changed files with 334 additions and 315 deletions
|
@ -16,7 +16,6 @@ import (
|
||||||
|
|
||||||
"github.com/Unknwon/com"
|
"github.com/Unknwon/com"
|
||||||
"github.com/go-xorm/builder"
|
"github.com/go-xorm/builder"
|
||||||
"github.com/go-xorm/xorm"
|
|
||||||
|
|
||||||
"code.gitea.io/git"
|
"code.gitea.io/git"
|
||||||
api "code.gitea.io/sdk/gitea"
|
api "code.gitea.io/sdk/gitea"
|
||||||
|
@ -91,13 +90,10 @@ type Action struct {
|
||||||
CreatedUnix int64 `xorm:"INDEX created"`
|
CreatedUnix int64 `xorm:"INDEX created"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet updates the webhook object upon setting a column.
|
// AfterLoad is invoked from XORM after setting the values of all fields of this object.
|
||||||
func (a *Action) AfterSet(colName string, _ xorm.Cell) {
|
func (a *Action) AfterLoad() {
|
||||||
switch colName {
|
|
||||||
case "created_unix":
|
|
||||||
a.Created = time.Unix(a.CreatedUnix, 0).Local()
|
a.Created = time.Unix(a.CreatedUnix, 0).Local()
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// GetOpType gets the ActionType of this action.
|
// GetOpType gets the ActionType of this action.
|
||||||
func (a *Action) GetOpType() ActionType {
|
func (a *Action) GetOpType() ActionType {
|
||||||
|
|
|
@ -12,7 +12,6 @@ import (
|
||||||
"code.gitea.io/gitea/modules/util"
|
"code.gitea.io/gitea/modules/util"
|
||||||
|
|
||||||
"github.com/Unknwon/com"
|
"github.com/Unknwon/com"
|
||||||
"github.com/go-xorm/xorm"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
//NoticeType describes the notice type
|
//NoticeType describes the notice type
|
||||||
|
@ -32,13 +31,10 @@ type Notice struct {
|
||||||
CreatedUnix int64 `xorm:"INDEX created"`
|
CreatedUnix int64 `xorm:"INDEX created"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet is invoked from XORM after setting the value of a field of this object.
|
// AfterLoad is invoked from XORM after setting the values of all fields of this object.
|
||||||
func (n *Notice) AfterSet(colName string, _ xorm.Cell) {
|
func (n *Notice) AfterLoad() {
|
||||||
switch colName {
|
|
||||||
case "created_unix":
|
|
||||||
n.Created = time.Unix(n.CreatedUnix, 0).Local()
|
n.Created = time.Unix(n.CreatedUnix, 0).Local()
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// TrStr returns a translation format string.
|
// TrStr returns a translation format string.
|
||||||
func (n *Notice) TrStr() string {
|
func (n *Notice) TrStr() string {
|
||||||
|
|
|
@ -12,7 +12,6 @@ import (
|
||||||
"path"
|
"path"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/go-xorm/xorm"
|
|
||||||
gouuid "github.com/satori/go.uuid"
|
gouuid "github.com/satori/go.uuid"
|
||||||
|
|
||||||
"code.gitea.io/gitea/modules/setting"
|
"code.gitea.io/gitea/modules/setting"
|
||||||
|
@ -31,14 +30,11 @@ type Attachment struct {
|
||||||
CreatedUnix int64 `xorm:"created"`
|
CreatedUnix int64 `xorm:"created"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet is invoked from XORM after setting the value of a field of
|
// AfterLoad is invoked from XORM after setting the value of a field of
|
||||||
// this object.
|
// this object.
|
||||||
func (a *Attachment) AfterSet(colName string, _ xorm.Cell) {
|
func (a *Attachment) AfterLoad() {
|
||||||
switch colName {
|
|
||||||
case "created_unix":
|
|
||||||
a.Created = time.Unix(a.CreatedUnix, 0).Local()
|
a.Created = time.Unix(a.CreatedUnix, 0).Local()
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// IncreaseDownloadCount is update download count + 1
|
// IncreaseDownloadCount is update download count + 1
|
||||||
func (a *Attachment) IncreaseDownloadCount() error {
|
func (a *Attachment) IncreaseDownloadCount() error {
|
||||||
|
@ -133,6 +129,10 @@ func GetAttachmentsByIssueID(issueID int64) ([]*Attachment, error) {
|
||||||
|
|
||||||
// GetAttachmentsByCommentID returns all attachments if comment by given ID.
|
// GetAttachmentsByCommentID returns all attachments if comment by given ID.
|
||||||
func GetAttachmentsByCommentID(commentID int64) ([]*Attachment, error) {
|
func GetAttachmentsByCommentID(commentID int64) ([]*Attachment, error) {
|
||||||
|
return getAttachmentsByCommentID(x, commentID)
|
||||||
|
}
|
||||||
|
|
||||||
|
func getAttachmentsByCommentID(e Engine, commentID int64) ([]*Attachment, error) {
|
||||||
attachments := make([]*Attachment, 0, 10)
|
attachments := make([]*Attachment, 0, 10)
|
||||||
return attachments, x.Where("comment_id=?", commentID).Find(&attachments)
|
return attachments, x.Where("comment_id=?", commentID).Find(&attachments)
|
||||||
}
|
}
|
||||||
|
|
|
@ -52,17 +52,15 @@ func (key *GPGKey) BeforeInsert() {
|
||||||
key.CreatedUnix = key.Created.Unix()
|
key.CreatedUnix = key.Created.Unix()
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet is invoked from XORM after setting the value of a field of this object.
|
// AfterLoad is invoked from XORM after setting the values of all fields of this object.
|
||||||
func (key *GPGKey) AfterSet(colName string, _ xorm.Cell) {
|
func (key *GPGKey) AfterLoad(session *xorm.Session) {
|
||||||
switch colName {
|
|
||||||
case "key_id":
|
|
||||||
x.Where("primary_key_id=?", key.KeyID).Find(&key.SubsKey)
|
|
||||||
case "added_unix":
|
|
||||||
key.Added = time.Unix(key.AddedUnix, 0).Local()
|
key.Added = time.Unix(key.AddedUnix, 0).Local()
|
||||||
case "expired_unix":
|
|
||||||
key.Expired = time.Unix(key.ExpiredUnix, 0).Local()
|
key.Expired = time.Unix(key.ExpiredUnix, 0).Local()
|
||||||
case "created_unix":
|
|
||||||
key.Created = time.Unix(key.CreatedUnix, 0).Local()
|
key.Created = time.Unix(key.CreatedUnix, 0).Local()
|
||||||
|
|
||||||
|
err := session.Where("primary_key_id=?", key.KeyID).Find(&key.SubsKey)
|
||||||
|
if err != nil {
|
||||||
|
log.Error(3, "Find Sub GPGkeys[%d]: %v", key.KeyID, err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -67,18 +67,13 @@ func (issue *Issue) BeforeUpdate() {
|
||||||
issue.DeadlineUnix = issue.Deadline.Unix()
|
issue.DeadlineUnix = issue.Deadline.Unix()
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet is invoked from XORM after setting the value of a field of
|
// AfterLoad is invoked from XORM after setting the value of a field of
|
||||||
// this object.
|
// this object.
|
||||||
func (issue *Issue) AfterSet(colName string, _ xorm.Cell) {
|
func (issue *Issue) AfterLoad() {
|
||||||
switch colName {
|
|
||||||
case "deadline_unix":
|
|
||||||
issue.Deadline = time.Unix(issue.DeadlineUnix, 0).Local()
|
issue.Deadline = time.Unix(issue.DeadlineUnix, 0).Local()
|
||||||
case "created_unix":
|
|
||||||
issue.Created = time.Unix(issue.CreatedUnix, 0).Local()
|
issue.Created = time.Unix(issue.CreatedUnix, 0).Local()
|
||||||
case "updated_unix":
|
|
||||||
issue.Updated = time.Unix(issue.UpdatedUnix, 0).Local()
|
issue.Updated = time.Unix(issue.UpdatedUnix, 0).Local()
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
func (issue *Issue) loadRepo(e Engine) (err error) {
|
func (issue *Issue) loadRepo(e Engine) (err error) {
|
||||||
if issue.Repo == nil {
|
if issue.Repo == nil {
|
||||||
|
|
|
@ -112,31 +112,26 @@ type Comment struct {
|
||||||
ShowTag CommentTag `xorm:"-"`
|
ShowTag CommentTag `xorm:"-"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet is invoked from XORM after setting the value of a field of this object.
|
// AfterLoad is invoked from XORM after setting the values of all fields of this object.
|
||||||
func (c *Comment) AfterSet(colName string, _ xorm.Cell) {
|
func (c *Comment) AfterLoad(session *xorm.Session) {
|
||||||
|
c.Created = time.Unix(c.CreatedUnix, 0).Local()
|
||||||
|
c.Updated = time.Unix(c.UpdatedUnix, 0).Local()
|
||||||
|
|
||||||
var err error
|
var err error
|
||||||
switch colName {
|
c.Attachments, err = getAttachmentsByCommentID(session, c.ID)
|
||||||
case "id":
|
|
||||||
c.Attachments, err = GetAttachmentsByCommentID(c.ID)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Error(3, "GetAttachmentsByCommentID[%d]: %v", c.ID, err)
|
log.Error(3, "getAttachmentsByCommentID[%d]: %v", c.ID, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
case "poster_id":
|
c.Poster, err = getUserByID(session, c.PosterID)
|
||||||
c.Poster, err = GetUserByID(c.PosterID)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if IsErrUserNotExist(err) {
|
if IsErrUserNotExist(err) {
|
||||||
c.PosterID = -1
|
c.PosterID = -1
|
||||||
c.Poster = NewGhostUser()
|
c.Poster = NewGhostUser()
|
||||||
} else {
|
} else {
|
||||||
log.Error(3, "GetUserByID[%d]: %v", c.ID, err)
|
log.Error(3, "getUserByID[%d]: %v", c.ID, err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
case "created_unix":
|
|
||||||
c.Created = time.Unix(c.CreatedUnix, 0).Local()
|
|
||||||
case "updated_unix":
|
|
||||||
c.Updated = time.Unix(c.UpdatedUnix, 0).Local()
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterDelete is invoked from XORM after the object is deleted.
|
// AfterDelete is invoked from XORM after the object is deleted.
|
||||||
|
|
|
@ -51,14 +51,10 @@ func (m *Milestone) BeforeUpdate() {
|
||||||
m.ClosedDateUnix = m.ClosedDate.Unix()
|
m.ClosedDateUnix = m.ClosedDate.Unix()
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet is invoked from XORM after setting the value of a field of
|
// AfterLoad is invoked from XORM after setting the value of a field of
|
||||||
// this object.
|
// this object.
|
||||||
func (m *Milestone) AfterSet(colName string, _ xorm.Cell) {
|
func (m *Milestone) AfterLoad() {
|
||||||
switch colName {
|
|
||||||
case "num_closed_issues":
|
|
||||||
m.NumOpenIssues = m.NumIssues - m.NumClosedIssues
|
m.NumOpenIssues = m.NumIssues - m.NumClosedIssues
|
||||||
|
|
||||||
case "deadline_unix":
|
|
||||||
m.Deadline = time.Unix(m.DeadlineUnix, 0).Local()
|
m.Deadline = time.Unix(m.DeadlineUnix, 0).Local()
|
||||||
if m.Deadline.Year() == 9999 {
|
if m.Deadline.Year() == 9999 {
|
||||||
return
|
return
|
||||||
|
@ -69,10 +65,8 @@ func (m *Milestone) AfterSet(colName string, _ xorm.Cell) {
|
||||||
m.IsOverDue = true
|
m.IsOverDue = true
|
||||||
}
|
}
|
||||||
|
|
||||||
case "closed_date_unix":
|
|
||||||
m.ClosedDate = time.Unix(m.ClosedDateUnix, 0).Local()
|
m.ClosedDate = time.Unix(m.ClosedDateUnix, 0).Local()
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// State returns string representation of milestone status.
|
// State returns string representation of milestone status.
|
||||||
func (m *Milestone) State() api.StateType {
|
func (m *Milestone) State() api.StateType {
|
||||||
|
|
|
@ -7,8 +7,6 @@ package models
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/go-xorm/xorm"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// Stopwatch represents a stopwatch for time tracking.
|
// Stopwatch represents a stopwatch for time tracking.
|
||||||
|
@ -26,14 +24,10 @@ func (s *Stopwatch) BeforeInsert() {
|
||||||
s.CreatedUnix = time.Now().Unix()
|
s.CreatedUnix = time.Now().Unix()
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet is invoked from XORM after setting the value of a field of this object.
|
// AfterLoad is invoked from XORM after setting the values of all fields of this object.
|
||||||
func (s *Stopwatch) AfterSet(colName string, _ xorm.Cell) {
|
func (s *Stopwatch) AfterLoad() {
|
||||||
switch colName {
|
|
||||||
|
|
||||||
case "created_unix":
|
|
||||||
s.Created = time.Unix(s.CreatedUnix, 0).Local()
|
s.Created = time.Unix(s.CreatedUnix, 0).Local()
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
func getStopwatch(e Engine, userID, issueID int64) (sw *Stopwatch, exists bool, err error) {
|
func getStopwatch(e Engine, userID, issueID int64) (sw *Stopwatch, exists bool, err error) {
|
||||||
sw = new(Stopwatch)
|
sw = new(Stopwatch)
|
||||||
|
|
|
@ -8,7 +8,6 @@ import (
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/go-xorm/builder"
|
"github.com/go-xorm/builder"
|
||||||
"github.com/go-xorm/xorm"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// TrackedTime represents a time that was spent for a specific issue.
|
// TrackedTime represents a time that was spent for a specific issue.
|
||||||
|
@ -17,23 +16,14 @@ type TrackedTime struct {
|
||||||
IssueID int64 `xorm:"INDEX" json:"issue_id"`
|
IssueID int64 `xorm:"INDEX" json:"issue_id"`
|
||||||
UserID int64 `xorm:"INDEX" json:"user_id"`
|
UserID int64 `xorm:"INDEX" json:"user_id"`
|
||||||
Created time.Time `xorm:"-" json:"created"`
|
Created time.Time `xorm:"-" json:"created"`
|
||||||
CreatedUnix int64 `json:"-"`
|
CreatedUnix int64 `xorm:"created" json:"-"`
|
||||||
Time int64 `json:"time"`
|
Time int64 `json:"time"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// BeforeInsert will be invoked by XORM before inserting a record
|
// AfterLoad is invoked from XORM after setting the values of all fields of this object.
|
||||||
// representing this object.
|
func (t *TrackedTime) AfterLoad() {
|
||||||
func (t *TrackedTime) BeforeInsert() {
|
|
||||||
t.CreatedUnix = time.Now().Unix()
|
|
||||||
}
|
|
||||||
|
|
||||||
// AfterSet is invoked from XORM after setting the value of a field of this object.
|
|
||||||
func (t *TrackedTime) AfterSet(colName string, _ xorm.Cell) {
|
|
||||||
switch colName {
|
|
||||||
case "created_unix":
|
|
||||||
t.Created = time.Unix(t.CreatedUnix, 0).Local()
|
t.Created = time.Unix(t.CreatedUnix, 0).Local()
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// FindTrackedTimesOptions represent the filters for tracked times. If an ID is 0 it will be ignored.
|
// FindTrackedTimesOptions represent the filters for tracked times. If an ID is 0 it will be ignored.
|
||||||
type FindTrackedTimesOptions struct {
|
type FindTrackedTimesOptions struct {
|
||||||
|
|
|
@ -3,8 +3,6 @@ package models
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/go-xorm/xorm"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// LFSMetaObject stores metadata for LFS tracked files.
|
// LFSMetaObject stores metadata for LFS tracked files.
|
||||||
|
@ -109,10 +107,7 @@ func RemoveLFSMetaObjectByOid(oid string) error {
|
||||||
return sess.Commit()
|
return sess.Commit()
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet stores the LFSMetaObject creation time in the database as local time.
|
// AfterLoad stores the LFSMetaObject creation time in the database as local time.
|
||||||
func (m *LFSMetaObject) AfterSet(colName string, _ xorm.Cell) {
|
func (m *LFSMetaObject) AfterLoad() {
|
||||||
switch colName {
|
|
||||||
case "created_unix":
|
|
||||||
m.Created = time.Unix(m.CreatedUnix, 0).Local()
|
m.Created = time.Unix(m.CreatedUnix, 0).Local()
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
|
@ -183,15 +183,11 @@ func (source *LoginSource) BeforeSet(colName string, val xorm.Cell) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet is invoked from XORM after setting the value of a field of this object.
|
// AfterLoad is invoked from XORM after setting the values of all fields of this object.
|
||||||
func (source *LoginSource) AfterSet(colName string, _ xorm.Cell) {
|
func (source *LoginSource) AfterLoad() {
|
||||||
switch colName {
|
|
||||||
case "created_unix":
|
|
||||||
source.Created = time.Unix(source.CreatedUnix, 0).Local()
|
source.Created = time.Unix(source.CreatedUnix, 0).Local()
|
||||||
case "updated_unix":
|
|
||||||
source.Updated = time.Unix(source.UpdatedUnix, 0).Local()
|
source.Updated = time.Unix(source.UpdatedUnix, 0).Local()
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// TypeName return name of this login source type.
|
// TypeName return name of this login source type.
|
||||||
func (source *LoginSource) TypeName() string {
|
func (source *LoginSource) TypeName() string {
|
||||||
|
|
|
@ -80,18 +80,15 @@ func (pr *PullRequest) BeforeUpdate() {
|
||||||
pr.MergedUnix = pr.Merged.Unix()
|
pr.MergedUnix = pr.Merged.Unix()
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet is invoked from XORM after setting the value of a field of this object.
|
// AfterLoad is invoked from XORM after setting the values of all fields of this object.
|
||||||
// Note: don't try to get Issue because will end up recursive querying.
|
// Note: don't try to get Issue because will end up recursive querying.
|
||||||
func (pr *PullRequest) AfterSet(colName string, _ xorm.Cell) {
|
func (pr *PullRequest) AfterLoad() {
|
||||||
switch colName {
|
|
||||||
case "merged_unix":
|
|
||||||
if !pr.HasMerged {
|
if !pr.HasMerged {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
pr.Merged = time.Unix(pr.MergedUnix, 0).Local()
|
pr.Merged = time.Unix(pr.MergedUnix, 0).Local()
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// Note: don't try to get Issue because will end up recursive querying.
|
// Note: don't try to get Issue because will end up recursive querying.
|
||||||
func (pr *PullRequest) loadAttributes(e Engine) (err error) {
|
func (pr *PullRequest) loadAttributes(e Engine) (err error) {
|
||||||
|
|
|
@ -15,7 +15,6 @@ import (
|
||||||
"code.gitea.io/gitea/modules/setting"
|
"code.gitea.io/gitea/modules/setting"
|
||||||
api "code.gitea.io/sdk/gitea"
|
api "code.gitea.io/sdk/gitea"
|
||||||
"github.com/go-xorm/builder"
|
"github.com/go-xorm/builder"
|
||||||
"github.com/go-xorm/xorm"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// Release represents a release of repository.
|
// Release represents a release of repository.
|
||||||
|
@ -50,13 +49,10 @@ func (r *Release) BeforeInsert() {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet is invoked from XORM after setting the value of a field of this object.
|
// AfterLoad is invoked from XORM after setting the values of all fields of this object.
|
||||||
func (r *Release) AfterSet(colName string, _ xorm.Cell) {
|
func (r *Release) AfterLoad() {
|
||||||
switch colName {
|
|
||||||
case "created_unix":
|
|
||||||
r.Created = time.Unix(r.CreatedUnix, 0).Local()
|
r.Created = time.Unix(r.CreatedUnix, 0).Local()
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Release) loadAttributes(e Engine) error {
|
func (r *Release) loadAttributes(e Engine) error {
|
||||||
var err error
|
var err error
|
||||||
|
|
|
@ -216,26 +216,19 @@ type Repository struct {
|
||||||
UpdatedUnix int64 `xorm:"INDEX updated"`
|
UpdatedUnix int64 `xorm:"INDEX updated"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet is invoked from XORM after setting the value of a field of this object.
|
// AfterLoad is invoked from XORM after setting the values of all fields of this object.
|
||||||
func (repo *Repository) AfterSet(colName string, _ xorm.Cell) {
|
func (repo *Repository) AfterLoad() {
|
||||||
switch colName {
|
|
||||||
case "default_branch":
|
|
||||||
// FIXME: use models migration to solve all at once.
|
// FIXME: use models migration to solve all at once.
|
||||||
if len(repo.DefaultBranch) == 0 {
|
if len(repo.DefaultBranch) == 0 {
|
||||||
repo.DefaultBranch = "master"
|
repo.DefaultBranch = "master"
|
||||||
}
|
}
|
||||||
case "num_closed_issues":
|
|
||||||
repo.NumOpenIssues = repo.NumIssues - repo.NumClosedIssues
|
repo.NumOpenIssues = repo.NumIssues - repo.NumClosedIssues
|
||||||
case "num_closed_pulls":
|
|
||||||
repo.NumOpenPulls = repo.NumPulls - repo.NumClosedPulls
|
repo.NumOpenPulls = repo.NumPulls - repo.NumClosedPulls
|
||||||
case "num_closed_milestones":
|
|
||||||
repo.NumOpenMilestones = repo.NumMilestones - repo.NumClosedMilestones
|
repo.NumOpenMilestones = repo.NumMilestones - repo.NumClosedMilestones
|
||||||
case "created_unix":
|
|
||||||
repo.Created = time.Unix(repo.CreatedUnix, 0).Local()
|
repo.Created = time.Unix(repo.CreatedUnix, 0).Local()
|
||||||
case "updated_unix":
|
|
||||||
repo.Updated = time.Unix(repo.UpdatedUnix, 0)
|
repo.Updated = time.Unix(repo.UpdatedUnix, 0)
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// MustOwner always returns a valid *User object to avoid
|
// MustOwner always returns a valid *User object to avoid
|
||||||
// conceptually impossible error handling.
|
// conceptually impossible error handling.
|
||||||
|
|
|
@ -55,25 +55,21 @@ func (m *Mirror) BeforeUpdate() {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet is invoked from XORM after setting the value of a field of this object.
|
// AfterLoad is invoked from XORM after setting the values of all fields of this object.
|
||||||
func (m *Mirror) AfterSet(colName string, _ xorm.Cell) {
|
func (m *Mirror) AfterLoad(session *xorm.Session) {
|
||||||
if m == nil {
|
if m == nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var err error
|
var err error
|
||||||
switch colName {
|
m.Repo, err = getRepositoryByID(session, m.RepoID)
|
||||||
case "repo_id":
|
|
||||||
m.Repo, err = GetRepositoryByID(m.RepoID)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Error(3, "GetRepositoryByID[%d]: %v", m.ID, err)
|
log.Error(3, "getRepositoryByID[%d]: %v", m.ID, err)
|
||||||
}
|
}
|
||||||
case "updated_unix":
|
|
||||||
m.Updated = time.Unix(m.UpdatedUnix, 0).Local()
|
m.Updated = time.Unix(m.UpdatedUnix, 0).Local()
|
||||||
case "next_update_unix":
|
|
||||||
m.NextUpdate = time.Unix(m.NextUpdateUnix, 0).Local()
|
m.NextUpdate = time.Unix(m.NextUpdateUnix, 0).Local()
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// ScheduleNextUpdate calculates and sets next update time.
|
// ScheduleNextUpdate calculates and sets next update time.
|
||||||
func (m *Mirror) ScheduleNextUpdate() {
|
func (m *Mirror) ScheduleNextUpdate() {
|
||||||
|
|
|
@ -106,13 +106,10 @@ func (r *RepoUnit) BeforeSet(colName string, val xorm.Cell) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet is invoked from XORM after setting the value of a field of this object.
|
// AfterLoad is invoked from XORM after setting the values of all fields of this object.
|
||||||
func (r *RepoUnit) AfterSet(colName string, _ xorm.Cell) {
|
func (r *RepoUnit) AfterLoad() {
|
||||||
switch colName {
|
|
||||||
case "created_unix":
|
|
||||||
r.Created = time.Unix(r.CreatedUnix, 0).Local()
|
r.Created = time.Unix(r.CreatedUnix, 0).Local()
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// Unit returns Unit
|
// Unit returns Unit
|
||||||
func (r *RepoUnit) Unit() Unit {
|
func (r *RepoUnit) Unit() Unit {
|
||||||
|
|
|
@ -56,23 +56,19 @@ type PublicKey struct {
|
||||||
|
|
||||||
Created time.Time `xorm:"-"`
|
Created time.Time `xorm:"-"`
|
||||||
CreatedUnix int64 `xorm:"created"`
|
CreatedUnix int64 `xorm:"created"`
|
||||||
Updated time.Time `xorm:"-"` // Note: Updated must below Created for AfterSet.
|
Updated time.Time `xorm:"-"`
|
||||||
UpdatedUnix int64 `xorm:"updated"`
|
UpdatedUnix int64 `xorm:"updated"`
|
||||||
HasRecentActivity bool `xorm:"-"`
|
HasRecentActivity bool `xorm:"-"`
|
||||||
HasUsed bool `xorm:"-"`
|
HasUsed bool `xorm:"-"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet is invoked from XORM after setting the value of a field of this object.
|
// AfterLoad is invoked from XORM after setting the values of all fields of this object.
|
||||||
func (key *PublicKey) AfterSet(colName string, _ xorm.Cell) {
|
func (key *PublicKey) AfterLoad() {
|
||||||
switch colName {
|
|
||||||
case "created_unix":
|
|
||||||
key.Created = time.Unix(key.CreatedUnix, 0).Local()
|
key.Created = time.Unix(key.CreatedUnix, 0).Local()
|
||||||
case "updated_unix":
|
|
||||||
key.Updated = time.Unix(key.UpdatedUnix, 0).Local()
|
key.Updated = time.Unix(key.UpdatedUnix, 0).Local()
|
||||||
key.HasUsed = key.Updated.After(key.Created)
|
key.HasUsed = key.Updated.After(key.Created)
|
||||||
key.HasRecentActivity = key.Updated.Add(7 * 24 * time.Hour).After(time.Now())
|
key.HasRecentActivity = key.Updated.Add(7 * 24 * time.Hour).After(time.Now())
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// OmitEmail returns content of public key without email address.
|
// OmitEmail returns content of public key without email address.
|
||||||
func (key *PublicKey) OmitEmail() string {
|
func (key *PublicKey) OmitEmail() string {
|
||||||
|
@ -612,23 +608,19 @@ type DeployKey struct {
|
||||||
|
|
||||||
Created time.Time `xorm:"-"`
|
Created time.Time `xorm:"-"`
|
||||||
CreatedUnix int64 `xorm:"created"`
|
CreatedUnix int64 `xorm:"created"`
|
||||||
Updated time.Time `xorm:"-"` // Note: Updated must below Created for AfterSet.
|
Updated time.Time `xorm:"-"`
|
||||||
UpdatedUnix int64 `xorm:"updated"`
|
UpdatedUnix int64 `xorm:"updated"`
|
||||||
HasRecentActivity bool `xorm:"-"`
|
HasRecentActivity bool `xorm:"-"`
|
||||||
HasUsed bool `xorm:"-"`
|
HasUsed bool `xorm:"-"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet is invoked from XORM after setting the value of a field of this object.
|
// AfterLoad is invoked from XORM after setting the values of all fields of this object.
|
||||||
func (key *DeployKey) AfterSet(colName string, _ xorm.Cell) {
|
func (key *DeployKey) AfterLoad() {
|
||||||
switch colName {
|
|
||||||
case "created_unix":
|
|
||||||
key.Created = time.Unix(key.CreatedUnix, 0).Local()
|
key.Created = time.Unix(key.CreatedUnix, 0).Local()
|
||||||
case "updated_unix":
|
|
||||||
key.Updated = time.Unix(key.UpdatedUnix, 0).Local()
|
key.Updated = time.Unix(key.UpdatedUnix, 0).Local()
|
||||||
key.HasUsed = key.Updated.After(key.Created)
|
key.HasUsed = key.Updated.After(key.Created)
|
||||||
key.HasRecentActivity = key.Updated.Add(7 * 24 * time.Hour).After(time.Now())
|
key.HasRecentActivity = key.Updated.Add(7 * 24 * time.Hour).After(time.Now())
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// GetContent gets associated public key content.
|
// GetContent gets associated public key content.
|
||||||
func (key *DeployKey) GetContent() error {
|
func (key *DeployKey) GetContent() error {
|
||||||
|
|
|
@ -71,16 +71,12 @@ type CommitStatus struct {
|
||||||
UpdatedUnix int64 `xorm:"INDEX updated"`
|
UpdatedUnix int64 `xorm:"INDEX updated"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet is invoked from XORM after setting the value of a field of
|
// AfterLoad is invoked from XORM after setting the value of a field of
|
||||||
// this object.
|
// this object.
|
||||||
func (status *CommitStatus) AfterSet(colName string, _ xorm.Cell) {
|
func (status *CommitStatus) AfterLoad() {
|
||||||
switch colName {
|
|
||||||
case "created_unix":
|
|
||||||
status.Created = time.Unix(status.CreatedUnix, 0).Local()
|
status.Created = time.Unix(status.CreatedUnix, 0).Local()
|
||||||
case "updated_unix":
|
|
||||||
status.Updated = time.Unix(status.UpdatedUnix, 0).Local()
|
status.Updated = time.Unix(status.UpdatedUnix, 0).Local()
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
func (status *CommitStatus) loadRepo(e Engine) (err error) {
|
func (status *CommitStatus) loadRepo(e Engine) (err error) {
|
||||||
if status.Repo == nil {
|
if status.Repo == nil {
|
||||||
|
|
|
@ -7,7 +7,6 @@ package models
|
||||||
import (
|
import (
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/go-xorm/xorm"
|
|
||||||
gouuid "github.com/satori/go.uuid"
|
gouuid "github.com/satori/go.uuid"
|
||||||
|
|
||||||
"code.gitea.io/gitea/modules/base"
|
"code.gitea.io/gitea/modules/base"
|
||||||
|
@ -22,23 +21,19 @@ type AccessToken struct {
|
||||||
|
|
||||||
Created time.Time `xorm:"-"`
|
Created time.Time `xorm:"-"`
|
||||||
CreatedUnix int64 `xorm:"INDEX created"`
|
CreatedUnix int64 `xorm:"INDEX created"`
|
||||||
Updated time.Time `xorm:"-"` // Note: Updated must below Created for AfterSet.
|
Updated time.Time `xorm:"-"`
|
||||||
UpdatedUnix int64 `xorm:"INDEX updated"`
|
UpdatedUnix int64 `xorm:"INDEX updated"`
|
||||||
HasRecentActivity bool `xorm:"-"`
|
HasRecentActivity bool `xorm:"-"`
|
||||||
HasUsed bool `xorm:"-"`
|
HasUsed bool `xorm:"-"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet is invoked from XORM after setting the value of a field of this object.
|
// AfterLoad is invoked from XORM after setting the values of all fields of this object.
|
||||||
func (t *AccessToken) AfterSet(colName string, _ xorm.Cell) {
|
func (t *AccessToken) AfterLoad() {
|
||||||
switch colName {
|
|
||||||
case "created_unix":
|
|
||||||
t.Created = time.Unix(t.CreatedUnix, 0).Local()
|
t.Created = time.Unix(t.CreatedUnix, 0).Local()
|
||||||
case "updated_unix":
|
|
||||||
t.Updated = time.Unix(t.UpdatedUnix, 0).Local()
|
t.Updated = time.Unix(t.UpdatedUnix, 0).Local()
|
||||||
t.HasUsed = t.Updated.After(t.Created)
|
t.HasUsed = t.Updated.After(t.Created)
|
||||||
t.HasRecentActivity = t.Updated.Add(7 * 24 * time.Hour).After(time.Now())
|
t.HasRecentActivity = t.Updated.Add(7 * 24 * time.Hour).After(time.Now())
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// NewAccessToken creates new access token.
|
// NewAccessToken creates new access token.
|
||||||
func NewAccessToken(t *AccessToken) error {
|
func NewAccessToken(t *AccessToken) error {
|
||||||
|
|
|
@ -11,7 +11,6 @@ import (
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/Unknwon/com"
|
"github.com/Unknwon/com"
|
||||||
"github.com/go-xorm/xorm"
|
|
||||||
"github.com/pquerna/otp/totp"
|
"github.com/pquerna/otp/totp"
|
||||||
|
|
||||||
"code.gitea.io/gitea/modules/base"
|
"code.gitea.io/gitea/modules/base"
|
||||||
|
@ -27,19 +26,15 @@ type TwoFactor struct {
|
||||||
|
|
||||||
Created time.Time `xorm:"-"`
|
Created time.Time `xorm:"-"`
|
||||||
CreatedUnix int64 `xorm:"INDEX created"`
|
CreatedUnix int64 `xorm:"INDEX created"`
|
||||||
Updated time.Time `xorm:"-"` // Note: Updated must below Created for AfterSet.
|
Updated time.Time `xorm:"-"`
|
||||||
UpdatedUnix int64 `xorm:"INDEX updated"`
|
UpdatedUnix int64 `xorm:"INDEX updated"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet is invoked from XORM after setting the value of a field of this object.
|
// AfterLoad is invoked from XORM after setting the values of all fields of this object.
|
||||||
func (t *TwoFactor) AfterSet(colName string, _ xorm.Cell) {
|
func (t *TwoFactor) AfterLoad() {
|
||||||
switch colName {
|
|
||||||
case "created_unix":
|
|
||||||
t.Created = time.Unix(t.CreatedUnix, 0).Local()
|
t.Created = time.Unix(t.CreatedUnix, 0).Local()
|
||||||
case "updated_unix":
|
|
||||||
t.Updated = time.Unix(t.UpdatedUnix, 0).Local()
|
t.Updated = time.Unix(t.UpdatedUnix, 0).Local()
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// GenerateScratchToken recreates the scratch token the user is using.
|
// GenerateScratchToken recreates the scratch token the user is using.
|
||||||
func (t *TwoFactor) GenerateScratchToken() error {
|
func (t *TwoFactor) GenerateScratchToken() error {
|
||||||
|
|
|
@ -153,17 +153,12 @@ func (u *User) UpdateDiffViewStyle(style string) error {
|
||||||
return UpdateUserCols(u, "diff_view_style")
|
return UpdateUserCols(u, "diff_view_style")
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet is invoked from XORM after setting the value of a field of this object.
|
// AfterLoad is invoked from XORM after setting the values of all fields of this object.
|
||||||
func (u *User) AfterSet(colName string, _ xorm.Cell) {
|
func (u *User) AfterLoad() {
|
||||||
switch colName {
|
|
||||||
case "created_unix":
|
|
||||||
u.Created = time.Unix(u.CreatedUnix, 0).Local()
|
u.Created = time.Unix(u.CreatedUnix, 0).Local()
|
||||||
case "updated_unix":
|
|
||||||
u.Updated = time.Unix(u.UpdatedUnix, 0).Local()
|
u.Updated = time.Unix(u.UpdatedUnix, 0).Local()
|
||||||
case "last_login_unix":
|
|
||||||
u.LastLogin = time.Unix(u.LastLoginUnix, 0).Local()
|
u.LastLogin = time.Unix(u.LastLoginUnix, 0).Local()
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// getEmail returns an noreply email, if the user has set to keep his
|
// getEmail returns an noreply email, if the user has set to keep his
|
||||||
// email address private, otherwise the primary email address.
|
// email address private, otherwise the primary email address.
|
||||||
|
|
|
@ -19,7 +19,6 @@ import (
|
||||||
"code.gitea.io/gitea/modules/sync"
|
"code.gitea.io/gitea/modules/sync"
|
||||||
api "code.gitea.io/sdk/gitea"
|
api "code.gitea.io/sdk/gitea"
|
||||||
|
|
||||||
"github.com/go-xorm/xorm"
|
|
||||||
gouuid "github.com/satori/go.uuid"
|
gouuid "github.com/satori/go.uuid"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -112,21 +111,16 @@ type Webhook struct {
|
||||||
UpdatedUnix int64 `xorm:"INDEX updated"`
|
UpdatedUnix int64 `xorm:"INDEX updated"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet updates the webhook object upon setting a column
|
// AfterLoad updates the webhook object upon setting a column
|
||||||
func (w *Webhook) AfterSet(colName string, _ xorm.Cell) {
|
func (w *Webhook) AfterLoad() {
|
||||||
var err error
|
|
||||||
switch colName {
|
|
||||||
case "events":
|
|
||||||
w.HookEvent = &HookEvent{}
|
w.HookEvent = &HookEvent{}
|
||||||
if err = json.Unmarshal([]byte(w.Events), w.HookEvent); err != nil {
|
if err := json.Unmarshal([]byte(w.Events), w.HookEvent); err != nil {
|
||||||
log.Error(3, "Unmarshal[%d]: %v", w.ID, err)
|
log.Error(3, "Unmarshal[%d]: %v", w.ID, err)
|
||||||
}
|
}
|
||||||
case "created_unix":
|
|
||||||
w.Created = time.Unix(w.CreatedUnix, 0).Local()
|
w.Created = time.Unix(w.CreatedUnix, 0).Local()
|
||||||
case "updated_unix":
|
|
||||||
w.Updated = time.Unix(w.UpdatedUnix, 0).Local()
|
w.Updated = time.Unix(w.UpdatedUnix, 0).Local()
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
// GetSlackHook returns slack metadata
|
// GetSlackHook returns slack metadata
|
||||||
func (w *Webhook) GetSlackHook() *SlackMeta {
|
func (w *Webhook) GetSlackHook() *SlackMeta {
|
||||||
|
@ -432,33 +426,18 @@ func (t *HookTask) BeforeUpdate() {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// AfterSet updates the webhook object upon setting a column
|
// AfterLoad updates the webhook object upon setting a column
|
||||||
func (t *HookTask) AfterSet(colName string, _ xorm.Cell) {
|
func (t *HookTask) AfterLoad() {
|
||||||
var err error
|
|
||||||
switch colName {
|
|
||||||
case "delivered":
|
|
||||||
t.DeliveredString = time.Unix(0, t.Delivered).Format("2006-01-02 15:04:05 MST")
|
t.DeliveredString = time.Unix(0, t.Delivered).Format("2006-01-02 15:04:05 MST")
|
||||||
|
|
||||||
case "request_content":
|
|
||||||
if len(t.RequestContent) == 0 {
|
if len(t.RequestContent) == 0 {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
t.RequestInfo = &HookRequest{}
|
t.RequestInfo = &HookRequest{}
|
||||||
if err = json.Unmarshal([]byte(t.RequestContent), t.RequestInfo); err != nil {
|
if err := json.Unmarshal([]byte(t.RequestContent), t.RequestInfo); err != nil {
|
||||||
log.Error(3, "Unmarshal[%d]: %v", t.ID, err)
|
log.Error(3, "Unmarshal[%d]: %v", t.ID, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
case "response_content":
|
|
||||||
if len(t.ResponseContent) == 0 {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
t.ResponseInfo = &HookResponse{}
|
|
||||||
if err = json.Unmarshal([]byte(t.ResponseContent), t.ResponseInfo); err != nil {
|
|
||||||
log.Error(3, "Unmarshal [%d]: %v", t.ID, err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (t *HookTask) simpleMarshalJSON(v interface{}) string {
|
func (t *HookTask) simpleMarshalJSON(v interface{}) string {
|
||||||
|
|
17
vendor/github.com/go-xorm/xorm/engine.go
generated
vendored
17
vendor/github.com/go-xorm/xorm/engine.go
generated
vendored
|
@ -1516,10 +1516,14 @@ func (engine *Engine) Import(r io.Reader) ([]sql.Result, error) {
|
||||||
return results, lastError
|
return results, lastError
|
||||||
}
|
}
|
||||||
|
|
||||||
// NowTime2 return current time
|
// nowTime return current time
|
||||||
func (engine *Engine) NowTime2(sqlTypeName string) (interface{}, time.Time) {
|
func (engine *Engine) nowTime(col *core.Column) (interface{}, time.Time) {
|
||||||
t := time.Now()
|
t := time.Now()
|
||||||
return engine.formatTime(sqlTypeName, t.In(engine.DatabaseTZ)), t.In(engine.TZLocation)
|
var tz = engine.DatabaseTZ
|
||||||
|
if !col.DisableTimeZone && col.TimeZone != nil {
|
||||||
|
tz = col.TimeZone
|
||||||
|
}
|
||||||
|
return engine.formatTime(col.SQLType.Name, t.In(tz)), t.In(engine.TZLocation)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (engine *Engine) formatColTime(col *core.Column, t time.Time) (v interface{}) {
|
func (engine *Engine) formatColTime(col *core.Column, t time.Time) (v interface{}) {
|
||||||
|
@ -1574,3 +1578,10 @@ func (engine *Engine) CondDeleted(colName string) builder.Cond {
|
||||||
}
|
}
|
||||||
return builder.IsNull{colName}.Or(builder.Eq{colName: zeroTime1})
|
return builder.IsNull{colName}.Or(builder.Eq{colName: zeroTime1})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// BufferSize sets buffer size for iterate
|
||||||
|
func (engine *Engine) BufferSize(size int) *Session {
|
||||||
|
session := engine.NewSession()
|
||||||
|
session.isAutoClose = true
|
||||||
|
return session.BufferSize(size)
|
||||||
|
}
|
||||||
|
|
2
vendor/github.com/go-xorm/xorm/helpers.go
generated
vendored
2
vendor/github.com/go-xorm/xorm/helpers.go
generated
vendored
|
@ -422,7 +422,7 @@ func genCols(table *core.Table, session *Session, bean interface{}, useCol bool,
|
||||||
|
|
||||||
if (col.IsCreated || col.IsUpdated) && session.statement.UseAutoTime /*&& isZero(fieldValue.Interface())*/ {
|
if (col.IsCreated || col.IsUpdated) && session.statement.UseAutoTime /*&& isZero(fieldValue.Interface())*/ {
|
||||||
// if time is non-empty, then set to auto time
|
// if time is non-empty, then set to auto time
|
||||||
val, t := session.engine.NowTime2(col.SQLType.Name)
|
val, t := session.engine.nowTime(col)
|
||||||
args = append(args, val)
|
args = append(args, val)
|
||||||
|
|
||||||
var colName = col.Name
|
var colName = col.Name
|
||||||
|
|
40
vendor/github.com/go-xorm/xorm/processors.go
generated
vendored
40
vendor/github.com/go-xorm/xorm/processors.go
generated
vendored
|
@ -29,13 +29,6 @@ type AfterSetProcessor interface {
|
||||||
AfterSet(string, Cell)
|
AfterSet(string, Cell)
|
||||||
}
|
}
|
||||||
|
|
||||||
// !nashtsai! TODO enable BeforeValidateProcessor when xorm start to support validations
|
|
||||||
//// Executed before an object is validated
|
|
||||||
//type BeforeValidateProcessor interface {
|
|
||||||
// BeforeValidate()
|
|
||||||
//}
|
|
||||||
// --
|
|
||||||
|
|
||||||
// AfterInsertProcessor executed after an object is persisted to the database
|
// AfterInsertProcessor executed after an object is persisted to the database
|
||||||
type AfterInsertProcessor interface {
|
type AfterInsertProcessor interface {
|
||||||
AfterInsert()
|
AfterInsert()
|
||||||
|
@ -50,3 +43,36 @@ type AfterUpdateProcessor interface {
|
||||||
type AfterDeleteProcessor interface {
|
type AfterDeleteProcessor interface {
|
||||||
AfterDelete()
|
AfterDelete()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// AfterLoadProcessor executed after an ojbect has been loaded from database
|
||||||
|
type AfterLoadProcessor interface {
|
||||||
|
AfterLoad()
|
||||||
|
}
|
||||||
|
|
||||||
|
// AfterLoadSessionProcessor executed after an ojbect has been loaded from database with session parameter
|
||||||
|
type AfterLoadSessionProcessor interface {
|
||||||
|
AfterLoad(*Session)
|
||||||
|
}
|
||||||
|
|
||||||
|
type executedProcessorFunc func(*Session, interface{}) error
|
||||||
|
|
||||||
|
type executedProcessor struct {
|
||||||
|
fun executedProcessorFunc
|
||||||
|
session *Session
|
||||||
|
bean interface{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (executor *executedProcessor) execute() error {
|
||||||
|
return executor.fun(executor.session, executor.bean)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (session *Session) executeProcessors() error {
|
||||||
|
processors := session.afterProcessors
|
||||||
|
session.afterProcessors = make([]executedProcessor, 0)
|
||||||
|
for _, processor := range processors {
|
||||||
|
if err := processor.execute(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
8
vendor/github.com/go-xorm/xorm/rows.go
generated
vendored
8
vendor/github.com/go-xorm/xorm/rows.go
generated
vendored
|
@ -99,15 +99,19 @@ func (rows *Rows) Scan(bean interface{}) error {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
scanResults, err := rows.session.row2Slice(rows.rows, rows.fields, len(rows.fields), bean)
|
scanResults, err := rows.session.row2Slice(rows.rows, rows.fields, bean)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err = rows.session.slice2Bean(scanResults, rows.fields, len(rows.fields), bean, &dataStruct, rows.session.statement.RefTable)
|
_, err = rows.session.slice2Bean(scanResults, rows.fields, bean, &dataStruct, rows.session.statement.RefTable)
|
||||||
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
return rows.session.executeProcessors()
|
||||||
|
}
|
||||||
|
|
||||||
// Close session if session.IsAutoClose is true, and claimed any opened resources
|
// Close session if session.IsAutoClose is true, and claimed any opened resources
|
||||||
func (rows *Rows) Close() error {
|
func (rows *Rows) Close() error {
|
||||||
if rows.session.isAutoClose {
|
if rows.session.isAutoClose {
|
||||||
|
|
60
vendor/github.com/go-xorm/xorm/session.go
generated
vendored
60
vendor/github.com/go-xorm/xorm/session.go
generated
vendored
|
@ -41,6 +41,8 @@ type Session struct {
|
||||||
beforeClosures []func(interface{})
|
beforeClosures []func(interface{})
|
||||||
afterClosures []func(interface{})
|
afterClosures []func(interface{})
|
||||||
|
|
||||||
|
afterProcessors []executedProcessor
|
||||||
|
|
||||||
prepareStmt bool
|
prepareStmt bool
|
||||||
stmtCache map[uint32]*core.Stmt //key: hash.Hash32 of (queryStr, len(queryStr))
|
stmtCache map[uint32]*core.Stmt //key: hash.Hash32 of (queryStr, len(queryStr))
|
||||||
|
|
||||||
|
@ -75,6 +77,8 @@ func (session *Session) Init() {
|
||||||
session.beforeClosures = make([]func(interface{}), 0)
|
session.beforeClosures = make([]func(interface{}), 0)
|
||||||
session.afterClosures = make([]func(interface{}), 0)
|
session.afterClosures = make([]func(interface{}), 0)
|
||||||
|
|
||||||
|
session.afterProcessors = make([]executedProcessor, 0)
|
||||||
|
|
||||||
session.lastSQL = ""
|
session.lastSQL = ""
|
||||||
session.lastSQLArgs = []interface{}{}
|
session.lastSQLArgs = []interface{}{}
|
||||||
}
|
}
|
||||||
|
@ -296,37 +300,40 @@ func (session *Session) getField(dataStruct *reflect.Value, key string, table *c
|
||||||
// Cell cell is a result of one column field
|
// Cell cell is a result of one column field
|
||||||
type Cell *interface{}
|
type Cell *interface{}
|
||||||
|
|
||||||
func (session *Session) rows2Beans(rows *core.Rows, fields []string, fieldsCount int,
|
func (session *Session) rows2Beans(rows *core.Rows, fields []string,
|
||||||
table *core.Table, newElemFunc func([]string) reflect.Value,
|
table *core.Table, newElemFunc func([]string) reflect.Value,
|
||||||
sliceValueSetFunc func(*reflect.Value, core.PK) error) error {
|
sliceValueSetFunc func(*reflect.Value, core.PK) error) error {
|
||||||
for rows.Next() {
|
for rows.Next() {
|
||||||
var newValue = newElemFunc(fields)
|
var newValue = newElemFunc(fields)
|
||||||
bean := newValue.Interface()
|
bean := newValue.Interface()
|
||||||
dataStruct := rValue(bean)
|
dataStruct := newValue.Elem()
|
||||||
|
|
||||||
// handle beforeClosures
|
// handle beforeClosures
|
||||||
scanResults, err := session.row2Slice(rows, fields, fieldsCount, bean)
|
scanResults, err := session.row2Slice(rows, fields, bean)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
pk, err := session.slice2Bean(scanResults, fields, fieldsCount, bean, &dataStruct, table)
|
pk, err := session.slice2Bean(scanResults, fields, bean, &dataStruct, table)
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
err = sliceValueSetFunc(&newValue, pk)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
session.afterProcessors = append(session.afterProcessors, executedProcessor{
|
||||||
|
fun: func(*Session, interface{}) error {
|
||||||
|
return sliceValueSetFunc(&newValue, pk)
|
||||||
|
},
|
||||||
|
session: session,
|
||||||
|
bean: bean,
|
||||||
|
})
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (session *Session) row2Slice(rows *core.Rows, fields []string, fieldsCount int, bean interface{}) ([]interface{}, error) {
|
func (session *Session) row2Slice(rows *core.Rows, fields []string, bean interface{}) ([]interface{}, error) {
|
||||||
for _, closure := range session.beforeClosures {
|
for _, closure := range session.beforeClosures {
|
||||||
closure(bean)
|
closure(bean)
|
||||||
}
|
}
|
||||||
|
|
||||||
scanResults := make([]interface{}, fieldsCount)
|
scanResults := make([]interface{}, len(fields))
|
||||||
for i := 0; i < len(fields); i++ {
|
for i := 0; i < len(fields); i++ {
|
||||||
var cell interface{}
|
var cell interface{}
|
||||||
scanResults[i] = &cell
|
scanResults[i] = &cell
|
||||||
|
@ -343,19 +350,48 @@ func (session *Session) row2Slice(rows *core.Rows, fields []string, fieldsCount
|
||||||
return scanResults, nil
|
return scanResults, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (session *Session) slice2Bean(scanResults []interface{}, fields []string, fieldsCount int, bean interface{}, dataStruct *reflect.Value, table *core.Table) (core.PK, error) {
|
func (session *Session) slice2Bean(scanResults []interface{}, fields []string, bean interface{}, dataStruct *reflect.Value, table *core.Table) (core.PK, error) {
|
||||||
defer func() {
|
defer func() {
|
||||||
if b, hasAfterSet := bean.(AfterSetProcessor); hasAfterSet {
|
if b, hasAfterSet := bean.(AfterSetProcessor); hasAfterSet {
|
||||||
for ii, key := range fields {
|
for ii, key := range fields {
|
||||||
b.AfterSet(key, Cell(scanResults[ii].(*interface{})))
|
b.AfterSet(key, Cell(scanResults[ii].(*interface{})))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
// handle afterClosures
|
// handle afterClosures
|
||||||
for _, closure := range session.afterClosures {
|
for _, closure := range session.afterClosures {
|
||||||
|
session.afterProcessors = append(session.afterProcessors, executedProcessor{
|
||||||
|
fun: func(sess *Session, bean interface{}) error {
|
||||||
closure(bean)
|
closure(bean)
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
session: session,
|
||||||
|
bean: bean,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
if a, has := bean.(AfterLoadProcessor); has {
|
||||||
|
session.afterProcessors = append(session.afterProcessors, executedProcessor{
|
||||||
|
fun: func(sess *Session, bean interface{}) error {
|
||||||
|
a.AfterLoad()
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
session: session,
|
||||||
|
bean: bean,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
if a, has := bean.(AfterLoadSessionProcessor); has {
|
||||||
|
session.afterProcessors = append(session.afterProcessors, executedProcessor{
|
||||||
|
fun: func(sess *Session, bean interface{}) error {
|
||||||
|
a.AfterLoad(sess)
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
session: session,
|
||||||
|
bean: bean,
|
||||||
|
})
|
||||||
}
|
}
|
||||||
}()
|
|
||||||
|
|
||||||
var tempMap = make(map[string]int)
|
var tempMap = make(map[string]int)
|
||||||
var pk core.PK
|
var pk core.PK
|
||||||
|
|
4
vendor/github.com/go-xorm/xorm/session_delete.go
generated
vendored
4
vendor/github.com/go-xorm/xorm/session_delete.go
generated
vendored
|
@ -184,12 +184,12 @@ func (session *Session) Delete(bean interface{}) (int64, error) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// !oinume! Insert NowTime to the head of session.statement.Params
|
// !oinume! Insert nowTime to the head of session.statement.Params
|
||||||
condArgs = append(condArgs, "")
|
condArgs = append(condArgs, "")
|
||||||
paramsLen := len(condArgs)
|
paramsLen := len(condArgs)
|
||||||
copy(condArgs[1:paramsLen], condArgs[0:paramsLen-1])
|
copy(condArgs[1:paramsLen], condArgs[0:paramsLen-1])
|
||||||
|
|
||||||
val, t := session.engine.NowTime2(deletedColumn.SQLType.Name)
|
val, t := session.engine.nowTime(deletedColumn)
|
||||||
condArgs[0] = val
|
condArgs[0] = val
|
||||||
|
|
||||||
var colName = deletedColumn.Name
|
var colName = deletedColumn.Name
|
||||||
|
|
7
vendor/github.com/go-xorm/xorm/session_find.go
generated
vendored
7
vendor/github.com/go-xorm/xorm/session_find.go
generated
vendored
|
@ -239,7 +239,12 @@ func (session *Session) noCacheFind(table *core.Table, containerValue reflect.Va
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
return session.rows2Beans(rows, fields, len(fields), tb, newElemFunc, containerValueSetFunc)
|
err = session.rows2Beans(rows, fields, tb, newElemFunc, containerValueSetFunc)
|
||||||
|
rows.Close()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return session.executeProcessors()
|
||||||
}
|
}
|
||||||
|
|
||||||
for rows.Next() {
|
for rows.Next() {
|
||||||
|
|
9
vendor/github.com/go-xorm/xorm/session_get.go
generated
vendored
9
vendor/github.com/go-xorm/xorm/session_get.go
generated
vendored
|
@ -87,7 +87,7 @@ func (session *Session) nocacheGet(beanKind reflect.Kind, table *core.Table, bea
|
||||||
return true, err
|
return true, err
|
||||||
}
|
}
|
||||||
|
|
||||||
scanResults, err := session.row2Slice(rows, fields, len(fields), bean)
|
scanResults, err := session.row2Slice(rows, fields, bean)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return false, err
|
return false, err
|
||||||
}
|
}
|
||||||
|
@ -95,7 +95,12 @@ func (session *Session) nocacheGet(beanKind reflect.Kind, table *core.Table, bea
|
||||||
rows.Close()
|
rows.Close()
|
||||||
|
|
||||||
dataStruct := rValue(bean)
|
dataStruct := rValue(bean)
|
||||||
_, err = session.slice2Bean(scanResults, fields, len(fields), bean, &dataStruct, table)
|
_, err = session.slice2Bean(scanResults, fields, bean, &dataStruct, table)
|
||||||
|
if err != nil {
|
||||||
|
return true, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return true, session.executeProcessors()
|
||||||
case reflect.Slice:
|
case reflect.Slice:
|
||||||
err = rows.ScanSlice(bean)
|
err = rows.ScanSlice(bean)
|
||||||
case reflect.Map:
|
case reflect.Map:
|
||||||
|
|
4
vendor/github.com/go-xorm/xorm/session_insert.go
generated
vendored
4
vendor/github.com/go-xorm/xorm/session_insert.go
generated
vendored
|
@ -126,7 +126,7 @@ func (session *Session) innerInsertMulti(rowsSlicePtr interface{}) (int64, error
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (col.IsCreated || col.IsUpdated) && session.statement.UseAutoTime {
|
if (col.IsCreated || col.IsUpdated) && session.statement.UseAutoTime {
|
||||||
val, t := session.engine.NowTime2(col.SQLType.Name)
|
val, t := session.engine.nowTime(col)
|
||||||
args = append(args, val)
|
args = append(args, val)
|
||||||
|
|
||||||
var colName = col.Name
|
var colName = col.Name
|
||||||
|
@ -181,7 +181,7 @@ func (session *Session) innerInsertMulti(rowsSlicePtr interface{}) (int64, error
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (col.IsCreated || col.IsUpdated) && session.statement.UseAutoTime {
|
if (col.IsCreated || col.IsUpdated) && session.statement.UseAutoTime {
|
||||||
val, t := session.engine.NowTime2(col.SQLType.Name)
|
val, t := session.engine.nowTime(col)
|
||||||
args = append(args, val)
|
args = append(args, val)
|
||||||
|
|
||||||
var colName = col.Name
|
var colName = col.Name
|
||||||
|
|
50
vendor/github.com/go-xorm/xorm/session_iterate.go
generated
vendored
50
vendor/github.com/go-xorm/xorm/session_iterate.go
generated
vendored
|
@ -23,6 +23,10 @@ func (session *Session) Iterate(bean interface{}, fun IterFunc) error {
|
||||||
defer session.Close()
|
defer session.Close()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if session.statement.bufferSize > 0 {
|
||||||
|
return session.bufferIterate(bean, fun)
|
||||||
|
}
|
||||||
|
|
||||||
rows, err := session.Rows(bean)
|
rows, err := session.Rows(bean)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
|
@ -44,3 +48,49 @@ func (session *Session) Iterate(bean interface{}, fun IterFunc) error {
|
||||||
}
|
}
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// BufferSize sets the buffersize for iterate
|
||||||
|
func (session *Session) BufferSize(size int) *Session {
|
||||||
|
session.statement.bufferSize = size
|
||||||
|
return session
|
||||||
|
}
|
||||||
|
|
||||||
|
func (session *Session) bufferIterate(bean interface{}, fun IterFunc) error {
|
||||||
|
if session.isAutoClose {
|
||||||
|
defer session.Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
var bufferSize = session.statement.bufferSize
|
||||||
|
var limit = session.statement.LimitN
|
||||||
|
if limit > 0 && bufferSize > limit {
|
||||||
|
bufferSize = limit
|
||||||
|
}
|
||||||
|
var start = session.statement.Start
|
||||||
|
v := rValue(bean)
|
||||||
|
sliceType := reflect.SliceOf(v.Type())
|
||||||
|
var idx = 0
|
||||||
|
for {
|
||||||
|
slice := reflect.New(sliceType)
|
||||||
|
if err := session.Limit(bufferSize, start).find(slice.Interface(), bean); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
for i := 0; i < slice.Elem().Len(); i++ {
|
||||||
|
if err := fun(idx, slice.Elem().Index(i).Addr().Interface()); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
idx++
|
||||||
|
}
|
||||||
|
|
||||||
|
start = start + slice.Elem().Len()
|
||||||
|
if limit > 0 && idx+bufferSize > limit {
|
||||||
|
bufferSize = limit - idx
|
||||||
|
}
|
||||||
|
|
||||||
|
if bufferSize <= 0 || slice.Elem().Len() < bufferSize || idx == limit {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
2
vendor/github.com/go-xorm/xorm/session_update.go
generated
vendored
2
vendor/github.com/go-xorm/xorm/session_update.go
generated
vendored
|
@ -205,7 +205,7 @@ func (session *Session) Update(bean interface{}, condiBean ...interface{}) (int6
|
||||||
if _, ok := session.statement.columnMap[strings.ToLower(table.Updated)]; !ok {
|
if _, ok := session.statement.columnMap[strings.ToLower(table.Updated)]; !ok {
|
||||||
colNames = append(colNames, session.engine.Quote(table.Updated)+" = ?")
|
colNames = append(colNames, session.engine.Quote(table.Updated)+" = ?")
|
||||||
col := table.UpdatedColumn()
|
col := table.UpdatedColumn()
|
||||||
val, t := session.engine.NowTime2(col.SQLType.Name)
|
val, t := session.engine.nowTime(col)
|
||||||
args = append(args, val)
|
args = append(args, val)
|
||||||
|
|
||||||
var colName = col.Name
|
var colName = col.Name
|
||||||
|
|
2
vendor/github.com/go-xorm/xorm/statement.go
generated
vendored
2
vendor/github.com/go-xorm/xorm/statement.go
generated
vendored
|
@ -73,6 +73,7 @@ type Statement struct {
|
||||||
decrColumns map[string]decrParam
|
decrColumns map[string]decrParam
|
||||||
exprColumns map[string]exprParam
|
exprColumns map[string]exprParam
|
||||||
cond builder.Cond
|
cond builder.Cond
|
||||||
|
bufferSize int
|
||||||
}
|
}
|
||||||
|
|
||||||
// Init reset all the statement's fields
|
// Init reset all the statement's fields
|
||||||
|
@ -111,6 +112,7 @@ func (statement *Statement) Init() {
|
||||||
statement.decrColumns = make(map[string]decrParam)
|
statement.decrColumns = make(map[string]decrParam)
|
||||||
statement.exprColumns = make(map[string]exprParam)
|
statement.exprColumns = make(map[string]exprParam)
|
||||||
statement.cond = builder.NewCond()
|
statement.cond = builder.NewCond()
|
||||||
|
statement.bufferSize = 0
|
||||||
}
|
}
|
||||||
|
|
||||||
// NoAutoCondition if you do not want convert bean's field as query condition, then use this function
|
// NoAutoCondition if you do not want convert bean's field as query condition, then use this function
|
||||||
|
|
6
vendor/vendor.json
vendored
6
vendor/vendor.json
vendored
|
@ -474,10 +474,10 @@
|
||||||
"revisionTime": "2016-08-11T02:11:45Z"
|
"revisionTime": "2016-08-11T02:11:45Z"
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
"checksumSHA1": "lAzHeuH461JyawhsGLi27JpWsgs=",
|
"checksumSHA1": "+KmPfckyKvrUZPIHBYHylg/7V8o=",
|
||||||
"path": "github.com/go-xorm/xorm",
|
"path": "github.com/go-xorm/xorm",
|
||||||
"revision": "3101e3bc440f16f151687d97bce94da063c486f5",
|
"revision": "29d4a0330a00b9be468b70e3fb0f74109348c358",
|
||||||
"revisionTime": "2017-09-15T01:51:15Z"
|
"revisionTime": "2017-09-30T01:26:13Z"
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
"checksumSHA1": "1ft/4j5MFa7C9dPI9whL03HSUzk=",
|
"checksumSHA1": "1ft/4j5MFa7C9dPI9whL03HSUzk=",
|
||||||
|
|
Loading…
Reference in a new issue