mirror of
https://github.com/stashapp/stash.git
synced 2025-12-15 21:03:22 +01:00
cb6dab3c5f
* Fix: config race conditions with RWMutex Added RWMutex to config.Instance which read or write locks all instances where viper is used. Refactored checksum manager to only use config and not viper directly anymore. All stash viper operations are now "behind" the config.Instance and thus mutex "protected".
73 lines
2.2 KiB
Go
73 lines
2.2 KiB
Go
package manager
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
"github.com/stashapp/stash/pkg/logger"
|
|
"github.com/stashapp/stash/pkg/manager/config"
|
|
"github.com/stashapp/stash/pkg/models"
|
|
)
|
|
|
|
func setInitialMD5Config(txnManager models.TransactionManager) {
|
|
// if there are no scene files in the database, then default the
|
|
// VideoFileNamingAlgorithm config setting to oshash and calculateMD5 to
|
|
// false, otherwise set them to true for backwards compatibility purposes
|
|
var count int
|
|
if err := txnManager.WithReadTxn(context.TODO(), func(r models.ReaderRepository) error {
|
|
var err error
|
|
count, err = r.Scene().Count()
|
|
return err
|
|
}); err != nil {
|
|
logger.Errorf("Error while counting scenes: %s", err.Error())
|
|
return
|
|
}
|
|
|
|
usingMD5 := count != 0
|
|
defaultAlgorithm := models.HashAlgorithmOshash
|
|
if usingMD5 {
|
|
defaultAlgorithm = models.HashAlgorithmMd5
|
|
}
|
|
|
|
config := config.GetInstance()
|
|
config.SetChecksumDefaultValues(defaultAlgorithm, usingMD5)
|
|
if err := config.Write(); err != nil {
|
|
logger.Errorf("Error while writing configuration file: %s", err.Error())
|
|
}
|
|
}
|
|
|
|
// ValidateVideoFileNamingAlgorithm validates changing the
|
|
// VideoFileNamingAlgorithm configuration flag.
|
|
//
|
|
// If setting VideoFileNamingAlgorithm to MD5, then this function will ensure
|
|
// that all checksum values are set on all scenes.
|
|
//
|
|
// Likewise, if VideoFileNamingAlgorithm is set to oshash, then this function
|
|
// will ensure that all oshash values are set on all scenes.
|
|
func ValidateVideoFileNamingAlgorithm(txnManager models.TransactionManager, newValue models.HashAlgorithm) error {
|
|
// if algorithm is being set to MD5, then all checksums must be present
|
|
return txnManager.WithReadTxn(context.TODO(), func(r models.ReaderRepository) error {
|
|
qb := r.Scene()
|
|
if newValue == models.HashAlgorithmMd5 {
|
|
missingMD5, err := qb.CountMissingChecksum()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if missingMD5 > 0 {
|
|
return errors.New("some checksums are missing on scenes. Run Scan with calculateMD5 set to true")
|
|
}
|
|
} else if newValue == models.HashAlgorithmOshash {
|
|
missingOSHash, err := qb.CountMissingOSHash()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if missingOSHash > 0 {
|
|
return errors.New("some oshash values are missing on scenes. Run Scan to populate")
|
|
}
|
|
}
|
|
|
|
return nil
|
|
})
|
|
}
|