2020-08-04 14:28:25 +08:00
|
|
|
package model
|
|
|
|
|
|
|
|
import sq "github.com/Masterminds/squirrel"
|
|
|
|
|
|
|
|
const installTraceTable = "`install_trace`"
|
|
|
|
|
2020-08-15 13:38:06 +08:00
|
|
|
// InstallTrace -
|
2020-08-04 14:28:25 +08:00
|
|
|
type InstallTrace struct {
|
|
|
|
ID int64 `json:"id"`
|
|
|
|
Token string `json:"token"`
|
|
|
|
ServerID int64 `json:"serverId"`
|
|
|
|
ServerName string `json:"serverName"`
|
|
|
|
Detail string `json:"detail"`
|
|
|
|
State uint8 `json:"state"`
|
|
|
|
OperatorID int64 `json:"operatorId"`
|
|
|
|
OperatorName string `json:"operatorName"`
|
|
|
|
Type int64 `json:"type"`
|
|
|
|
Ext string `json:"ext"`
|
|
|
|
InstallState int64 `json:"installState"`
|
|
|
|
InsertTime string `json:"insertTime"`
|
|
|
|
UpdateTime string `json:"updateTime"`
|
|
|
|
}
|
|
|
|
|
2020-08-15 13:38:06 +08:00
|
|
|
// InstallTraces -
|
2020-08-04 14:28:25 +08:00
|
|
|
type InstallTraces []InstallTrace
|
|
|
|
|
|
|
|
// install trace type
|
|
|
|
const (
|
|
|
|
Rsync = 1
|
|
|
|
SSH = 2
|
|
|
|
Script = 3
|
|
|
|
)
|
|
|
|
|
2020-08-15 13:38:06 +08:00
|
|
|
// AddRow return LastInsertId
|
2020-08-04 14:28:25 +08:00
|
|
|
func (it InstallTrace) AddRow() (int64, error) {
|
|
|
|
result, err := sq.
|
|
|
|
Insert(installTraceTable).
|
|
|
|
Columns("token", "server_id", "server_name", "detail", "state", "operator_id", "operator_name", "type", "ext").
|
|
|
|
Values(it.Token, it.ServerID, it.ServerName, it.Detail, it.State, it.OperatorID, it.OperatorName, it.Type, it.Ext).
|
|
|
|
RunWith(DB).
|
|
|
|
Exec()
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
return 0, err
|
|
|
|
}
|
|
|
|
|
|
|
|
id, err := result.LastInsertId()
|
|
|
|
return id, err
|
|
|
|
}
|
|
|
|
|
2020-08-15 13:38:06 +08:00
|
|
|
// GetListByToken -
|
2020-08-04 14:28:25 +08:00
|
|
|
func (it InstallTrace) GetListByToken() (InstallTraces, error) {
|
|
|
|
rows, err := sq.
|
|
|
|
Select("id, token, server_id, server_name, detail, state, operator_id, operator_name, type, ext, insert_time, update_time").
|
|
|
|
From(installTraceTable).
|
|
|
|
Where(sq.Eq{"token": it.Token}).
|
|
|
|
RunWith(DB).
|
|
|
|
Query()
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
installTraces := InstallTraces{}
|
|
|
|
for rows.Next() {
|
|
|
|
var installTrace InstallTrace
|
|
|
|
|
|
|
|
if err := rows.Scan(
|
|
|
|
&installTrace.ID,
|
|
|
|
&installTrace.Token,
|
|
|
|
&installTrace.ServerID,
|
|
|
|
&installTrace.ServerName,
|
|
|
|
&installTrace.Detail,
|
|
|
|
&installTrace.State,
|
|
|
|
&installTrace.OperatorID,
|
|
|
|
&installTrace.OperatorName,
|
|
|
|
&installTrace.Type,
|
|
|
|
&installTrace.Ext,
|
|
|
|
&installTrace.InsertTime,
|
|
|
|
&installTrace.UpdateTime); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
installTraces = append(installTraces, installTrace)
|
|
|
|
}
|
|
|
|
return installTraces, nil
|
|
|
|
}
|
|
|
|
|
2020-08-15 13:38:06 +08:00
|
|
|
// GetPreviewByProjectID -
|
2020-08-04 14:28:25 +08:00
|
|
|
func (it InstallTrace) GetPreviewByProjectID() (InstallTraces, error) {
|
|
|
|
rows, err := sq.
|
|
|
|
Select("id,token,server_id,server_name,detail,state,operator_id,operator_name,type,ext,create_time,update_time").
|
|
|
|
Column("!EXISTS (SELECT id FROM " + installTraceTable + " AS it where it.state = 0 AND it.token = install_trace.token) as install_state").
|
|
|
|
From(installTraceTable).
|
|
|
|
Where(sq.Eq{"project_id": it.ServerID, "type": Rsync}).
|
|
|
|
Limit(15).
|
|
|
|
RunWith(DB).
|
|
|
|
Query()
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
installTraces := InstallTraces{}
|
|
|
|
for rows.Next() {
|
|
|
|
var installTrace InstallTrace
|
|
|
|
|
|
|
|
if err := rows.Scan(
|
|
|
|
&installTrace.ID,
|
|
|
|
&installTrace.Token,
|
|
|
|
&installTrace.ServerID,
|
|
|
|
&installTrace.ServerName,
|
|
|
|
&installTrace.Detail,
|
|
|
|
&installTrace.State,
|
|
|
|
&installTrace.OperatorID,
|
|
|
|
&installTrace.OperatorName,
|
|
|
|
&installTrace.Type,
|
|
|
|
&installTrace.Ext,
|
|
|
|
&installTrace.InsertTime,
|
|
|
|
&installTrace.UpdateTime,
|
|
|
|
&installTrace.InstallState); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
installTraces = append(installTraces, installTrace)
|
|
|
|
}
|
|
|
|
return installTraces, nil
|
|
|
|
}
|
|
|
|
|
2020-08-15 13:38:06 +08:00
|
|
|
// GetListGroupByToken -
|
2020-08-04 14:28:25 +08:00
|
|
|
func (it InstallTrace) GetListGroupByToken() (InstallTraces, error) {
|
|
|
|
rows, err := sq.
|
|
|
|
Select("token, MIN(state) as install_state").
|
|
|
|
From(installTraceTable).
|
|
|
|
Where(sq.Eq{"server_id": it.ServerID}).
|
|
|
|
GroupBy("token").
|
|
|
|
RunWith(DB).
|
|
|
|
Query()
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
installTraces := InstallTraces{}
|
|
|
|
for rows.Next() {
|
|
|
|
var installTrace InstallTrace
|
|
|
|
|
|
|
|
if err := rows.Scan(
|
|
|
|
&installTrace.Token,
|
|
|
|
&installTrace.InstallState); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
installTraces = append(installTraces, installTrace)
|
|
|
|
}
|
|
|
|
return installTraces, nil
|
|
|
|
}
|