Пример #1
0
// 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
}
Пример #2
0
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
}
Пример #3
0
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()
}
Пример #4
0
// 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
	}
}
Пример #5
0
// 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
}