func Patrol(rw http.ResponseWriter, req *http.Request) {
notify := false
now := time.Now()
req.ParseForm()
if req.Form.Get("devmod") == "1" {
notify = true
} else {
s := req.Form.Get("lasttime")
lasttime, err := strconv.Atoi(s)
if err != nil || lasttime == 0 {
if now.After(NotifyStartTime) && now.Before(NotifyEndTime) {
notify = true
}
} else {
t := time.Unix(int64(lasttime), 0)
if t.Before(NotifyStartTime) && now.After(NotifyStartTime) && now.Before(NotifyEndTime) {
notify = true
}
}
}
result := []string{
"bkimage=",
BACKGROUND_IMAGE,
"&url=",
ClickURL,
"&lasttime=",
strconv.Itoa(int(now.Unix())),
"&clsbtimage1=",
CLOSE_IMAGE1,
"&clsbtimage2=",
CLOSE_IMAGE2,
"&clsbtx=",
CLOSE_BUTTON_X,
"&clsbty=",
CLOSE_BUTTON_Y,
"&nextmin=",
strconv.Itoa(NextMinute),
}
if notify {
result = append(result, "¬ify")
}
rw.Write([]byte(strings.Join(result, "")))
remote_addr := req.Header.Get("X-Forwarded-For")
if remote_addr == "" {
remote_addr = req.Header.Get("X-Real-IP")
}
if remote_addr == "" {
remote_addr = req.RemoteAddr
}
logging.Info("[%s] %s?%s %v\n", remote_addr, req.URL.Path, req.URL.RawQuery, notify)
}
func main() {
var host = flag.String("host", "", "Server listen host, default 0.0.0.0")
var port = flag.Int("port", 80, "Server listen port, default 80")
var url = flag.String("url", "", "Click open url")
var nst = flag.String("nst", "", "Notify start time, format: YYYY-mm-dd HH:MM")
var nvt = flag.String("nvt", "", "Notify over time, format as start time")
var cpu = flag.Int("cpu", 0, "Go runtime max process")
var logpath = flag.String("logpath", "./", "Log path")
var logfile = flag.String("logfile", "", "Log filename")
var nextmin = flag.Int("nextmin", NextMinute, "Next request duration(minute)")
flag.Parse()
if *logfile != "" {
if logger, err := logging.NewHourRotationLogger(*logfile, *logpath, LOGFILE_SUFFIX); err != nil {
panic(err)
} else {
logging.SetDefaultLogger(logger)
logging.SetPrefix("BWService")
}
}
if *cpu > 0 {
runtime.GOMAXPROCS(*cpu)
}
loc, err := time.LoadLocation("Asia/Shanghai")
if err != nil {
panic(err)
}
if t, err := time.ParseInLocation("2006-01-02 15:04", *nst, loc); err != nil {
panic(err)
} else {
NotifyStartTime = t
}
if t, err := time.ParseInLocation("2006-01-02 15:04", *nvt, loc); err != nil {
panic(err)
} else {
NotifyEndTime = t
}
if *url == "" {
panic(*url)
} else {
ClickURL = *url
}
if *nextmin > 0 {
NextMinute = *nextmin
}
var addr = net.JoinHostPort(*host, strconv.Itoa(*port))
http.HandleFunc("/patrol", Patrol)
logging.Info("[%s] (%s) - (%s) => %s, %d\n", addr, NotifyStartTime.String(), NotifyEndTime.String(), ClickURL, NextMinute)
if err := http.ListenAndServe(addr, nil); err != nil {
logging.Error("%s", err.Error())
}
}
func (self *Entry) Info(format string, v ...interface{}) {
logging.Info(self.formatHead(format), v...)
}