leaf游戏服务器模块源码-module

原文链接

leaf模块

1. package leaf
   
2. 
   
3. import (
   
4.         "github.com/name5566/leaf/cluster"
   
5.         "github.com/name5566/leaf/conf"
   
6.         "github.com/name5566/leaf/console"
   
7.         "github.com/name5566/leaf/log"
   
8.         "github.com/name5566/leaf/module"
   
9.         "os"
   
10.         "os/signal"
    
11. )
    
12. 
    
13. func Run(mods ...module.Module) {
    
14.         // logger
    
15.         if conf.LogLevel != "" {
    
16.                 logger, err := log.New(conf.LogLevel, conf.LogPath, conf.LogFlag)
    
17.                 if err != nil {
    
18.                         panic(err)
    
19.                 }
    
20.                 log.Export(logger)
    
21.                 defer logger.Close()
    
22.         }
    
23. 
    
24.         log.Release("Leaf %v starting up", version)
    
25. 
    
26.         // module
    
27.         for i := 0; i < len(mods); i++ {
    
28.                 module.Register(mods[i])
    
29.         }
    
30.         module.Init()
    
31. 
    
32.         // cluster
    
33.         cluster.Init()
    
34. 
    
35.         // console
    
36.         console.Init()
    
37. 
    
38.         // close
    
39.         c := make(chan os.Signal, 1)
    
40.         signal.Notify(c, os.Interrupt, os.Kill)
    
41.         sig := <-c
    
42.         log.Release("Leaf closing down (signal: %v)", sig)
    
43.         console.Destroy()
    
44.         cluster.Destroy()
    
45.         module.Destroy()
    
46. }
    

复制代码

main()模块调用

1. func main() {
   
2.         lconf.LogLevel = conf.Server.LogLevel
   
3.         lconf.LogPath = conf.Server.LogPath
   
4.         lconf.LogFlag = conf.LogFlag
   
5.         lconf.ConsolePort = conf.Server.ConsolePort
   
6.         lconf.ProfilePath = conf.Server.ProfilePath
   
7. 
   
8.         leaf.Run(
   
9.                 game.Module,
   
10.                 gate.Module,
    
11.                 login.Module,
    
12.         )
    
13. }
    

复制代码

1. package module
   
2. 
   
3. import (
   
4.         "github.com/name5566/leaf/conf"
   
5.         "github.com/name5566/leaf/log"
   
6.         "runtime"
   
7.         "sync"
   
8. )
   
9. 
   
10. type Module interface {
    
11.         OnInit()
    
12.         OnDestroy()
    
13.         Run(closeSig chan bool)
    
14. }
    
15. 
    
16. type module struct {
    
17.         mi       Module
    
18.         closeSig chan bool
    
19.         wg       sync.WaitGroup
    
20. }
    
21. 
    
22. var mods []*module
    
23. 
    
24. func Register(mi Module) {
    
25.         m := new(module)
    
26.         m.mi = mi
    
27.         m.closeSig = make(chan bool, 1)
    
28. 
    
29.         mods = append(mods, m)
    
30. }
    
31. 
    
32. func Init() {
    
33.         for i := 0; i < len(mods); i++ {
    
34.                 mods[i].mi.OnInit()
    
35.         }
    
36. 
    
37.         for i := 0; i < len(mods); i++ {
    
38.                 m := mods[i]
    
39.                 m.wg.Add(1)
    
40.                 go run(m)
    
41.         }
    
42. }
    
43. 
    
44. func Destroy() {
    
45.         for i := len(mods) - 1; i >= 0; i-- {
    
46.                 m := mods[i]
    
47.                 m.closeSig <- true
    
48.                 m.wg.Wait()
    
49.                 destroy(m)
    
50.         }
    
51. }
    
52. 
    
53. func run(m *module) {
    
54.         m.mi.Run(m.closeSig)
    
55.         m.wg.Done()
    
56. }
    
57. 
    
58. func destroy(m *module) {
    
59.         defer func() {
    
60.                 if r := recover(); r != nil {
    
61.                         if conf.LenStackBuf > 0 {
    
62.                                 buf := make([]byte, conf.LenStackBuf)
    
63.                                 l := runtime.Stack(buf, false)
    
64.                                 log.Error("%v: %s", r, buf[:l])
    
65.                         } else {
    
66.                                 log.Error("%v", r)
    
67.                         }
    
68.                 }
    
69.         }()
    
70. 
    
71.         m.mi.OnDestroy()
    
72. }
    

复制代码