in traffic_ops/traffic_ops_golang/invalidationjobs/invalidationjobs.go [1042:1215]
func Update(w http.ResponseWriter, r *http.Request) {
inf, userErr, sysErr, errCode := api.NewInfo(r, nil, nil)
if userErr != nil || sysErr != nil {
api.HandleErr(w, r, inf.Tx.Tx, errCode, userErr, sysErr)
return
}
defer inf.Close()
var oFQDN string
var dsid uint
var uid uint
job := tc.InvalidationJob{}
row := inf.Tx.Tx.QueryRow(putInfoQuery, inf.Params["id"])
err := row.Scan(&job.ID,
&job.CreatedBy,
&uid,
&dsid,
&job.DeliveryService,
&job.AssetURL,
&job.Parameters,
&job.StartTime,
&oFQDN)
if err != nil {
if err == sql.ErrNoRows {
userErr = fmt.Errorf("No job by id '%s'!", inf.Params["id"])
errCode = http.StatusNotFound
} else {
sysErr = fmt.Errorf("fetching job update info: %v", err)
errCode = http.StatusInternalServerError
}
api.HandleErr(w, r, inf.Tx.Tx, errCode, userErr, sysErr)
return
}
if ok, err := IsUserAuthorizedToModifyDSID(inf, dsid); err != nil {
sysErr = fmt.Errorf("Checking user permissions on DS #%d: %v", dsid, err)
errCode = http.StatusInternalServerError
api.HandleErr(w, r, inf.Tx.Tx, errCode, nil, sysErr)
return
} else if !ok {
userErr = errors.New("No such Delivery Service!")
errCode = http.StatusNotFound
api.HandleErr(w, r, inf.Tx.Tx, errCode, userErr, nil)
return
}
if ok, err := IsUserAuthorizedToModifyJobsMadeByUsername(inf, *job.CreatedBy); err != nil {
sysErr = fmt.Errorf("Checking user permissions against user %s: %v", *job.CreatedBy, err)
errCode = http.StatusInternalServerError
api.HandleErr(w, r, inf.Tx.Tx, errCode, nil, sysErr)
return
} else if !ok {
userErr = fmt.Errorf("No job by id '%s'!", inf.Params["id"])
errCode = http.StatusNotFound
api.HandleErr(w, r, inf.Tx.Tx, errCode, userErr, nil)
return
}
input := tc.InvalidationJob{}
if err := json.NewDecoder(r.Body).Decode(&input); err != nil {
userErr = fmt.Errorf("Unable to parse input: %v", err)
sysErr = fmt.Errorf("parsing input to PUT jobs?id=%s: %v", inf.Params["id"], err)
errCode = http.StatusBadRequest
api.HandleErr(w, r, inf.Tx.Tx, errCode, userErr, sysErr)
return
}
if err := input.Validate(); err != nil {
api.HandleErr(w, r, inf.Tx.Tx, http.StatusBadRequest, err, nil)
return
}
if !strings.HasPrefix(*input.AssetURL, oFQDN) {
userErr = fmt.Errorf("Cannot set asset URL that does not start with Delivery Service origin URL: %s", oFQDN)
errCode = http.StatusBadRequest
api.HandleErr(w, r, inf.Tx.Tx, errCode, userErr, nil)
return
}
if job.StartTime.Before(time.Now()) {
userErr = errors.New("Cannot modify a job that has already started!")
errCode = http.StatusMethodNotAllowed
w.Header().Set(http.CanonicalHeaderKey("allow"), "GET,HEAD,DELETE")
api.HandleErr(w, r, inf.Tx.Tx, errCode, userErr, nil)
return
}
if *job.DeliveryService != *input.DeliveryService {
userErr = errors.New("Cannot change 'deliveryService' of existing invalidation job!")
errCode = http.StatusConflict
api.HandleErr(w, r, inf.Tx.Tx, errCode, userErr, nil)
return
}
if *job.CreatedBy != *input.CreatedBy {
userErr = errors.New("Cannot change 'createdBy' of existing invalidation jobs!")
errCode = http.StatusConflict
api.HandleErr(w, r, inf.Tx.Tx, errCode, userErr, nil)
return
}
if *job.ID != *input.ID {
userErr = errors.New("Cannot change an invalidation job 'id'!")
errCode = http.StatusConflict
api.HandleErr(w, r, inf.Tx.Tx, errCode, userErr, nil)
return
}
_, cdnName, _, err := dbhelpers.GetDSNameAndCDNFromID(inf.Tx.Tx, int(dsid))
if err != nil {
api.HandleErr(w, r, inf.Tx.Tx, http.StatusInternalServerError, nil, errors.New("getting delivery service and CDN name from ID: "+err.Error()))
return
}
userErr, sysErr, statusCode := dbhelpers.CheckIfCurrentUserCanModifyCDN(inf.Tx.Tx, string(cdnName), inf.User.UserName)
if userErr != nil || sysErr != nil {
api.HandleErr(w, r, inf.Tx.Tx, statusCode, userErr, sysErr)
return
}
row = inf.Tx.Tx.QueryRow(updateQuery,
input.AssetURL,
strings.TrimSuffix(strings.TrimPrefix(*input.Parameters, "TTL:"), "h"), // Strip TTL: and h from 'TTL:##h'
input.StartTime.Time,
*job.ID)
err = row.Scan(&job.AssetURL,
&job.CreatedBy,
&job.DeliveryService,
&job.ID,
&job.Keyword,
&job.Parameters,
&job.StartTime)
if err != nil {
sysErr = fmt.Errorf("Updating a job: %v", err)
errCode = http.StatusInternalServerError
api.HandleErr(w, r, inf.Tx.Tx, errCode, nil, sysErr)
return
}
if err = setRevalFlags(*job.DeliveryService, inf.Tx.Tx); err != nil {
api.HandleErr(w, r, inf.Tx.Tx, http.StatusInternalServerError, nil, fmt.Errorf("Setting reval flags: %v", err))
return
}
ttlHours := input.TTLHours()
conflicts := tc.ValidateJobUniqueness(inf.Tx.Tx, dsid, input.StartTime.Time, *input.AssetURL, ttlHours)
response := apiResponse{
make([]tc.Alert, len(conflicts)+1),
job,
}
for i, conflict := range conflicts {
response.Alerts[i] = tc.Alert{
Text: conflict,
Level: tc.WarnLevel.String(),
}
}
response.Alerts[len(conflicts)] = tc.Alert{
Text: fmt.Sprintf("Invalidation request created for %v, start:%v end %v", *job.AssetURL, job.StartTime.Time,
job.StartTime.Add(time.Hour*time.Duration(ttlHours))),
Level: tc.SuccessLevel.String(),
}
resp, err := json.Marshal(response)
if err != nil {
sysErr = fmt.Errorf("encoding response: %v", err)
errCode = http.StatusInternalServerError
api.HandleErr(w, r, inf.Tx.Tx, errCode, nil, sysErr)
return
}
w.Header().Set(http.CanonicalHeaderKey("content-type"), rfc.ApplicationJSON)
api.WriteAndLogErr(w, r, append(resp, '\n'))
api.CreateChangeLogRawTx(api.ApiChange, api.Updated+" content invalidation job - ID: "+strconv.FormatUint(*job.ID, 10)+" DS: "+*job.DeliveryService+" URL: '"+*job.AssetURL+"' Params: '"+*job.Parameters+"'", inf.User, inf.Tx.Tx)
}