batch.go

/*
* Copyright 2022-2025 Thorsten A. Knieling
*
* 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
*
 */

package clu

import (
	"os"
	"sync"
	"time"

	"github.com/tknie/errorrepo"
	"github.com/tknie/flynn"
	"github.com/tknie/flynn/common"
	"github.com/tknie/log"
	"github.com/tknie/services"
)

// BatchEntry entry of batch repository
type BatchEntry struct {
	Name       string
	Query      string `flynn:":BLOB"`
	Database   string
	ParamCount int
}

var batchDbRef *common.Reference
var batchDbPassword = ""
var batchtablename = ""

var batchLock sync.Mutex
var batchStoreOnline = false

func openBatchRepository() (common.RegDbID, error) {
	var err error
	if userDbPassword == "" {
		userDbPassword = os.Getenv("REST_BATCH_PASS")
	}
	sessionStoreID, err := flynn.Handler(batchDbRef, batchDbPassword)
	if err != nil {
		services.ServerMessage("Register error log: %v", err)
		return 0, err
	}
	return sessionStoreID, nil
}

// InitBatchWatcherThread initialize batch watcher
func InitBatchWatcherThread() {
	// dm := Viewer.Database.BatchRepository
	// if dm != nil {
	// 	r, err := Handles(dm)
	// 	if err == nil {
	// 		clu.InitBatchRepository(r, os.ExpandEnv(dm.Password), os.ExpandEnv(dm.Table))
	// 	} else {
	// 		log.Fatal("batch repository store not being able to start:", err)
	// 	}
	// }
	dm := Viewer.Database.BatchRepository
	if dm != nil {
		for {
			r, err := dm.Handles()
			if err == nil {
				if InitBatchRepository(r, os.ExpandEnv(dm.Password), os.ExpandEnv(dm.Table)) {
					break
				}
				time.Sleep(5 * time.Minute)
			} else {
				services.ServerMessage("Batch repository store not being able to start: %v", err)
			}
		}
	}
}

// InitBatchRepository init batch repository
func InitBatchRepository(dbRef *common.Reference, dbPassword, tablename string) bool {
	batchDbRef = dbRef
	batchDbPassword = dbPassword
	batchStoreID, err := openBatchRepository()
	if err != nil {
		services.ServerMessage("Register error log: %v", err)
		return false
	}
	log.Log.Debugf("Receive batch store handler %s", batchStoreID)
	defer batchStoreID.FreeHandler()
	defer batchStoreID.Close()

	dbTables := flynn.Maps()
	for _, d := range dbTables {
		if d == tablename {
			batchtablename = tablename
			batchStoreOnline = true
			log.Log.Debugf("batch store online = %v", batchStoreOnline)
			services.ServerMessage("Using batch repository on table '%s'", batchtablename)
			return true
		}
	}
	su := &BatchEntry{}
	err = batchStoreID.CreateTable(tablename, su)
	if err != nil {
		services.ServerMessage("Database batch store creating failed: %v", err)
		return false
	}
	batchtablename = tablename
	batchStoreOnline = true
	services.ServerMessage("Database batch store '%s' created successfully", batchtablename)
	return true
}

// BatchSelect search for batchname in an batch repository
func BatchSelect(batchname string) (*BatchEntry, error) {
	log.Log.Debugf("batch select online = %v", batchStoreOnline)
	if !batchStoreOnline {
		log.Log.Debugf("batch rep disabled online = %v", batchStoreOnline)
		return nil, errorrepo.NewError("REST00003")
	}
	batchLock.Lock()
	defer batchLock.Unlock()
	batchStoreID, err := openBatchRepository()
	if err != nil {
		services.ServerMessage("Register error log: %v", err)
		return nil, err
	}
	log.Log.Debugf("Receive batch store handler %s", batchStoreID)
	defer batchStoreID.FreeHandler()
	defer batchStoreID.Close()
	var b *BatchEntry
	q := &common.Query{TableName: batchtablename,
		Search:     "name='" + batchname + "'",
		DataStruct: &BatchEntry{},
		Fields:     []string{"*"}}
	_, err = batchStoreID.Query(q, func(search *common.Query, result *common.Result) error {
		if b == nil {
			b = result.Data.(*BatchEntry)
		}
		return nil
	})
	if err != nil {
		log.Log.Debugf("Failure batch select online = %v", batchStoreOnline)
		log.Log.Errorf("Query batch store failure: %v", err)
		return nil, err
	}
	if b == nil {
		log.Log.Debugf("Fail batch select online = %v", batchStoreOnline)
		return nil, errorrepo.NewError("REST00004")
	}
	log.Log.Debugf("Return batch select online = %v", batchStoreOnline)
	return b, nil
}