gf/g/util/gutil/gutil.go

// Copyright 2017 gf Author(https://gitee.com/johng/gf). All Rights Reserved.
//
// This Source Code Form is subject to the terms of the MIT License.
// If a copy of the MIT was not distributed with this file,
// You can obtain one at https://gitee.com/johng/gf.

// Package gutil provides some uncategorized util functions.
// 
// 工具包.
package gutil

import (
    "bytes"
    "encoding/json"
    "fmt"
    "gitee.com/johng/gf/g/util/gconv"
    "os"
    "reflect"
    "runtime"
)

// 格式化打印变量(类似于PHP-vardump)
func Dump(i...interface{}) {
    for _, v := range i {
        if b, ok := v.([]byte); ok {
            fmt.Print(string(b))
        } else {
            // 主要针对 map[interface{}]* 进行处理，json无法进行encode，
            // 这里强制对所有map进行反射处理转换
            refValue := reflect.ValueOf(v)
            if refValue.Kind() == reflect.Map {
                m := make(map[string]interface{})
                keys := refValue.MapKeys()
                for _, k := range keys {
                    key   := gconv.String(k.Interface())
                    m[key] = refValue.MapIndex(k).Interface()
                }
                v = m
            }
            // json encode并打印到终端
            buffer  := &bytes.Buffer{}
            encoder := json.NewEncoder(buffer)
            encoder.SetEscapeHTML(false)
            encoder.SetIndent("", "\t")
            if err := encoder.Encode(v); err == nil {
                fmt.Print(buffer.String())
            } else {
                fmt.Fprintln(os.Stderr, err.Error())
            }
        }
        //fmt.Println()
    }
}

// 打印完整的调用回溯信息
func PrintBacktrace() {
    index  := 1
    buffer := bytes.NewBuffer(nil)
    for i := 0; i < 10000; i++ {
        if _, cfile, cline, ok := runtime.Caller(i); ok {
            buffer.WriteString(fmt.Sprintf(`%d. %s:%d%s`, index, cfile, cline, "\n"))
            index++
        } else {
            break
        }
    }
    fmt.Print(buffer.String())
}

// 抛出一个异常
func Throw(exception interface{}) {
    panic(exception)
}

// try...catch...
func TryCatch(try func(), catch ... func(exception interface{})) {
    if len(catch) > 0 {
        defer func() {
            if e := recover(); e != nil {
                catch[0](e)
            }
        }()
    }
    try()
}