add version from flag

Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>
This commit is contained in:
Bo-Yi Wu
2016-05-09 14:48:07 +08:00
parent 822c8038d2
commit d9f7c16432
4 changed files with 13 additions and 8 deletions

View File

@@ -1,10 +1,5 @@
package gorush
const (
// Version is gorush server version.
Version = "1.2.1"
)
const (
// PlatFormIos constant is 1 for iOS
PlatFormIos = iota + 1

View File

@@ -6,10 +6,16 @@ import (
"runtime"
)
var version string
func SetVersion(ver string) {
version = ver
}
// PrintGoRushVersion provide print server engine
func PrintGoRushVersion() {
fmt.Printf(`GoRush %s, Compiler: %s %s, Copyright (C) 2016 Bo-Yi Wu, Inc.`,
Version,
version,
runtime.Compiler,
runtime.Version())
}
@@ -18,7 +24,7 @@ func PrintGoRushVersion() {
func VersionMiddleware() gin.HandlerFunc {
// Set out header value for each response
return func(c *gin.Context) {
c.Writer.Header().Set("Server-Version", "GoRush/"+Version)
c.Writer.Header().Set("Server-Version", "GoRush/"+version)
c.Next()
}
}