Golang创建一个HTTP和HTTPS服务器

使用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>&#128151;&#128151;&#128151;&#128151;&#128151;&#128151;&#128151;&#128151;&#128151;</p><p>山高路远坑深,大军纵横驰奔;</p><p>谁敢横刀立马,唯我彭大将军。</p><p>&#128151;&#128151;&#128151;&#128151;&#128151;&#128151;&#128151;&#128151;&#128151;</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