// Copyright (C) The Arvados Authors. All rights reserved.
//
// SPDX-License-Identifier: AGPL-3.0

package localdb

import (
	"context"
	"database/sql"
	"fmt"
	"time"

	"git.arvados.org/arvados.git/lib/ctrlctx"
	"git.arvados.org/arvados.git/sdk/go/arvados"
	"git.arvados.org/arvados.git/sdk/go/ctxlog"
	"github.com/sirupsen/logrus"
)

// ContainerUpdate defers to railsProxy and then notifies the
// container priority updater thread.
func (conn *Conn) ContainerUpdate(ctx context.Context, opts arvados.UpdateOptions) (arvados.Container, error) {
	resp, err := conn.railsProxy.ContainerUpdate(ctx, opts)
	if err == nil {
		select {
		case conn.wantContainerPriorityUpdate <- struct{}{}:
		default:
			// update already pending
		}
	}
	return resp, err
}

var containerPriorityUpdateInterval = 5 * time.Minute

// runContainerPriorityUpdateThread periodically (and immediately
// after each container update request) corrects any inconsistent
// container priorities caused by races.
func (conn *Conn) runContainerPriorityUpdateThread(ctx context.Context) {
	ctx = ctrlctx.NewWithToken(ctx, conn.cluster, conn.cluster.SystemRootToken)
	log := ctxlog.FromContext(ctx).WithField("worker", "runContainerPriorityUpdateThread")
	ticker := time.NewTicker(containerPriorityUpdateInterval)
	for ctx.Err() == nil {
		select {
		case <-ticker.C:
		case <-conn.wantContainerPriorityUpdate:
		case <-ctx.Done():
			return
		}
		err := conn.containerPriorityUpdate(ctx, log)
		if err != nil {
			log.WithError(err).Warn("error updating container priorities")
		}
	}
}

func (conn *Conn) containerPriorityUpdate(ctx context.Context, log logrus.FieldLogger) error {
	db, err := conn.getdb(ctx)
	if err != nil {
		return fmt.Errorf("getdb: %w", err)
	}
	// Stage 1: Fix containers that have priority>0 but should
	// have priority=0 because there are no active
	// container_requests (unfinished, priority>0) associated with
	// them.
	res, err := db.ExecContext(ctx, `
		UPDATE containers
		SET priority=0
		WHERE state IN ('Queued', 'Locked', 'Running')
		 AND priority>0
		 AND uuid NOT IN (
			SELECT container_uuid
			FROM container_requests
			WHERE priority > 0
			 AND state = 'Committed')`)
	if err != nil {
		return fmt.Errorf("update: %w", err)
	} else if rows, err := res.RowsAffected(); err != nil {
		return fmt.Errorf("update: %w", err)
	} else if rows > 0 {
		log.Infof("found %d containers with priority>0 and no active requests, updated to priority=0", rows)
	}

	// Stage 2: Fix containers that have priority=0 but should
	// have priority>0 because there are active container_requests
	// (priority>0, unfinished, and not children of cancelled
	// containers).
	//
	// Fixing here means calling out to RailsAPI to compute the
	// correct priority for the contianer and (if needed)
	// propagate that change to child containers.

	// In this loop we look for a single container that needs
	// fixing, call out to Rails to fix it, and repeat until we
	// don't find any more.
	//
	// We could get a batch of UUIDs that need attention by
	// increasing LIMIT 1, however, updating priority on one
	// container typically cascades to other containers, so we
	// would often end up repeating work.
	for lastUUID := ""; ; {
		var uuid string
		err := db.QueryRowxContext(ctx, `
			SELECT containers.uuid from containers
			JOIN container_requests
			 ON container_requests.container_uuid = containers.uuid
			 AND container_requests.state = 'Committed' AND container_requests.priority > 0
			LEFT JOIN containers parent
			 ON parent.uuid = container_requests.requesting_container_uuid
			WHERE containers.state IN ('Queued', 'Locked', 'Running')
			 AND containers.priority = 0
			 AND (parent.uuid IS NULL OR parent.priority > 0)
			ORDER BY containers.created_at
			LIMIT 1`).Scan(&uuid)
		if err == sql.ErrNoRows {
			break
		}
		if err != nil {
			return fmt.Errorf("join: %w", err)
		}
		if uuid == lastUUID {
			// We don't want to keep hammering this
			// forever if the ContainerPriorityUpdate call
			// didn't achieve anything.
			return fmt.Errorf("possible lack of progress: container %s still has priority=0 after updating", uuid)
		}
		lastUUID = uuid
		upd, err := conn.railsProxy.ContainerPriorityUpdate(ctx, arvados.UpdateOptions{UUID: uuid, Select: []string{"uuid", "priority"}})
		if err != nil {
			return err
		}
		log.Debugf("updated container %s priority from 0 to %d", uuid, upd.Priority)
	}
	return nil
}