sim1222-mirror/gitea
1
0
Fork 0
mirror of https://github.com/go-gitea/gitea.git synced 2025-08-18 06:34:07 +09:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

feat: receive task logs

Browse Source
This commit is contained in:
Jason Song
2022-10-12 17:58:36 +08:00
parent 7497155d14
commit 848a1ae309
2 changed files with 67 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
models/bots/task_log.go
View File

@ -8,14 +8,15 @@ import (
"fmt"
"code.gitea.io/gitea/models/db"
"code.gitea.io/gitea/modules/log"
"code.gitea.io/gitea/modules/timeutil"
)
// TaskLog represents a task's log, every task has a standalone table
type TaskLog struct {
ID int64
Content string `xorm:"BINARY"`
Created timeutil.TimeStamp `xorm:"created"`
ID int64
Timestamp timeutil.TimeStamp
Content string `xorm:"LONGTEXT"`
}
func init() {
@ -26,16 +27,16 @@ func GetTaskLogTableName(taskID int64) string {
return fmt.Sprintf("bots_task_log_%d", taskID)
}
// CreateTaskLog table for a build
func CreateTaskLog(buildID int64) error {
// CreateTaskLog table for a task
func CreateTaskLog(taskID int64) error {
return db.GetEngine(db.DefaultContext).
Table(GetBuildLogTableName(buildID)).
Sync2(new(BuildLog))
Table(GetTaskLogTableName(taskID)).
Sync(new(TaskLog))
}
func GetTaskLogs(taskID, index, length int64) (logs []*TaskLog, err error) {
sess := db.GetEngine(db.DefaultContext).Table(GetBuildLogTableName(taskID)).
Where("id>=?", index)
Where("id>=?", index).OrderBy("id")
if length > 0 {
sess.Limit(int(length))
@ -45,3 +46,36 @@ func GetTaskLogs(taskID, index, length int64) (logs []*TaskLog, err error) {
return
}
func InsertTaskLogs(taskID int64, logs []*TaskLog) (int64, error) {
if err := CreateTaskLog(taskID); err != nil {
return 0, err
}
table := GetTaskLogTableName(taskID)
// TODO: A more complete way to insert logs
// Be careful:
// - the id of a log can be 0
// - some logs may already exist in db
// - if use exec, consider different databases
// - the input should be ordered by id
// - the ids should be continuously increasing
// - the min id of input should be 1 + (the max id in db)
if len(logs) == 0 {
return 0, fmt.Errorf("no logs")
}
ack := logs[0].ID - 1
sess := db.GetEngine(db.DefaultContext)
for _, v := range logs {
_, err := sess.Exec(fmt.Sprintf("INSERT IGNORE INTO %s (id, timestamp, content) VALUES (?,?,?)", table), v.ID, v.Timestamp, []byte(v.Content))
if err != nil {
log.Error("insert log %d of task %d: %v", v.ID, taskID, err)
break
}
ack = v.ID
}
return ack, nil
}