From a7f89c033ab5b53943165fca9a9557b7b2c3dd76 Mon Sep 17 00:00:00 2001
From: SisMaker <1713699517@qq.com>
Date: Fri, 30 Oct 2020 22:39:42 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=B9=E5=8A=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 fileSync.go | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/fileSync.go b/fileSync.go
index 29b95ac..02b153d 100644
--- a/fileSync.go
+++ b/fileSync.go
@@ -23,6 +23,7 @@ var Conn net.Conn
 const (
 	hrl = ".hrl"
 	erl = ".erl"
+	beam = ".beam"
 	dtl = ".dtl"
 	lfe = "lfe"
 	ex  = "ex"
@@ -37,7 +38,7 @@ func CollectFile(File string) {
 	ext := filepath.Ext(File)
 	fmt.Println("IMY****************收集数据 : ", File)
 	fmt.Println("IMY****************收集数据ext : ", ext)
-	if ext != "" && (ext == erl || ext == hrl || ext == ex || ext == dtl || ext == lfe) {
+	if ext != "" && (ext == erl || ext == beam || ext == hrl || ext == ex || ext == dtl || ext == lfe) {
 		CollectFiles[File] = struct{}{}
 		SendTimer.Reset(time.Second * SendDur)
 		fmt.Println("IMY****************收集数据成功: ", File)
@@ -46,11 +47,10 @@ func CollectFile(File string) {
 	}
 }
 
-// 发送文件到erl层
+// 发送文件列表到erl层
 func SendToErl() {
 	fmt.Println("IMY****************发送数据到tcp : ", CollectFiles)
 
-	// IMY-todo 通过tcp 发送数据
 	// 拼写数据
 	var buffer bytes.Buffer
 	for k := range CollectFiles {
@@ -65,6 +65,7 @@ func SendToErl() {
 	_ = binary.Write(msg, binary.BigEndian, length)
 	//写入消息体
 	_ = binary.Write(msg, binary.BigEndian, buffer.Bytes())
+	fmt.Println("IMY****************发送数据到sock : ", msg)
 	Conn.Write(msg.Bytes())
 
 	SendTimer.Reset(time.Second * SleepDur)
@@ -174,7 +175,8 @@ func main() {
 	defer SendTimer.Stop()
 
 	Addr := "localhost:" + os.Args[2]
-	Conn, err := net.Dial("tcp", Addr)
+	var err error
+	Conn, err = net.Dial("tcp", Addr)
 	if err != nil {
 		fmt.Println("IMY****************建立tcp失败 : ", Addr)
 		return
@@ -188,4 +190,5 @@ func main() {
 
 	_, _ = Conn.Read(data)
 	Conn.Close()
+	fmt.Println("IMY****************建立tcp关闭了 : ", os.Args[0])
 }