public/tools/tools.go

package tools

import (
	"crypto/md5"
	crand "crypto/rand"
	"crypto/sha1"
	"encoding/base64"
	"encoding/hex"
	"errors"
	"fmt"
	"html/template"
	"io"
	"net/http"
	"os"
	"path/filepath"
	"public/mylog"
	"reflect"
	"sort"
	"strings"

	"data/config"

	"github.com/ant0ine/go-json-rest/rest"
)

/*
获取程序运行路径
*/
func GetCurrentDirectory() string {
	dir, _ := filepath.Abs(filepath.Dir(os.Args[0]))
	return strings.Replace(dir, "\\", "/", -1)
}

func Md5Encoder(src string) string {
	h := md5.New()
	h.Write([]byte(src)) // 需要加密的字符串
	//    fmt.Printf("%x\n", h.Sum(nil)) // 输出加密结
	ret := fmt.Sprintf("%x", h.Sum(nil))
	return strings.ToUpper(ret)
}

//合并数组
func Copy(dest []interface{}, src []interface{}) (result []interface{}) {
	result = make([]interface{}, len(dest)+len(src))
	copy(result, dest)
	copy(result[len(dest):], src)
	return
}

//删除数组
func DeleteArray(src []interface{}, index int) (result []interface{}) {
	result = append(src[:index], src[(index+1):]...)
	return
}

//生成32位md5字串
func GetMd5String(s string) string {
	h := md5.New()
	h.Write([]byte(s))
	return hex.EncodeToString(h.Sum(nil))
}

//获取总页数
func GetTotalPageNum(pageSize, totalCount int) int {
	return (totalCount + pageSize - 1) / pageSize
}

//生成32位guid
func UniqueId() string {
	b := make([]byte, 48)
	if _, err := io.ReadFull(crand.Reader, b); err != nil {
		return ""
	}
	return GetMd5String(base64.URLEncoding.EncodeToString(b))
}

//删除切片index
func DeleteSlice(slice interface{}, index int) (interface{}, error) {
	sliceValue := reflect.ValueOf(slice)
	length := sliceValue.Len()
	if slice == nil || length == 0 || (length-1) < index {
		return nil, errors.New("error")
	}
	if length-1 == index {
		return sliceValue.Slice(0, index).Interface(), nil
	} else {
		return reflect.AppendSlice(sliceValue.Slice(0, index), sliceValue.Slice(index+1, length)).Interface(), nil
	}
}

//查找int最小值
func MinimumInt(rest []int) int {
	minimum := rest[0]
	for _, v := range rest {
		if v < minimum {
			minimum = v
		}
	}
	return minimum
}

func LoadTemplate(list ...string) *template.Template {
	var tmp []string
	for _, v := range list {
		if CheckFileIsExist(GetModelPath() + config.Static_host[0] + v) {
			tmp = append(tmp, GetModelPath()+config.Static_host[0]+v)
		} else {
			mylog.Debug("file does not exist:" + GetModelPath() + config.Static_host[0] + v)
			panic(GetModelPath() + config.Static_host[0] + v)
		}
	}
	return template.Must(template.ParseFiles(tmp...))
}

/*
	执行模版渲染，
	name:模版名字，""则无名字
	data:传参列表
	list:模版列表
*/
func ExecuteTemplate(w rest.ResponseWriter, name string, data interface{}, list ...string) error {
	t := LoadTemplate(list...)
	w.(http.ResponseWriter).Header().Set("Content-Type", "text/html; charset=utf-8")

	if len(name) == 0 {
		return t.Execute(w.(http.ResponseWriter), data)
	} else {
		return t.ExecuteTemplate(w.(http.ResponseWriter), name, data)
	}
}

//按字典顺序排序
func DictSort(res []string) (str string) {
	sort.Strings(res)
	if len(res) > 0 {
		for _, v := range res {
			str += v
		}
	}
	return
}

//SHA1加密
func Sha1Encrypt(str string) string {
	h := sha1.New()
	h.Write([]byte(str))
	bs := h.Sum(nil)
	return fmt.Sprintf("%x", bs)
}

//中文字符切割时有问题。采用此方式不会有问题
func GetUtf8Str(str string) []rune {
	return []rune(str)
}