(taskData adaptix.TaskData)
| 16 | } |
| 17 | |
| 18 | func (dbms *DBMS) DbTaskInsert(taskData adaptix.TaskData) error { |
| 19 | ok := dbms.DatabaseExists() |
| 20 | if !ok { |
| 21 | return errors.New("database does not exist") |
| 22 | } |
| 23 | |
| 24 | var ( |
| 25 | result sql.Result |
| 26 | err error |
| 27 | ) |
| 28 | if dbms.stmtTaskInsert != nil { |
| 29 | result, err = dbms.stmtTaskInsert.Exec(taskData.TaskId, taskData.AgentId, taskData.Type, taskData.Client, taskData.User, taskData.Computer, taskData.StartDate, taskData.FinishDate, taskData.CommandLine, taskData.MessageType, taskData.Message, taskData.ClearText, taskData.Completed) |
| 30 | } else { |
| 31 | result, err = dbms.database.Exec(`INSERT OR IGNORE INTO Tasks (TaskId, AgentId, TaskType, Client, User, Computer, StartDate, FinishDate, CommandLine, MessageType, Message, ClearText, Completed) values(?,?,?,?,?,?,?,?,?,?,?,?,?);`, taskData.TaskId, taskData.AgentId, taskData.Type, taskData.Client, taskData.User, taskData.Computer, taskData.StartDate, taskData.FinishDate, taskData.CommandLine, taskData.MessageType, taskData.Message, taskData.ClearText, taskData.Completed) |
| 32 | } |
| 33 | if err != nil { |
| 34 | return err |
| 35 | } |
| 36 | rows, _ := result.RowsAffected() |
| 37 | if rows == 0 { |
| 38 | return fmt.Errorf("task %s already exists", taskData.TaskId) |
| 39 | } |
| 40 | return nil |
| 41 | } |
| 42 | |
| 43 | func (dbms *DBMS) DbTaskUpdate(taskData adaptix.TaskData) error { |
| 44 | ok := dbms.DatabaseExists() |
no test coverage detected