github.com/go-kivik/kivik/v4@v4.3.2/couchdb/scheduler.go

// Licensed 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.

package couchdb

import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	"net/http"
	"time"

	kivik "github.com/go-kivik/kivik/v4"
	"github.com/go-kivik/kivik/v4/couchdb/chttp"
	"github.com/go-kivik/kivik/v4/driver"
)

type schedulerDoc struct {
	Database      string    `json:"database"`
	DocID         string    `json:"doc_id"`
	ReplicationID string    `json:"id"`
	Source        string    `json:"source"`
	Target        string    `json:"target"`
	StartTime     time.Time `json:"start_time"`
	LastUpdated   time.Time `json:"last_updated"`
	State         string    `json:"state"`
	Info          repInfo   `json:"info"`
}

type repInfo struct {
	Error            error
	DocsRead         int64 `json:"docs_read"`
	DocsWritten      int64 `json:"docs_written"`
	DocWriteFailures int64 `json:"doc_write_failures"`
	Pending          int64 `json:"changes_pending"`
}

func (i *repInfo) UnmarshalJSON(data []byte) error {
	switch {
	case string(data) == "null":
		return nil
	case bytes.HasPrefix(data, []byte(`{"error":`)):
		var e struct {
			Error *replicationError `json:"error"`
		}
		if err := json.Unmarshal(data, &e); err != nil {
			return err
		}
		i.Error = e.Error
	case data[0] == '{':
		type repInfoClone repInfo
		var x repInfoClone
		if err := json.Unmarshal(data, &x); err != nil {
			return err
		}
		*i = repInfo(x)
	default:
		var e replicationError
		if err := json.Unmarshal(data, &e); err != nil {
			return err
		}
		i.Error = &e
	}
	return nil
}

type schedulerReplication struct {
	docID         string
	database      string
	replicationID string
	source        string
	target        string
	startTime     time.Time
	lastUpdated   time.Time
	state         string
	info          repInfo

	*db
}

var _ driver.Replication = &schedulerReplication{}

func (c *client) schedulerSupported(ctx context.Context) (bool, error) {
	c.sdMU.Lock()
	defer c.sdMU.Unlock()
	if c.schedulerDetected != nil {
		return *c.schedulerDetected, nil
	}
	resp, err := c.DoReq(ctx, http.MethodHead, "_scheduler/jobs", nil)
	if err != nil {
		return false, err
	}
	var supported bool
	switch resp.StatusCode {
	case http.StatusBadRequest:
		// 1.6.x, 1.7.x
		supported = false
	case http.StatusNotFound:
		// 2.0.x
		supported = false
	case http.StatusOK, http.StatusUnauthorized:
		// 2.1.x +
		supported = true
	default:
		// Assume not supported
		supported = false
	}
	c.schedulerDetected = &supported
	return supported, nil
}

func (c *client) newSchedulerReplication(doc *schedulerDoc) *schedulerReplication {
	rep := &schedulerReplication{
		db: &db{
			client: c,
			dbName: doc.Database,
		},
	}
	rep.setFromDoc(doc)
	return rep
}

func (r *schedulerReplication) setFromDoc(doc *schedulerDoc) {
	if r.source == "" {
		r.docID = doc.DocID
		r.database = doc.Database
		r.replicationID = doc.ReplicationID
		r.source = doc.Source
		r.target = doc.Target
		r.startTime = doc.StartTime
	}
	r.lastUpdated = doc.LastUpdated
	r.state = doc.State
	r.info = doc.Info
}

func (c *client) fetchSchedulerReplication(ctx context.Context, docID string) (*schedulerReplication, error) {
	rep := &schedulerReplication{
		docID:    docID,
		database: "_replicator",
		db: &db{
			client: c,
			dbName: "_replicator",
		},
	}
	for rep.source == "" {
		if err := rep.update(ctx); err != nil {
			return rep, err
		}
		time.Sleep(100 * time.Millisecond) // nolint:gomnd
	}
	return rep, nil
}

func (r *schedulerReplication) StartTime() time.Time { return r.startTime }
func (r *schedulerReplication) EndTime() time.Time {
	if r.state == "failed" || r.state == "completed" {
		return r.lastUpdated
	}
	return time.Time{}
}
func (r *schedulerReplication) Err() error            { return r.info.Error }
func (r *schedulerReplication) ReplicationID() string { return r.replicationID }
func (r *schedulerReplication) Source() string        { return r.source }
func (r *schedulerReplication) Target() string        { return r.target }
func (r *schedulerReplication) State() string         { return r.state }

func (r *schedulerReplication) Update(ctx context.Context, rep *driver.ReplicationInfo) error {
	if err := r.update(ctx); err != nil {
		return err
	}
	rep.DocWriteFailures = r.info.DocWriteFailures
	rep.DocsRead = r.info.DocsRead
	rep.DocsWritten = r.info.DocsWritten
	return nil
}

func (r *schedulerReplication) Delete(ctx context.Context) error {
	rev, err := r.GetRev(ctx, r.docID, kivik.Params(nil))
	if err != nil {
		return err
	}
	_, err = r.db.Delete(ctx, r.docID, kivik.Rev(rev))
	return err
}

// isBug1000 detects a race condition bug in CouchDB 2.1.x so the attempt can
// be retried. See https://github.com/apache/couchdb/issues/1000
func isBug1000(err error) bool {
	if err == nil {
		return false
	}
	cerr, ok := err.(*chttp.HTTPError)
	if !ok {
		// should never happen
		return false
	}
	if cerr.Response.StatusCode != http.StatusInternalServerError {
		return false
	}
	return cerr.Reason == "function_clause"
}

func (r *schedulerReplication) update(ctx context.Context) error {
	path := fmt.Sprintf("/_scheduler/docs/%s/%s", r.database, chttp.EncodeDocID(r.docID))
	var doc schedulerDoc
	if err := r.db.Client.DoJSON(ctx, http.MethodGet, path, nil, &doc); err != nil {
		if isBug1000(err) {
			return r.update(ctx)
		}
		return err
	}
	r.setFromDoc(&doc)
	return nil
}

func (c *client) getReplicationsFromScheduler(ctx context.Context, opts map[string]interface{}) ([]driver.Replication, error) {
	params, err := optionsToParams(opts)
	if err != nil {
		return nil, err
	}
	var result struct {
		Docs []schedulerDoc `json:"docs"`
	}
	path := "/_scheduler/docs"
	if params != nil {
		path = path + "?" + params.Encode()
	}
	if err = c.DoJSON(ctx, http.MethodGet, path, nil, &result); err != nil {
		return nil, err
	}
	reps := make([]driver.Replication, 0, len(result.Docs))
	for _, row := range result.Docs {
		rep := c.newSchedulerReplication(&row)
		reps = append(reps, rep)
	}
	return reps, nil
}