123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114 |
- package plugin_manager
- import (
- "fmt"
- "sync"
- "github.com/langgenius/dify-plugin-daemon/internal/cluster"
- "github.com/langgenius/dify-plugin-daemon/internal/core/dify_invocation"
- "github.com/langgenius/dify-plugin-daemon/internal/core/plugin_manager/media_manager"
- "github.com/langgenius/dify-plugin-daemon/internal/types/app"
- "github.com/langgenius/dify-plugin-daemon/internal/types/entities/plugin_entities"
- "github.com/langgenius/dify-plugin-daemon/internal/utils/cache"
- "github.com/langgenius/dify-plugin-daemon/internal/utils/lock"
- "github.com/langgenius/dify-plugin-daemon/internal/utils/log"
- "github.com/langgenius/dify-plugin-daemon/internal/utils/mapping"
- )
- type PluginManager struct {
- m sync.Map
- cluster *cluster.Cluster
- maxPluginPackageSize int64
- workingDirectory string
- // mediaManager is used to manage media files like plugin icons, images, etc.
- mediaManager *media_manager.MediaManager
- // running plugin in storage contains relations between plugin packages and their running instances
- runningPluginInStorage mapping.Map[string, string]
- // start process lock
- startProcessLock *lock.HighGranularityLock
- }
- var (
- manager *PluginManager
- )
- func InitGlobalPluginManager(cluster *cluster.Cluster, configuration *app.Config) {
- manager = &PluginManager{
- cluster: cluster,
- maxPluginPackageSize: configuration.MaxPluginPackageSize,
- workingDirectory: configuration.PluginWorkingPath,
- mediaManager: media_manager.NewMediaManager(
- configuration.PluginMediaCachePath,
- configuration.PluginMediaCacheSize,
- ),
- startProcessLock: lock.NewHighGranularityLock(),
- }
- manager.Init(configuration)
- }
- func GetGlobalPluginManager() *PluginManager {
- return manager
- }
- func (p *PluginManager) Add(plugin plugin_entities.PluginRuntimeInterface) error {
- identity, err := plugin.Identity()
- if err != nil {
- return err
- }
- p.m.Store(identity.String(), plugin)
- return nil
- }
- func (p *PluginManager) List() []plugin_entities.PluginRuntimeInterface {
- var runtimes []plugin_entities.PluginRuntimeInterface
- p.m.Range(func(key, value interface{}) bool {
- if v, ok := value.(plugin_entities.PluginRuntimeInterface); ok {
- runtimes = append(runtimes, v)
- }
- return true
- })
- return runtimes
- }
- func (p *PluginManager) Get(identity plugin_entities.PluginUniqueIdentifier) plugin_entities.PluginRuntimeInterface {
- if v, ok := p.m.Load(identity.String()); ok {
- if r, ok := v.(plugin_entities.PluginRuntimeInterface); ok {
- return r
- }
- }
- return nil
- }
- func (p *PluginManager) GetAsset(id string) ([]byte, error) {
- return p.mediaManager.Get(id)
- }
- func (p *PluginManager) Init(configuration *app.Config) {
- // TODO: init plugin manager
- log.Info("start plugin manager daemon...")
- // init redis client
- if err := cache.InitRedisClient(
- fmt.Sprintf("%s:%d", configuration.RedisHost, configuration.RedisPort),
- configuration.RedisPass,
- ); err != nil {
- log.Panic("init redis client failed: %s", err.Error())
- }
- if err := dify_invocation.InitDifyInvocationDaemon(
- configuration.PluginInnerApiURL, configuration.PluginInnerApiKey,
- ); err != nil {
- log.Panic("init dify invocation daemon failed: %s", err.Error())
- }
- // start local watcher
- p.startLocalWatcher(configuration)
- // start remote watcher
- p.startRemoteWatcher(configuration)
- }
|