使用Go语言同时创建一个HTTP服务器和一个HTTPS服务器。
package main
import (
"bufio"
"encoding/json"
"fmt"
"io/ioutil"
"log"
"net/http"
"os"
"strings"
"sync"
"time"
"github.com/360EntSecGroup-Skylar/excelize"
)
//安装excel库
// PS C:\Users\TopFish\Desktop\https模拟器> go mod tidy
// PS C:\Users\TopFish\Desktop\https模拟器> go mod init github.com/tealeg/xlsx
// PS C:\Users\TopFish\Desktop\https模拟器> go get github.com/360EntSecGroup-Skylar/excelize
type SuccessResponse struct {
Name string `json:"name"`
Age int64 `json:"age"`
Type int64 `json:"type"`
Msg string `json:"msg"`
}
var lock sync.RWMutex
var flag int = 0
var fileName string = "./httpLog.txt"
func main() {
os.Remove(fileName)
num := len(os.Args)
if num > 1 {
// fmt.Println(os.Args[1])
switch os.Args[1] {
case "-D", "-d":
flag = 1
fmt.Println("调试模式")
case "-F", "-f":
flag = 2
fmt.Println("日志信息输入到文件:", fileName)
case "-E", "-e":
flag = 3
fmt.Println("日志信息输入到Excel文件:", fileName)
excelInit()
case "-h", "--help":
fmt.Println("Usage: ")
fmt.Println(" ./main.exe -f|-F :日志输出到文件")
fmt.Println(" ./main.exe -d|-D :日志输出到终端")
fmt.Println(" ./main.exe -h|--help :显示用法")
return
}
} else {
fmt.Println("默认模式:无任何日志信息")
}
http.HandleFunc("/", handler)
http.HandleFunc("/book/tcpip", tcpip) // 默认处理函数
http.HandleFunc("/book/datastruct", datastruct) // 数据结构
http.HandleFunc("/book/linuxKernel", linuxKernel) // Linux内核
go httpServer()
go httpsServer()
hearBeat()
}
func httpServer() {
fmt.Println("HTTP服务已开启,监听端口:8080")
err := http.ListenAndServe(":8080", nil)
if err != nil {
fmt.Println(err)
}
}
func httpsServer() {
fmt.Println("HTTPS服务已开启,监听端口:8443")
err := http.ListenAndServeTLS(":8443", "./cert/server.crt", "./cert/server.key", nil)
if err != nil {
fmt.Println(err)
}
}
var timer *time.Timer
func hearBeat() { //心跳检测
timer = time.NewTimer(time.Second * 120)
for {
select {
case <-timer.C:
debugLog(fmt.Sprintf("[心跳报文(%s)]\n", time.Now().Format("2006-01-02 15:04:05")))
}
timer.Reset(time.Second * 120)
}
}
func handler(rw http.ResponseWriter, r *http.Request) {
_, _ = rw.Write([]byte("<h1><marquee><font color=\"#0000ff\"><p>💗💗💗💗💗💗💗💗💗</p><p>山高路远坑深,大军纵横驰奔;</p><p>谁敢横刀立马,唯我彭大将军。</p><p>💗💗💗💗💗💗💗💗💗</p></font></marquee></h1>"))
}
func debugLog(msg string) {
if flag == 1 {
fmt.Printf("%s\n", msg)
} else if flag == 2 {
writeLog(msg)
fmt.Print("->")
} else if flag == 3 {
writeExcelLog(msg)
fmt.Print("->")
} else {
fmt.Print("->")
}
}
// tcpip
func tcpip(rw http.ResponseWriter, r *http.Request) {
body, ok := ioutil.ReadAll(r.Body)
if ok != nil {
fmt.Println("格式有误")
}
defer r.Body.Close()
debugLog(fmt.Sprintf("[TCIPIP(%s)]: URL=%s, body=%s\n", time.Now().Format("2006-01-02 15:04:05"), r.URL.String(), body))
Success(rw)
}
// 数据结构
func datastruct(rw http.ResponseWriter, r *http.Request) {
body, ok := ioutil.ReadAll(r.Body)
if ok != nil {
fmt.Println("格式有误")
}
defer r.Body.Close()
debugLog(fmt.Sprintf("[数据结构(%s)]: URL=%s, %s\n", time.Now().Format("2006-01-02 15:04:05"), r.URL.String(), body))
Success(rw)
}
func CheckFileExist(fileName string) bool {
_, err := os.Stat(fileName)
if os.IsNotExist(err) {
return false
}
return true
}
func linuxKernel(rw http.ResponseWriter, r *http.Request) {
useragent := r.Header.Get("User-Agent")
debugLog(fmt.Sprintf("[Linux内核(%s)]: URL=%s, 设备ID=%s 已注册\n", time.Now().Format("2006-01-02 15:04:05"), r.URL.String(), useragent))
Success(rw)
}
func writeLog(msg string) {
lock.Lock()
defer lock.Unlock()
fileHandle, err := os.OpenFile(fileName, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)
if err != nil {
log.Println("open file error :", err)
return
}
defer fileHandle.Close()
// NewWriter 默认缓冲区大小是 4096
// 需要使用自定义缓冲区的writer 使用 NewWriterSize()方法
buf := bufio.NewWriterSize(fileHandle, len(msg))
buf.WriteString(msg)
err = buf.Flush()
if err != nil {
log.Println("flush error :", err)
}
}
var xlsxName string = "日志记录.xlsx"
var logType = [...]string{
"tcpip",
"datastruct",
"linux-kernel",
}
func excelInit() {
var xlsx *excelize.File
os.Remove(xlsxName)
xlsx = excelize.NewFile()
for _, v := range logType {
xlsx.NewSheet(v)
}
xlsx.DeleteSheet("Sheet1")
//保存文件
err := xlsx.SaveAs(xlsxName)
if err != nil {
log.Fatal(err)
}
fmt.Println("创建日志文件:", xlsxName)
}
func writeExcelLog(msg string) {
var xlsx *excelize.File
xlsx, err := excelize.OpenFile(xlsxName)
if err != nil {
fmt.Println(err)
if strings.Contains(err.Error(), "cannot find the file specified") {
fmt.Println("创建日志文件:")
xlsx = excelize.NewFile()
}
}
for index, name := range xlsx.GetSheetMap() {
fmt.Println(index, name)
}
//保存文件到指定路径
err = xlsx.SaveAs(xlsxName)
if err != nil {
log.Fatal(err)
}
}
func Success(rw http.ResponseWriter) {
var success = new(SuccessResponse)
success.Name = "张三"
success.Age = 18
refund, _ := json.Marshal(success)
_, _ = rw.Write(refund)
}
func Failure(rw http.ResponseWriter) {
var success = new(SuccessResponse)
success.Name = "李四"
success.Age = 18
refund, _ := json.Marshal(success)
_, _ = rw.Write(refund)
}
cert.zip
