dify-plugin-daemon/internal/types/app/default.go

package app

import "golang.org/x/exp/constraints"

func (config *Config) SetDefault() {
	setDefaultInt(&config.ServerPort, 5002)
	setDefaultInt(&config.RoutinePoolSize, 1000)
	setDefaultInt(&config.LifetimeCollectionGCInterval, 60)
	setDefaultInt(&config.LifetimeCollectionHeartbeatInterval, 5)
	setDefaultInt(&config.LifetimeStateGCInterval, 300)
	setDefaultInt(&config.DifyInvocationConnectionIdleTimeout, 120)
	setDefaultInt(&config.PluginRemoteInstallServerEventLoopNums, 8)
	setDefaultInt(&config.PluginRemoteInstallingMaxConn, 256)
	setDefaultInt(&config.MaxPluginPackageSize, 52428800)
	setDefaultInt(&config.MaxAWSLambdaTransactionTimeout, 150)
	setDefaultInt(&config.PluginMaxExecutionTimeout, 240)
	setDefaultString(&config.PluginStorageType, "local")
	setDefaultInt(&config.PluginMediaCacheSize, 1024)
	setDefaultInt(&config.PluginRemoteInstallingMaxSingleTenantConn, 5)
	setDefaultBool(&config.PluginRemoteInstallingEnabled, true)
	setDefaultBool(&config.PluginEndpointEnabled, true)
	setDefaultString(&config.DBSslMode, "disable")
	setDefaultString(&config.PluginStorageLocalRoot, "./storage")
	setDefaultString(&config.PluginInstalledPath, "./plugin")
	setDefaultString(&config.PluginMediaCachePath, "./assets")
	setDefaultString(&config.PersistenceStoragePath, "./persistence")
	setDefaultInt(&config.PersistenceStorageMaxSize, 100*1024*1024)
	setDefaultString(&config.ProcessCachingPath, "./subprocesses")
	setDefaultString(&config.PluginPackageCachePath, "./plugin_packages")
	setDefaultString(&config.PythonInterpreterPath, "/usr/bin/python3")
}

func setDefaultInt[T constraints.Integer](value *T, defaultValue T) {
	if *value == 0 {
		*value = defaultValue
	}
}

func setDefaultString(value *string, defaultValue string) {
	if *value == "" {
		*value = defaultValue
	}
}

func setDefaultBool(value *bool, defaultValue bool) {
	if !*value {
		*value = defaultValue
	}
}