config.go 3.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  1. package app
  2. import (
  3. "fmt"
  4. "github.com/go-playground/validator/v10"
  5. )
  6. type Config struct {
  7. ServerPort uint16 `envconfig:"SERVER_PORT" validate:"required"`
  8. PluginInnerApiKey string `envconfig:"PLUGIN_INNER_API_KEY" validate:"required"`
  9. PluginInnerApiURL string `envconfig:"PLUGIN_INNER_API_URL" validate:"required"`
  10. PluginRemoteInstallingHost string `envconfig:"PLUGIN_REMOTE_INSTALLING_HOST"`
  11. PluginRemoteInstallingPort uint16 `envconfig:"PLUGIN_REMOTE_INSTALLING_PORT"`
  12. PluginRemoteInstallingEnabled bool `envconfig:"PLUGIN_REMOTE_INSTALLING_ENABLED"`
  13. PluginRemoteInstallingMaxConn int `envconfig:"PLUGIN_REMOTE_INSTALLING_MAX_CONN"`
  14. PluginRemoteInstallServerEventLoopNums int `envconfig:"PLUGIN_REMOTE_INSTALL_SERVER_EVENT_LOOP_NUMS"`
  15. PluginWebhookEnabled bool `envconfig:"PLUGIN_WEBHOOK_ENABLED"`
  16. StoragePath string `envconfig:"STORAGE_PLUGIN_PATH" validate:"required"`
  17. ProcessCachingPath string `envconfig:"PROCESS_CACHING_PATH" validate:"required"`
  18. Platform PlatformType `envconfig:"PLATFORM" validate:"required"`
  19. RoutinePoolSize int `envconfig:"ROUTINE_POOL_SIZE" validate:"required"`
  20. RedisHost string `envconfig:"REDIS_HOST" validate:"required"`
  21. RedisPort uint16 `envconfig:"REDIS_PORT" validate:"required"`
  22. RedisPass string `envconfig:"REDIS_PASS" validate:"required"`
  23. DBUsername string `envconfig:"DB_USERNAME" validate:"required"`
  24. DBPassword string `envconfig:"DB_PASSWORD" validate:"required"`
  25. DBHost string `envconfig:"DB_HOST" validate:"required"`
  26. DBPort uint16 `envconfig:"DB_PORT" validate:"required"`
  27. DBDatabase string `envconfig:"DB_DATABASE" validate:"required"`
  28. DBSslMode string `envconfig:"DB_SSL_MODE" validate:"required,oneof=disable require"`
  29. AWSLambdaRegion *string `envconfig:"AWS_LAMBDA_REGION" validate:"omitempty"`
  30. AWSLambdaAccessKey *string `envconfig:"AWS_LAMBDA_ACCESS_KEY" validate:"omitempty"`
  31. AWSLambdaSecretKey *string `envconfig:"AWS_LAMBDA_SECRET_KEY" validate:"omitempty"`
  32. LifetimeCollectionHeartbeatInterval int `envconfig:"LIFETIME_COLLECTION_HEARTBEAT_INTERVAL" validate:"required"`
  33. LifetimeCollectionGCInterval int `envconfig:"LIFETIME_COLLECTION_GC_INTERVAL" validate:"required"`
  34. LifetimeStateGCInterval int `envconfig:"LIFETIME_STATE_GC_INTERVAL" validate:"required"`
  35. DifyInvocationConnectionIdleTimeout int `envconfig:"DIFY_INVOCATION_CONNECTION_IDLE_TIMEOUT" validate:"required"`
  36. }
  37. func (c *Config) Validate() error {
  38. validator := validator.New()
  39. err := validator.Struct(c)
  40. if err != nil {
  41. return err
  42. }
  43. if c.PluginRemoteInstallingEnabled {
  44. if c.PluginRemoteInstallingHost == "" {
  45. return fmt.Errorf("plugin remote installing host is empty")
  46. }
  47. if c.PluginRemoteInstallingPort == 0 {
  48. return fmt.Errorf("plugin remote installing port is empty")
  49. }
  50. if c.PluginRemoteInstallingMaxConn == 0 {
  51. return fmt.Errorf("plugin remote installing max connection is empty")
  52. }
  53. if c.PluginRemoteInstallServerEventLoopNums == 0 {
  54. return fmt.Errorf("plugin remote install server event loop nums is empty")
  55. }
  56. }
  57. return nil
  58. }
  59. type PlatformType string
  60. const (
  61. PLATFORM_LOCAL PlatformType = "local"
  62. PLATFORM_AWS_LAMBDA PlatformType = "aws_lambda"
  63. )