// ScheduleList returns a list of all saved schedules func (d *SQLite) ScheduleList() ([]*spec.Schedule, error) { rows, err := d.Connection.Query("SELECT * FROM "+schedulesTable, nil) if err != nil { return nil, err } defer rows.Close() var ss []*spec.Schedule for rows.Next() { var id, backup int var body []byte err = rows.Scan(&id, &backup, &body) if err != nil { return nil, err } var s spec.Schedule json.Unmarshal(body, &s) s.Backup = backup s.ID = id ss = append(ss, &s) } return ss, nil }
func updateSchedule(w http.ResponseWriter, r *http.Request) (*httpapi.APIResponse, error) { body, err := ioutil.ReadAll(io.LimitReader(r.Body, 1048576)) if err != nil { return nil, httpapi.NewError("", "Request too large", 413) } if err := r.Body.Close(); err != nil { return nil, errors.New("") } s := new(spec.Schedule) if err := json.Unmarshal(body, &s); err != nil { return nil, httpapi.NewError(err.Error(), "Invalid Request", 400) } vars := mux.Vars(r) sID, err := strconv.ParseInt(vars["sID"], 10, 64) if err != nil { return nil, httpapi.NewError(err.Error(), "Invalid schedule ID", 400) } s.ID = int(sID) if err := manager.UpdateSchedule(s); err != nil { return nil, httpapi.NewError(err.Error(), "Could not update schedule", 400) } return &httpapi.APIResponse{HTTPCode: 200}, nil }
func UpdateSchedule(s *spec.Schedule) error { _, err := cron.Parse(s.String()) if err != nil { return err } if err = gDb.UpdateSchedule(s); err != nil { return err } schedules.Stop() return initCron() }
// GetSchedule returns single schedule func (d *SQLite) GetSchedule(id int) (*spec.Schedule, error) { var s []byte var backup int err := d.Connection.QueryRow("SELECT * FROM "+schedulesTable+" WHERE id=?", id).Scan(&id, &backup, &s) switch { case err == sql.ErrNoRows: return nil, errors.New("No schedule with that ID") case err != nil: return nil, err default: var sched spec.Schedule json.Unmarshal(s, &sched) sched.ID = id return &sched, nil } }
// AddSchedule saves a schedule to the DB func (d *SQLite) AddSchedule(s *spec.Schedule) error { sql := "INSERT INTO " + schedulesTable + " (backup, schedule) values(?, ?)" sched, err := json.Marshal(s) if err != nil { return errors.New("Could not marshal schedule") } result, err := d.Connection.Exec(sql, s.Backup, sched) if err != nil { return err } id, err := result.LastInsertId() if err != nil { return nil } s.ID = int(id) return nil }