!77 ppeof

Merge pull request !77 from bandl/feat-pporf-plugins

conf/public_conf.go

@@ -2,6 +2,7 @@ package conf
 
 import (
 	"log"
+	"sync"
 
 	"github.com/spf13/viper"
 )
@@ -10,7 +11,10 @@ const (
 	linuxPath = "/etc/wheat-cache/"
 )
 
+var confLock sync.Once
+
 func init() {
+	confLock.Do(func() {
 		setDefaultConfValue()
 		err := LoadConf("")
 		switch err.(type) {
@@ -22,6 +26,8 @@ func init() {
 			log.Fatalf("the resolution of the profile failed, err: %v", err)
 
 		}
+	},
+	)
 }
 
 func setDefaultConfValue() {

conf/wheat-cache.yaml

@@ -49,3 +49,5 @@ gateway:
   port: 5891
   target: ["127.0.0.1:5890"]
 
+mock-plugin:
+  pprof-addr: "127.0.0.1:8000"  

plugins/config/middle.gen.go

@@ -12,8 +12,9 @@ import (
 func GetMiddlewareMap() map[string]plugins.PluginInterface {
 
 	mockPlugin := mockPlugin.NewPlugin()
-	return map[string]plugins.PluginInterface{
+	mockPlugin.Init()
 
+	return map[string]plugins.PluginInterface{
 		mockPlugin.Name(): mockPlugin,
 	}
 }

plugins/config/middle.template

@@ -14,9 +14,10 @@ import (
 func GetMiddlewareMap() map[string]plugins.PluginInterface {
     {%for dir in dirs %}
         {{dir[0]}}:={{dir[0]}}.NewPlugin()
-        {%- endfor%}
+        {{dir[0]}}.Init()
+        {% endfor%}
     return map[string]plugins.PluginInterface{
-		{%for dir in dirs %}
+		{%for dir in dirs -%}
         {{dir[0]}}.Name():{{dir[0]}},
         {%- endfor%}
 	}

plugins/mock-plugin/mock.go

@@ -1,11 +1,27 @@
 package mockplugin
 
-import "fmt"
+import (
+	"fmt"
+	"net/http"
+	_ "net/http/pprof"
+
+	_ "gitee.com/timedb/wheatCache/conf"
+	"gitee.com/timedb/wheatCache/pkg/logx"
+	"github.com/spf13/viper"
+)
 
 type MockPlugin struct {
 }
 
 func (m *MockPlugin) Init() {
+	addr := viper.GetString("mock-plugin.pprof-addr")
+	if addr != "" {
+		go func() {
+			logx.Debugln(http.ListenAndServe(addr, nil))
+		}()
+	}
+
+	logx.Debug("mock-plugins status: start")
 }
 
 func (m *MockPlugin) Exec(msg interface{}) (interface{}, error) {