Upgrade to go 1.8 version, supports shutdown and reboot gracefully.

Author: andeya

README.md

@@ -20,7 +20,7 @@ v1.0
 
 ### Requirements
 
-Go Version ≥1.6
+Go Version ≥1.8
 
 ## Quick Start
 
@@ -68,6 +68,7 @@ think run [appname]
 - Supports XSRF security filtering
 - Supports near-LRU memory caching (mainly used for static file cache)
 - Nice and easy to use configuration file, automatically write default values
+- Supports shutdown and reboot gracefully
 
 - `struct Handler` multi-usage
 
@@ -112,7 +113,7 @@ func main() {
   // )
 
   // Start the service
-  app.Run()
+  thinkgo.Run()
 }
 
 /*
@@ -256,6 +257,20 @@ app2.Group("home")
 app2.Run()
 ```
 
+## shutdown and reboot
+
+- shutdown gracefully
+
+```sh
+kill [pid]
+```
+
+- reboot gracefully
+
+```sh
+kill -USR2 [pid]
+```
+
 ## Configuration
 
 - Each instance of the application has a single configuration (file name format `config/{appname}[_{version}].ini`). Refer to the following:

README_ZH.md

@@ -22,7 +22,7 @@ v1.0
 
 ### 安装要求
 
-Go Version ≥1.6
+Go Version ≥1.8
 
 ## 快速使用
 
@@ -69,6 +69,7 @@ think run [appname]
 - 支持Gzip全局配置
 - 提供XSRF跨站请求伪造安全过滤
 - 简单整洁的配置文件，且自动补填默认值方便设置
+- 支持平滑关闭与重启
 
 - `struct Handler` 的多用途合一
 
@@ -257,6 +258,20 @@ app2.Group("home")
 app2.Run()
 ```
 
+## 平滑关闭与重启
+
+- 平滑关闭
+
+```sh
+kill [pid]
+```
+
+- 平滑重启
+
+```sh
+kill -USR2 [pid]
+```
+
 ## 配置文件说明
 
 - 应用的各实例均有单独一份配置，其文件名格式 `config/{appname}[_{version}].ini`，配置详情：

global.go

@@ -15,6 +15,7 @@
 package thinkgo
 
 import (
+	"context"
 	"encoding/json"
 	"fmt"
 	"net/http"
@@ -70,6 +71,11 @@ type (
 
 		syslog *logging.Logger
 		bizlog *logging.Logger
+
+		// finalizer is called after the services shutdown.
+		finalizer func(context.Context) error
+
+		graceOnce sync.Once
 	}
 	// PresetStatic is the system default static file routing information
 	PresetStatic struct {
@@ -201,31 +207,16 @@ func GetFrame(name string, version ...string) (*Framework, bool) {
 // Run starts all web services.
 func Run() {
 	global.framesLock.Lock()
-	count := len(global.frames)
-	if count == 0 {
-		global.framesLock.Unlock()
-		return
-	}
-	var last = -1
-	for i := count - 1; i >= 0; i-- {
-		if !global.frames[i].Running() {
-			last = i
-			break
-		}
-	}
-	if last == -1 {
-		global.framesLock.Unlock()
-		return
-	}
-	for _, frame := range global.frames[:last] {
+	for _, frame := range global.frames {
 		if !frame.Running() {
 			go frame.run()
 			time.Sleep(time.Second)
 		}
 	}
-	frame := global.frames[last]
 	global.framesLock.Unlock()
-	frame.run()
+	global.graceOnce.Do(func() {
+		graceSignal()
+	})
 }
 
 // Running returns whether the frame service is running.
@@ -237,14 +228,42 @@ func Running(name string, version ...string) bool {
 	return frame.Running()
 }
 
-// Close closes all the frame services.
-// TODO: close listeners
-func Close() {
+// SetFinalizer sets the function which is called after the services shutdown.
+func SetFinalizer(finalizer func(context.Context) error) {
+	global.finalizer = finalizer
+}
+
+const (
+	// SHUTDOWN_TIMEOUT the time-out period for closing the service
+	SHUTDOWN_TIMEOUT = 1 * time.Minute
+)
+
+// Shutdown closes all the frame services gracefully.
+func Shutdown(timeout ...time.Duration) {
+	Print("\x1b[46m[SYS]\x1b[0m shutting down servers...")
 	global.framesLock.Lock()
 	defer global.framesLock.Unlock()
+	var d = SHUTDOWN_TIMEOUT
+	if len(timeout) > 0 {
+		d = timeout[0]
+	}
+	// shut down gracefully, but wait no longer than d before halting
+	ctxTimeout, _ := context.WithTimeout(context.Background(), d)
+	count := new(sync.WaitGroup)
 	for _, frame := range global.frames {
-		frame.Close()
+		count.Add(1)
+		go func(fm *Framework) {
+			fm.shutdown(ctxTimeout)
+			count.Done()
+		}(frame)
+	}
+	count.Wait()
+	if global.finalizer != nil {
+		if err := global.finalizer(ctxTimeout); err != nil {
+			Error(err.Error())
+		}
 	}
+	Print("\x1b[46m[SYS]\x1b[0m servers gracefully stopped.")
 	CloseLog()
 }