Merge pull request #6 from bubble-diff/dev

优化代码

Author: peanutzhen

const.go

@@ -2,8 +2,8 @@ package main
 
 // 参数常量
 const (
-	SnapshotLen         int32 = 65536 * 4
-	SettingsFilePath          = "./settings.json"
+	SnapshotLen      int32 = 65536 * 4
+	SettingsFilePath       = "./settings.json"
 )
 
 // 应用层协议类型
@@ -14,4 +14,4 @@ const (
 
 const (
 	ApiAddRecord = "/record/add"
-)
+)

main.go

@@ -16,7 +16,7 @@ import (
 
 var debugmode bool
 
-func init() {
+func Init() {
 	flag.BoolVar(&debugmode, "debug", false, "Run as debug mode, read settings file to override task configuration if existsed.")
 	flag.Parse()
 
@@ -37,12 +37,14 @@ func init() {
 }
 
 func main() {
+	Init()
+
 	handle, err := pcap.OpenLive(configuration.Device, SnapshotLen, false, pcap.BlockForever)
-	defer handle.Close()
 	if err != nil {
 		logrus.Error(err)
 		logrus.Fatal("Try sudo.")
 	}
+	defer handle.Close()
 
 	// 过滤出当前服务的流量
 	filter := fmt.Sprintf(