server.go 2.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394
  1. package server
  2. import (
  3. "github.com/getsentry/sentry-go"
  4. "github.com/langgenius/dify-plugin-daemon/internal/cluster"
  5. "github.com/langgenius/dify-plugin-daemon/internal/core/persistence"
  6. "github.com/langgenius/dify-plugin-daemon/internal/core/plugin_manager"
  7. "github.com/langgenius/dify-plugin-daemon/internal/db"
  8. "github.com/langgenius/dify-plugin-daemon/internal/oss"
  9. "github.com/langgenius/dify-plugin-daemon/internal/oss/local"
  10. "github.com/langgenius/dify-plugin-daemon/internal/oss/s3"
  11. "github.com/langgenius/dify-plugin-daemon/internal/oss/tencent_cos"
  12. "github.com/langgenius/dify-plugin-daemon/internal/types/app"
  13. "github.com/langgenius/dify-plugin-daemon/internal/utils/log"
  14. "github.com/langgenius/dify-plugin-daemon/internal/utils/routine"
  15. )
  16. func initOSS(config *app.Config) oss.OSS {
  17. // init oss
  18. var oss oss.OSS
  19. var err error
  20. if config.PluginStorageType == "aws_s3" {
  21. oss, err = s3.NewAWSS3Storage(
  22. config.AWSAccessKey,
  23. config.AWSSecretKey,
  24. config.AWSRegion,
  25. config.PluginStorageOSSBucket,
  26. )
  27. if err != nil {
  28. log.Panic("Failed to create aws s3 storage: %s", err)
  29. }
  30. } else if config.PluginStorageType == "local" {
  31. oss = local.NewLocalStorage(config.PluginStorageLocalRoot)
  32. } else if config.PluginStorageType == "tencent_cos" {
  33. oss, err = tencent_cos.NewTencentCOSStorage(
  34. config.TencentCOSSecretId,
  35. config.TencentCOSSecretKey,
  36. config.TencentCOSRegion,
  37. config.PluginStorageOSSBucket)
  38. if err != nil {
  39. log.Panic("Failed to create tencent cos storage: %s", err)
  40. }
  41. } else {
  42. log.Panic("Invalid plugin storage type: %s", config.PluginStorageType)
  43. }
  44. return oss
  45. }
  46. func (app *App) Run(config *app.Config) {
  47. // init routine pool
  48. if config.SentryEnabled {
  49. routine.InitPool(config.RoutinePoolSize, sentry.ClientOptions{
  50. Dsn: config.SentryDSN,
  51. AttachStacktrace: config.SentryAttachStacktrace,
  52. TracesSampleRate: config.SentryTracesSampleRate,
  53. SampleRate: config.SentrySampleRate,
  54. EnableTracing: config.SentryTracingEnabled,
  55. })
  56. } else {
  57. routine.InitPool(config.RoutinePoolSize)
  58. }
  59. // init db
  60. db.Init(config)
  61. // init oss
  62. oss := initOSS(config)
  63. // create manager
  64. manager := plugin_manager.InitGlobalManager(oss, config)
  65. // create cluster
  66. app.cluster = cluster.NewCluster(config, manager)
  67. // register plugin lifetime event
  68. manager.AddPluginRegisterHandler(app.cluster.RegisterPlugin)
  69. // init manager
  70. manager.Launch(config)
  71. // init persistence
  72. persistence.InitPersistence(oss, config)
  73. // launch cluster
  74. app.cluster.Launch()
  75. // start http server
  76. app.server(config)
  77. // block
  78. select {}
  79. }