mirror of
https://github.com/beego/bee.git
synced 2024-11-24 03:20:55 +00:00
74 lines
1.5 KiB
Go
74 lines
1.5 KiB
Go
package version
|
|
|
|
import (
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"runtime"
|
|
"text/template"
|
|
|
|
"time"
|
|
|
|
beeLogger "github.com/beego/bee/v2/logger"
|
|
"github.com/beego/bee/v2/utils"
|
|
)
|
|
|
|
// RuntimeInfo holds information about the current runtime.
|
|
type RuntimeInfo struct {
|
|
GoVersion string
|
|
GOOS string
|
|
GOARCH string
|
|
NumCPU int
|
|
GOPATH string
|
|
GOROOT string
|
|
Compiler string
|
|
BeeVersion string
|
|
Published string
|
|
}
|
|
|
|
// InitBanner loads the banner and prints it to output
|
|
// All errors are ignored, the application will not
|
|
// print the banner in case of error.
|
|
func InitBanner(out io.Writer, in io.Reader) {
|
|
if in == nil {
|
|
beeLogger.Log.Fatal("The input is nil")
|
|
}
|
|
|
|
banner, err := ioutil.ReadAll(in)
|
|
if err != nil {
|
|
beeLogger.Log.Fatalf("Error while trying to read the banner: %s", err)
|
|
}
|
|
|
|
show(out, string(banner))
|
|
}
|
|
|
|
func show(out io.Writer, content string) {
|
|
t, err := template.New("banner").
|
|
Funcs(template.FuncMap{"Now": Now}).
|
|
Parse(content)
|
|
|
|
if err != nil {
|
|
beeLogger.Log.Fatalf("Cannot parse the banner template: %s", err)
|
|
}
|
|
|
|
err = t.Execute(out, RuntimeInfo{
|
|
GetGoVersion(),
|
|
runtime.GOOS,
|
|
runtime.GOARCH,
|
|
runtime.NumCPU(),
|
|
os.Getenv("GOPATH"),
|
|
runtime.GOROOT(),
|
|
runtime.Compiler,
|
|
version,
|
|
utils.GetLastPublishedTime(),
|
|
})
|
|
if err != nil {
|
|
beeLogger.Log.Error(err.Error())
|
|
}
|
|
}
|
|
|
|
// Now returns the current local time in the specified layout
|
|
func Now(layout string) string {
|
|
return time.Now().Format(layout)
|
|
}
|