run.go:
package main
import (
"./uas"
"flag"
"math/rand"
"net/url"
"os"
"strconv"
"strings"
"time"
)
type resource struct {
url string
target string
start int
end int
}
func ruleResource() []resource{
var res []resource
//首页
r1 := resource{
url: "https://localhost/",
target: "",
start: 0,
end: 0,
}
//列表页
r2 := resource{
url: "https://localhost/list/{$id}.html",
target: "{$id}",
start: 1,
end: 21,
}
//详情页
r3 := resource{
url: "https://localhost/movie/{$id}.html",
target: "{$id}",
start: 1,
end: 12924,
}
res = append(res, r1, r2, r3)
return res
}
func buildUrl(res []resource) []string {
var list []string
for _, resItem := range res {
if len(resItem.target) == 0 {
list = append(list, resItem.url)
}else {
for i := resItem.start; i <= resItem.end; i++ {
urlStr := strings.Replace(resItem.url, resItem.target, strconv.Itoa(i), -1)
list = append(list, urlStr)
}
}
}
return list
}
func makeLog(current, refer, ua string) string {
u := url.Values{}
u.Set("time", "1")
u.Set("url", current)
u.Set("refer", refer)
u.Set("ua", ua)
paramsStr := u.Encode()
logTemplate := "127.0.0.1 - - [14/Apr/2019:20:07:42 +0800] \"OPTIONS /dig?{$paramsStr} HTTP/1.1\" 499 0 \"-\" \"{$ua}\" \"-\""
log := strings.Replace(logTemplate, "{$paramsStr}", paramsStr, -1)
log = strings.Replace(log, "{$ua}", ua, -1)
return log
}
//随机
func randInt(min, max int) int {
r := rand.New(rand.NewSource(time.Now().UnixNano()))
if min > max {
return max
}
return r.Intn(max-min) + min
}
func main() {
total := flag.Int("total", 100, "创建多少行日志")
filePath := flag.String("filePath", "F:/phpStudy/PHPTutorial/nginx/logs/access.log", "日志文件路径")
//想生效还要调用解析方法
flag.Parse()
//fmt.Println(*total, *filePath)
//构造出真实的网站url集合
res := ruleResource()
list := buildUrl(res)
//按照要求,生成$total行日志内容
logStr := ""
for i := 1; i <= *total; i++ {
//随机选
currentUrl := list[randInt(0, len(list)-1)]
referUrl := list[randInt(0, len(list)-1)]
ua := uas.UaList[randInt(0, len(uas.UaList)-1)]
logStr = logStr + makeLog(currentUrl, referUrl, ua) + "\n"
//和随机种子一样的间隔,尽可能保证随机性
time.Sleep(time.Nanosecond)
}
//放到循环外一次写入
fd, _ := os.OpenFile(*filePath, os.O_RDWR|os.O_APPEND, 0644 )
fd.Write([]byte(logStr))
fd.Close()
}package main
import (
"./uas"