server calls AfterInit

hook.go

@@ -5,6 +5,7 @@ import (
 )
 
 type Hook interface {
+	Name() string
 	AfterInit()
 	AfterComm(*AfterCommData)
 	AfterConn(*AfterConnData)

hook_file.go

@@ -18,7 +18,7 @@ type HookFile struct {
 	file io.Writer
 }
 
-func (h *HookFile) prefix() string {
+func (h *HookFile) Name() string {
 	return "file"
 }
 
@@ -47,23 +47,23 @@ func (h *HookFile) AfterInit() {
 func (h *HookFile) AfterComm(d *AfterCommData) {
 	writer, err := h.writer()
 	if err != nil {
-		fmt.Printf("[%s] %s\n", h.prefix(), err)
+		fmt.Printf("[%s] %s\n", h.Name(), err)
 		return
 	}
 
 	if _, err := fmt.Fprintf(writer, fileCommJson, d.OccurredAt.Format(time.RFC3339), d.ConnID, d.Direction, d.Data); err != nil {
-		fmt.Printf("[%s] file append error: %s\n", h.prefix(), err)
+		fmt.Printf("[%s] file append error: %s\n", h.Name(), err)
 	}
 }
 
 func (h *HookFile) AfterConn(d *AfterConnData) {
 	writer, err := h.writer()
 	if err != nil {
-		fmt.Printf("[%s] %s\n", h.prefix(), err)
+		fmt.Printf("[%s] %s\n", h.Name(), err)
 		return
 	}
 
 	if _, err := fmt.Fprintf(writer, fileConnJson, d.OccurredAt.Format(time.RFC3339), d.ConnID, d.MailFrom, d.MailTo, d.Elapse); err != nil {
-		fmt.Printf("[%s] file append error: %s\n", h.prefix(), err)
+		fmt.Printf("[%s] file append error: %s\n", h.Name(), err)
 	}
 }

hook_file_test.go

@@ -49,10 +49,10 @@ func TestHookFileConst(t *testing.T) {
 	}
 }
 
-func TestHookFilePrefix(t *testing.T) {
+func TestHookFileName(t *testing.T) {
 	f := &HookFile{}
 	expect := "file"
-	got := f.prefix()
+	got := f.Name()
 	if got != expect {
 		t.Errorf("expected %s, got %s", expect, got)
 	}

hook_mysql.go

@@ -15,7 +15,7 @@ type HookMysql struct {
 	pool *sql.DB // Database connection pool.
 }
 
-func (h *HookMysql) prefix() string {
+func (h *HookMysql) Name() string {
 	return "mysql"
 }
 
@@ -44,7 +44,7 @@ func (h *HookMysql) AfterInit() {
 func (h *HookMysql) AfterComm(d *AfterCommData) {
 	conn, err := h.conn()
 	if err != nil {
-		fmt.Printf("[%s] %s\n", h.prefix(), err)
+		fmt.Printf("[%s] %s\n", h.Name(), err)
 		return
 	}
 
@@ -57,14 +57,14 @@ func (h *HookMysql) AfterComm(d *AfterCommData) {
 		d.Data,
 	)
 	if err != nil {
-		fmt.Printf("[%s] db exec error: %s\n", h.prefix(), err)
+		fmt.Printf("[%s] db exec error: %s\n", h.Name(), err)
 	}
 }
 
 func (h *HookMysql) AfterConn(d *AfterConnData) {
 	conn, err := h.conn()
 	if err != nil {
-		fmt.Printf("[%s] %s\n", h.prefix(), err)
+		fmt.Printf("[%s] %s\n", h.Name(), err)
 		return
 	}
 
@@ -77,6 +77,6 @@ func (h *HookMysql) AfterConn(d *AfterConnData) {
 		d.Elapse,
 	)
 	if err != nil {
-		fmt.Printf("[%s] db exec error: %s\n", h.prefix(), err)
+		fmt.Printf("[%s] db exec error: %s\n", h.Name(), err)
 	}
 }

hook_mysql_test.go

@@ -27,13 +27,13 @@ func TestHookMysqlConst(t *testing.T) {
 	}
 }
 
-func TestHookMysqlPrefix(t *testing.T) {
+func TestHookMysqlName(t *testing.T) {
 	mysql := &HookMysql{}
 	var expect string
 	var got string
 
 	expect = "mysql"
-	got = mysql.prefix()
+	got = mysql.Name()
 	if got != expect {
 		t.Errorf("expected %s, got %s", expect, got)
 	}

hook_sqlite.go

@@ -31,7 +31,7 @@ type HookSqlite struct {
 	pool *sql.DB // Database connection pool.
 }
 
-func (h *HookSqlite) prefix() string {
+func (h *HookSqlite) Name() string {
 	return "sqlite"
 }
 
@@ -57,25 +57,25 @@ func (h *HookSqlite) conn() (*sql.DB, error) {
 func (h *HookSqlite) AfterInit() {
 	conn, err := h.conn()
 	if err != nil {
-		fmt.Printf("[%s] %s\n", h.prefix(), err)
+		fmt.Printf("[%s] %s\n", h.Name(), err)
 		return
 	}
 
 	_, err = conn.Exec(sqliteConnCreateTable)
 	if err != nil {
-		fmt.Printf("[%s] db exec error: %s\n", h.prefix(), err)
+		fmt.Printf("[%s] db exec error: %s\n", h.Name(), err)
 	}
 
 	_, err = conn.Exec(sqliteCommCreateTable)
 	if err != nil {
-		fmt.Printf("[%s] db exec error: %s\n", h.prefix(), err)
+		fmt.Printf("[%s] db exec error: %s\n", h.Name(), err)
 	}
 }
 
 func (h *HookSqlite) AfterComm(d *AfterCommData) {
 	conn, err := h.conn()
 	if err != nil {
-		fmt.Printf("[%s] %s\n", h.prefix(), err)
+		fmt.Printf("[%s] %s\n", h.Name(), err)
 		return
 	}
 
@@ -88,14 +88,14 @@ func (h *HookSqlite) AfterComm(d *AfterCommData) {
 		d.Data,
 	)
 	if err != nil {
-		fmt.Printf("[%s] db exec error: %s\n", h.prefix(), err)
+		fmt.Printf("[%s] db exec error: %s\n", h.Name(), err)
 	}
 }
 
 func (h *HookSqlite) AfterConn(d *AfterConnData) {
 	conn, err := h.conn()
 	if err != nil {
-		fmt.Printf("[%s] %s\n", h.prefix(), err)
+		fmt.Printf("[%s] %s\n", h.Name(), err)
 		return
 	}
 
@@ -108,6 +108,6 @@ func (h *HookSqlite) AfterConn(d *AfterConnData) {
 		d.Elapse,
 	)
 	if err != nil {
-		fmt.Printf("[%s] db exec error: %s\n", h.prefix(), err)
+		fmt.Printf("[%s] db exec error: %s\n", h.Name(), err)
 	}
 }

hook_sqlite_test.go

@@ -27,10 +27,10 @@ func TestHookSqliteConst(t *testing.T) {
 	}
 }
 
-func TestHookSqlitePrefix(t *testing.T) {
+func TestHookSqliteName(t *testing.T) {
 	sqlite := &HookSqlite{}
 	expect := "sqlite"
-	got := sqlite.prefix()
+	got := sqlite.Name()
 	if got != expect {
 		t.Errorf("expected %s, got %s", expect, got)
 	}

plugins.go

@@ -76,7 +76,6 @@ func (p *Plugins) load() error {
 			continue
 		}
 
-		plug.AfterInit()
 		p.hooks = append(p.hooks, plug)
 	}
 

server.go

@@ -35,6 +35,10 @@ func (s *Server) Start() error {
 		return err
 	}
 	s.Hooks = append(s.Hooks, pl.hooks...)
+	for _, hook := range s.Hooks {
+		s.log.Printf("use %s hook", hook.Name())
+		hook.AfterInit()
+	}
 
 	addr, err := net.ResolveTCPAddr("tcp", fmt.Sprintf("%s:%d", s.Addr, s.Port))
 	if err != nil {