analyze/utility/userTask.go

package utility

import (
	"analyze/internal/model/entity"
	"fmt"
	"math"
	"time"
)

var (
	y_m_day = map[int]int{1: 31, 2: 28, 3: 31, 4: 30, 5: 31, 6: 30, 7: 31, 8: 31, 9: 30, 10: 31, 11: 30, 12: 31}
)

func GetPreviousMarket(sTime, eTime time.Time) int64 {
	var os_time int64
	s_time := sTime
	if SEMonth(sTime, eTime) {
		var min int
		//统计月份
		for sTime.Before(eTime) {
			sTime = sTime.AddDate(0, 1, 0)
			min++
		}
		os_time = s_time.AddDate(0, -min, 0).Unix()
	} else {
		os_time = s_time.AddDate(0, 0, -int(math.Ceil(eTime.Sub(sTime).Hours()/24))).Unix()
	}

	return os_time
}

// 判断是否月初到月末
func SEMonth(sTime, eTime time.Time) bool {
	var day int
	month := int(eTime.Month())
	if month == 2 {
		if eTime.Year()%4 == 0 {
			day = 29
		} else {
			day = 28
		}
	} else {
		day = y_m_day[month]
	}

	if sTime.Day() == 1 && eTime.Day() == day {
		return true
	}
	return false
}

func GetMonthData(sTime, eTime time.Time) (_b bool, dateSpans []entity.DateSpan) {
	//整月多取一个月进行环比
	if SEMonth(sTime, eTime) {
		_b = true
		sTime = sTime.AddDate(0, -1, 0)
	}
	dateSpans = getBidAmountStatistics(sTime, eTime)
	return
}

func getBidAmountStatistics(sTime, eTime time.Time) (dateSpans []entity.DateSpan) {
	tmpTime, rTime, tEndTime := sTime, sTime, getMonthRange(eTime, false)
	for rTime.Before(tEndTime) {
		ts, te := getMonthRange(tmpTime, true), getMonthRange(tmpTime, false)
		if sTime == tmpTime {
			ts = sTime
		}
		if te == tEndTime {
			te = eTime
		}
		if ts.Before(te) {
			dateSpans = append(dateSpans, entity.DateSpan{
				Key:  fmt.Sprintf("%d-%d", ts.Year(), ts.Month()),
				From: ts.Unix(),
				To:   te.Unix(),
			})
		}
		rTime = rTime.AddDate(0, 1, 0)
		if int(rTime.Month())-int(tmpTime.Month()) > 1 {
			rTime = rTime.AddDate(0, -1, 0)
		}
		tmpTime = rTime
	}
	return
}

// getMonthRange获取月份范围
// isStart true本月月初  false 本月月末(下月月初)
func getMonthRange(t time.Time, isStart bool) time.Time {
	if isStart {
		return time.Date(t.Year(), t.Month(), 1, 0, 0, 0, 0, t.Location())
	}
	return time.Date(t.Year(), t.Month()+1, 1, 0, 0, 0, 0, t.Location())
}

func GetYearData(sTime, eTime time.Time) (_b bool, dateSpans []entity.DateSpan) {
	//整年多取一年进行环比
	if sTime.Month() == 1 && sTime.Day() == 1 && eTime.Month() == 12 && eTime.Day() == 31 {
		_b = true
		sTime = sTime.AddDate(-1, 0, 0)
	}
	dateSpans = getCommonYearStatistics(sTime, eTime)
	return
}

// 年份统计
func getCommonYearStatistics(sTime, eTime time.Time) (dateSpans []entity.DateSpan) {
	tmpTime, rTime, tEndTime := sTime, sTime, getYearRange(eTime, false)
	for rTime.Before(tEndTime) {
		ts, te := getYearRange(tmpTime, true), getYearRange(tmpTime, false)
		if sTime == tmpTime {
			ts = sTime
		}
		if te == tEndTime {
			te = eTime
		}
		if ts.Before(te) {
			dateSpans = append(dateSpans, entity.DateSpan{
				Key:  fmt.Sprintf("%d", ts.Year()),
				From: ts.Unix(),
				To:   te.Unix(),
			})
		}
		rTime = rTime.AddDate(1, 0, 0)
		tmpTime = rTime
	}
	return
}

// getYearRange获取月份范围
// isStart true本月月初  false 本月月末(下月月初)
func getYearRange(t time.Time, isStart bool) time.Time {
	if isStart {
		return time.Date(t.Year(), 1, 1, 0, 0, 0, 0, t.Location())
	}
	return time.Date(t.Year()+1, 1, 1, 0, 0, 0, 0, t.Location())
}

// GetAllKeywordArr  获取所有匹配词
func GetAllKeywordArr(res []entity.KeyWordGroup) (rData []entity.ViewKeyWord) {
	for _, kwg := range res {
		rData = append(rData, getGroupKeywordArr(kwg.A_Key)...)
	}
	return
}

// getGroupKeywordArr 模糊拆分为多个精准匹配
func getGroupKeywordArr(res []entity.ViewKeyWord) (rData []entity.ViewKeyWord) {
	for _, kw := range res {
		if kw.MatchWay == 1 {
			for _, kk := range kw.Keyword {
				rData = append(rData, entity.ViewKeyWord{
					Keyword: []string{kk},
					Exclude: kw.Exclude,
				})
			}
			for _, kk := range kw.Appended {
				rData = append(rData, entity.ViewKeyWord{
					Keyword: []string{kk},
					Exclude: kw.Exclude,
				})
			}
		} else {
			rData = append(rData, kw)
		}
	}
	return
}

func DateIsIn(from, to, n, m int64) bool {
	if m == 0 {
		return to >= n && from <= n
	}
	return to > n && from < n
}