blob: b489e4939096baa4932d10c613336e408cf1e26f [file] [log] [blame]
package cachegroup
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
import (
"database/sql"
"encoding/json"
"errors"
"net/http"
"strconv"
"github.com/apache/trafficcontrol/lib/go-tc"
"github.com/apache/trafficcontrol/traffic_ops/traffic_ops_golang/api"
)
func QueueUpdates(w http.ResponseWriter, r *http.Request) {
inf, userErr, sysErr, errCode := api.NewInfo(r, []string{"id"}, []string{"id"})
if userErr != nil || sysErr != nil {
api.HandleErr(w, r, inf.Tx.Tx, errCode, userErr, sysErr)
return
}
defer inf.Close()
reqObj := tc.CachegroupQueueUpdatesRequest{}
if err := json.NewDecoder(r.Body).Decode(&reqObj); err != nil {
api.HandleErr(w, r, inf.Tx.Tx, http.StatusBadRequest, errors.New("malformed JSON: "+err.Error()), nil)
return
}
if reqObj.Action != "queue" && reqObj.Action != "dequeue" {
api.HandleErr(w, r, inf.Tx.Tx, http.StatusBadRequest, errors.New("action must be 'queue' or 'dequeue'"), nil)
return
}
if reqObj.CDN == nil && reqObj.CDNID == nil {
api.HandleErr(w, r, inf.Tx.Tx, http.StatusBadRequest, errors.New("cdn does not exist"), nil)
return
}
if reqObj.CDN == nil || *reqObj.CDN == "" {
cdn, ok, err := getCDNNameFromID(inf.Tx.Tx, int64(*reqObj.CDNID))
if err != nil {
api.HandleErr(w, r, inf.Tx.Tx, http.StatusInternalServerError, nil, errors.New("getting CDN name from ID '"+strconv.Itoa(int(*reqObj.CDNID))+"': "+err.Error()))
return
}
if !ok {
api.HandleErr(w, r, inf.Tx.Tx, http.StatusBadRequest, errors.New("cdn "+strconv.Itoa(int(*reqObj.CDNID))+" does not exist"), nil)
return
}
reqObj.CDN = &cdn
}
cgID := int64(inf.IntParams["id"])
cgName, ok, err := getCGNameFromID(inf.Tx.Tx, cgID)
if err != nil {
api.HandleErr(w, r, inf.Tx.Tx, http.StatusInternalServerError, nil, errors.New("getting cachegroup name from ID '"+inf.Params["id"]+"': "+err.Error()))
return
}
if !ok {
api.HandleErr(w, r, inf.Tx.Tx, http.StatusBadRequest, errors.New("cachegroup "+inf.Params["id"]+" does not exist"), nil)
return
}
queue := reqObj.Action == "queue"
updatedCaches, err := queueUpdates(inf.Tx.Tx, cgID, *reqObj.CDN, queue)
if err != nil {
api.HandleErr(w, r, inf.Tx.Tx, http.StatusInternalServerError, nil, errors.New("queueing updates: "+err.Error()))
return
}
api.WriteResp(w, r, QueueUpdatesResp{
CacheGroupName: cgName,
Action: reqObj.Action,
ServerNames: updatedCaches,
CDN: *reqObj.CDN,
CacheGroupID: cgID,
})
api.CreateChangeLogRawTx(api.ApiChange, "Server updates "+reqObj.Action+"d for "+string(cgName), inf.User, inf.Tx.Tx)
}
type QueueUpdatesResp struct {
CacheGroupName tc.CacheGroupName `json:"cachegroupName"`
Action string `json:"action"`
ServerNames []tc.CacheName `json:"serverNames"`
CDN tc.CDNName `json:"cdn"`
CacheGroupID int64 `json:"cachegroupID"`
}
func getCDNNameFromID(tx *sql.Tx, id int64) (tc.CDNName, bool, error) {
name := ""
if err := tx.QueryRow(`SELECT name FROM cdn WHERE id = $1`, id).Scan(&name); err != nil {
if err == sql.ErrNoRows {
return "", false, nil
}
return "", false, errors.New("querying CDN ID: " + err.Error())
}
return tc.CDNName(name), true, nil
}
func getCGNameFromID(tx *sql.Tx, id int64) (tc.CacheGroupName, bool, error) {
name := ""
if err := tx.QueryRow(`SELECT name FROM cachegroup WHERE id = $1`, id).Scan(&name); err != nil {
if err == sql.ErrNoRows {
return "", false, nil
}
return "", false, errors.New("querying cachegroup ID: " + err.Error())
}
return tc.CacheGroupName(name), true, nil
}
func queueUpdates(tx *sql.Tx, cgID int64, cdn tc.CDNName, queue bool) ([]tc.CacheName, error) {
q := `
UPDATE server SET upd_pending = $1
WHERE server.cachegroup = $2
AND server.cdn_id = (select id from cdn where name = $3)
RETURNING server.host_name
`
rows, err := tx.Query(q, queue, cgID, cdn)
if err != nil {
return nil, errors.New("querying queue updates: " + err.Error())
}
defer rows.Close()
names := []tc.CacheName{}
for rows.Next() {
name := ""
if err := rows.Scan(&name); err != nil {
return nil, errors.New("scanning queue updates: " + err.Error())
}
names = append(names, tc.CacheName(name))
}
return names, nil
}