model_capabilities.go

package worker

import (
	"fmt"
	"time"

	"github.com/ovh/cds/engine/api/cache"
	"github.com/ovh/cds/engine/api/database"
	"github.com/ovh/cds/engine/log"
	"github.com/ovh/cds/sdk"
)

//ModelCapabilititiesCacheLoader set all model Capabilities in the cache
func ModelCapabilititiesCacheLoader(delay time.Duration) {
	for {
		time.Sleep(delay * time.Second)
		db := database.DB()
		dbmap := database.DBMap(db)
		if db != nil {
			var mayIWork string
			loaderKey := cache.Key("worker", "modelcapabilitites", "loading")
			if cache.Get(loaderKey, &mayIWork) {
				cache.SetWithTTL(loaderKey, "true", 60)
				wms, err := LoadWorkerModels(dbmap)
				if err != nil {
					log.Warning("ModelCapabilititiesCacheLoader> Unable to load worker models: %s", err)
					continue
				}
				for _, wm := range wms {
					modelKey := cache.Key("worker", "modelcapabilitites", fmt.Sprintf("%d", wm.ID))
					cache.Set(modelKey, wm.Capabilities)
				}
				cache.Delete(loaderKey)
			}
		}
	}
}

//GetModelCapabilities load model capabilities from cache
func GetModelCapabilities(db database.Querier, modelID int64) ([]sdk.Requirement, error) {
	modelKey := cache.Key("worker", "modelcapabilitites", fmt.Sprintf("%d", modelID))
	req := []sdk.Requirement{}
	//if we didn't got any data, try to load from DB
	if !cache.Get(modelKey, &req) {
		var err error
		req, err = LoadWorkerModelCapabilities(db, modelID)
		if err != nil {
			return nil, fmt.Errorf("GetModelCapabilities> cannot loadWorkerModelCapabilities: %s\n", err)
		}
		cache.Set(modelKey, req)
	}
	return req, nil
}