manager.go 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114
  1. package plugin_manager
  2. import (
  3. "fmt"
  4. "sync"
  5. "github.com/langgenius/dify-plugin-daemon/internal/cluster"
  6. "github.com/langgenius/dify-plugin-daemon/internal/core/dify_invocation"
  7. "github.com/langgenius/dify-plugin-daemon/internal/core/plugin_manager/media_manager"
  8. "github.com/langgenius/dify-plugin-daemon/internal/types/app"
  9. "github.com/langgenius/dify-plugin-daemon/internal/types/entities/plugin_entities"
  10. "github.com/langgenius/dify-plugin-daemon/internal/utils/cache"
  11. "github.com/langgenius/dify-plugin-daemon/internal/utils/lock"
  12. "github.com/langgenius/dify-plugin-daemon/internal/utils/log"
  13. "github.com/langgenius/dify-plugin-daemon/internal/utils/mapping"
  14. )
  15. type PluginManager struct {
  16. m sync.Map
  17. cluster *cluster.Cluster
  18. maxPluginPackageSize int64
  19. workingDirectory string
  20. // mediaManager is used to manage media files like plugin icons, images, etc.
  21. mediaManager *media_manager.MediaManager
  22. // running plugin in storage contains relations between plugin packages and their running instances
  23. runningPluginInStorage mapping.Map[string, string]
  24. // start process lock
  25. startProcessLock *lock.HighGranularityLock
  26. }
  27. var (
  28. manager *PluginManager
  29. )
  30. func InitGlobalPluginManager(cluster *cluster.Cluster, configuration *app.Config) {
  31. manager = &PluginManager{
  32. cluster: cluster,
  33. maxPluginPackageSize: configuration.MaxPluginPackageSize,
  34. workingDirectory: configuration.PluginWorkingPath,
  35. mediaManager: media_manager.NewMediaManager(
  36. configuration.PluginMediaCachePath,
  37. configuration.PluginMediaCacheSize,
  38. ),
  39. startProcessLock: lock.NewHighGranularityLock(),
  40. }
  41. manager.Init(configuration)
  42. }
  43. func GetGlobalPluginManager() *PluginManager {
  44. return manager
  45. }
  46. func (p *PluginManager) Add(plugin plugin_entities.PluginRuntimeInterface) error {
  47. identity, err := plugin.Identity()
  48. if err != nil {
  49. return err
  50. }
  51. p.m.Store(identity.String(), plugin)
  52. return nil
  53. }
  54. func (p *PluginManager) List() []plugin_entities.PluginRuntimeInterface {
  55. var runtimes []plugin_entities.PluginRuntimeInterface
  56. p.m.Range(func(key, value interface{}) bool {
  57. if v, ok := value.(plugin_entities.PluginRuntimeInterface); ok {
  58. runtimes = append(runtimes, v)
  59. }
  60. return true
  61. })
  62. return runtimes
  63. }
  64. func (p *PluginManager) Get(identity plugin_entities.PluginUniqueIdentifier) plugin_entities.PluginRuntimeInterface {
  65. if v, ok := p.m.Load(identity.String()); ok {
  66. if r, ok := v.(plugin_entities.PluginRuntimeInterface); ok {
  67. return r
  68. }
  69. }
  70. return nil
  71. }
  72. func (p *PluginManager) GetAsset(id string) ([]byte, error) {
  73. return p.mediaManager.Get(id)
  74. }
  75. func (p *PluginManager) Init(configuration *app.Config) {
  76. // TODO: init plugin manager
  77. log.Info("start plugin manager daemon...")
  78. // init redis client
  79. if err := cache.InitRedisClient(
  80. fmt.Sprintf("%s:%d", configuration.RedisHost, configuration.RedisPort),
  81. configuration.RedisPass,
  82. ); err != nil {
  83. log.Panic("init redis client failed: %s", err.Error())
  84. }
  85. if err := dify_invocation.InitDifyInvocationDaemon(
  86. configuration.PluginInnerApiURL, configuration.PluginInnerApiKey,
  87. ); err != nil {
  88. log.Panic("init dify invocation daemon failed: %s", err.Error())
  89. }
  90. // start local watcher
  91. p.startLocalWatcher(configuration)
  92. // start remote watcher
  93. p.startRemoteWatcher(configuration)
  94. }