Blame view

common/components/plugin.go 1.12 KB
77f5eec7   zhangqijia   plugin 插件热更 接口
1
2
3
4
  package components
  
  import (
  	"plugin"
765431a4   zhangqijia   增加schema接口, 抽象 mo...
5
  	"pro2d/common/logger"
77f5eec7   zhangqijia   plugin 插件热更 接口
6
  	"pro2d/pb"
77f5eec7   zhangqijia   plugin 插件热更 接口
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
  	"sync"
  )
  
  type ActionMap 		sync.Map//map[pb.ProtoCode]ActionHandler
  
  type PluginOption func(*Plugin)
  
  type Plugin struct {
  	pluginPath string
  
  	Actions 	sync.Map
  }
  
  func NewPlugin(path string, options ...PluginOption) IPlugin{
  	p := &Plugin{
  		pluginPath: path,
  	}
  	for _, option := range options {
  		option(p)
  	}
  	return p
  }
  
  func (p *Plugin) LoadPlugin() error {
  	plu, err := plugin.Open(p.pluginPath)
  
  	if err != nil {
  		return err
  	}
  	logger.Debug("func LoadPlugin open success...")
  
  	f, err := plu.Lookup("GetActionMap")
  	if err != nil {
  		return err
  	}
  	logger.Debug("func LoadPlugin Lookup success...")
  
  	if x, ok := f.(func()map[interface{}]interface{}); ok {
  		logger.Debug("func LoadPlugin GetActionMap success...")
765431a4   zhangqijia   增加schema接口, 抽象 mo...
46
  		p.SetActions(x())
77f5eec7   zhangqijia   plugin 插件热更 接口
47
48
49
50
51
52
53
54
55
56
57
58
  	}
  
  	return nil
  }
  
  func (p *Plugin) GetAction(cmd uint32) interface{} {
  	f, ok := p.Actions.Load(pb.ProtoCode(cmd))
  	if !ok {
  		return nil
  	}
  	return f
  }
765431a4   zhangqijia   增加schema接口, 抽象 mo...
59
60
61
62
63
64
65
66
  
  func (p *Plugin) SetActions(am map[interface{}]interface{}) {
  	for k, v := range am {
  		cmd := k.(pb.ProtoCode)
  		p.Actions.Delete(cmd)
  		p.Actions.Store(cmd, v)
  	}
  }