misc.go

package utils

import (
	"fmt"
	"math"
	"math/rand"
	"strings"
	"time"
	"net"
)

type Misc struct{}

func NewMisc() *Misc {
	return &Misc{}
}

// get map default
func (m *Misc) GetMapDefault(mapValue map[string]interface{}, key string, def interface{}) interface{} {
	value, ok := mapValue[key]
	if ok {
		return value
	}else {
		return def
	}
}

// rand string
func (m *Misc) RandString(strlen int) string {
	codes := "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"
	codeLen := len(codes)
	data := make([]byte, strlen)
	rand.Seed(time.Now().UnixNano() + rand.Int63() + rand.Int63() + rand.Int63() + rand.Int63())
	for i := 0; i < strlen; i++ {
		idx := rand.Intn(codeLen)
		data[i] = byte(codes[idx])
	}
	return string(data)
}

// rand int
func (m *Misc) RandInt(strLen int) string {
	codes := "0123456789"
	codeLen := len(codes)
	data := make([]byte, strLen)
	rand.Seed(time.Now().UnixNano() + rand.Int63() + rand.Int63() + rand.Int63() + rand.Int63())
	for i := 0; i < strLen; i++ {
		idx := rand.Intn(codeLen)
		data[i] = byte(codes[idx])
	}
	return string(data)
}

// get local ip
func (m *Misc) GetLocalIp() string {
	addrSlice, err := net.InterfaceAddrs()
	if nil != err {
		return "localhost"
	}
	for _, addr := range addrSlice {
		if ipnet, ok := addr.(*net.IPNet); ok && !ipnet.IP.IsLoopback() {
			if nil != ipnet.IP.To4() {
				return  ipnet.IP.String()
			}
		}
	}
	return "localhost"
}

// 分页方法
// total: 一共多少记录
// page:  当前是第几页
// pagesize: 每页多少
// url: url里面的{page}会被替换成页码
// args:
//  order 分页条的组成，是一个数组，可以按着1-6的序号，选择分页条组成部分和每个部分的顺序
//  a_count   分页条中a页码链接的总数量,不包含当前页的a标签，默认10个。
//
func (m *Misc) Page(total, page, pagesize int, url string, args ...interface{}) string {
	order := []int{1, 2, 3, 4, 5, 6}
	a_count := 10
	if len(args) >= 1 {
		order = args[0].([]int)
	}
	if len(args) >= 2 {
		a_count = args[1].(int)
	}
	a_num := a_count
	first := "首页"
	last := "尾页"
	pre := "上页"
	next := "下页"
	if a_num%2 == 0 {
		a_num++
	}
	pages := int(math.Ceil(float64(total) / float64(pagesize)))
	curpage := page
	if curpage > pages || curpage <= 0 {
		curpage = 1
	}
	body := `<span class="page_body">`
	prefix := ""
	subfix := ""
	start := curpage - ((a_num - 1) / 2)
	end := curpage + ((a_num - 1) / 2)
	if start <= 0 {
		start = 1
	}
	if end > pages {
		end = pages
	}
	if pages >= a_num {
		if curpage <= (a_num-1)/2 {
			end = a_num
		}
		if end-curpage <= (a_num-1)/2 {
			start -= int(math.Floor(float64(a_num)/float64(2))) - (end - curpage)
		}
	}
	for i := start; i <= end; i++ {
		if i == curpage {
			body += fmt.Sprintf(`<a class="page_cur_page" href="javascript:void(0);"><b>%d</b></a>`, i)
		} else {
			body += fmt.Sprintf(`<a href="%s">%d</a>`, strings.Replace(url, "{page}", fmt.Sprintf("%d", i), 1), i)

		}
	}
	body += "</span>"
	if curpage > 1 {
		prefix = fmt.Sprintf(`<span class="page_bar_prefix"><a href="%s">%s</a><a href="%s">%s</a></span>`, strings.Replace(url, "{page}", fmt.Sprintf("%d", 1), 1), first, strings.Replace(url, "{page}", fmt.Sprintf("%d", curpage-1), 1), pre)
	}
	if curpage != pages {
		subfix = fmt.Sprintf(`<span class="page_bar_subfix"><a href="%s">%s</a><a href="%s">%s</a></span>`, strings.Replace(url, "{page}", fmt.Sprintf("%d", curpage+1), 1), next, strings.Replace(url, "{page}", fmt.Sprintf("%d", pages), 1), last)
	}
	info := fmt.Sprintf(`<span class="page_cur">第%d/%d页</span>`, curpage, pages)
	id := fmt.Sprintf("gsd09fhas9d%d%d%d", rand.Intn(1000), rand.Intn(1000), rand.Intn(1000))
	gostr := fmt.Sprintf(`<script>function ekup(){if(event.keyCode==13){clkyup();}}function clkyup(){var num=document.getElementById('%s').value;if(!/^\d+$/.test(num)||num<=0||num>%d){alert('请输入正确页码!');return;};location='%s'.replace(/\{page\}/,document.getElementById('%s').value);}</script><span class="page_input_num"><input onkeyup="ekup()" type="text" id="%s" style="width:40px;vertical-align:text-baseline;padding:0 2px;font-size:10px;border:1px solid gray;"/></span><span class="page_btn_go" onclick="clkyup();" style="cursor:pointer;">转到</span>`, id, pages, url, id, id)
	totalstr := fmt.Sprintf(`<span class="page_total">共%d条</span>`, total)
	pagenation := []string{totalstr, info, prefix, body, subfix, gostr}
	output := []string{}
	for _, v := range order {
		if v-1 < len(pagenation) && v-1 >= 0 {
			output = append(output, pagenation[v-1])
		}
	}
	if pages > 1 {
		return strings.Join(output, "")
	}
	return ""
}