go并发之美·B操作伴随A操作(B和A同步)

迈不过友情╰ 2022-12-11 13:51 68阅读 0赞

时维九月，序属三秋。潦水尽而寒潭清，烟光凝而暮山紫。地势极而南溟深，天柱高而北辰远。

> 人的思想是无限的，想法是很多的，我们每天都能碰到很多事情，也能想到很多事情。平时习惯于在桌面创建临时文件，接下来又会常常修改，大多数最后下来就成了必备的、常查看的文件，而桌面上的东西又经常会删除，备用性文件留在桌面上不太安全。今天想到了这个情景，结合在本地修改后能同步到云端能备份新文件等类似的情景，这才有了这篇博文，于此，想写点代码趁机实现下我的想法。有了想法就准备行动，当自己能驾驭代码完成一件事、实现自己的想法时，你会发现世界如此美妙！

**go并发之美·干货好文合集：**

[go并发之美·redis篇·接口幂等性设计实现][go_redis]

[go并发之美·redis篇·实现实时排行榜][go_redis 1]

[go并发日记·多goroutine并发安全的操作共享变量][go_goroutine]

[go并发日记·常用并发场景中的并发控制思路与方案][go]

[go并发日记·避免goroutine泄漏/实现协程可控][go_goroutine 1]

[go并发日记·深入理解go语言channel与goroutine阻塞][go_go_channel_goroutine]

[go并发日记·基于channel优雅的进行并发控制][go_channel]

[go并发日记·WaitGroup并发控制探索及实际场景运用][go_WaitGroup]

[channel死锁的情况分析][channel]

[go并发之美·B操作伴随A操作(B和A同步)][go_B_A_B_A]

[go多函数并发通用工具的设计与实现][go 1]

talk is cheep, show you my code

/*
    以文件操作为场景。
    A目录：D:\go_files\fileA
    B目录：D:\go_files\fileB
    背景：A目录有什么文件，B目录也要有，B目录时刻与A进行同步，但是B目录下额外创建的文件不会影响A目录。
    往大了说，可以写一个工具，作为两个存储空间之间的同步工具，A目录可以是本地，B目录可以是另一个地方。
    */
    
    const pathSrc = "D:\\go_files\\dirA"
    const pathDist = "D:\\go_files\\dirB"
    
    func main() {
    	fileCh := make(chan *WillCreate)
    	go WriteBDir(pathDist, fileCh)
    	ticker := time.NewTicker(time.Second * 1)
    	//i := 0
    	for {
    		select {
    		case <-ticker.C:
    			//i++
    			getAllFile(pathSrc, fileCh) // 每秒检测一次所有文件，判断其变动
    			//fmt.Println("第", i, "次检测完成。")
    		}
    	}
    }
    
    type WillCreate struct {
    	IsCreate bool   // 本次需要新创建  还是更新源文件
    	FileName string
    	Content  []byte
    }
    
    // 监控B目录
    func WriteBDir(path string, fileCh chan *WillCreate) {
    	for {
    		select {
    		case obj := <-fileCh:
    			if nil == obj {
    				continue
    			}
    			fmt.Println("从fileCh中获取到：", obj.FileName)
    			// 先得判断文件夹是否在B目录下存在
    			willCreateFileName := pathDist + "\\" + obj.FileName
    			isExist, err := IsExist(willCreateFileName)
    			if err != nil {
    				fmt.Println("[WriteBDir]识别", willCreateFileName, "失败。")
    				continue
    			}
    			if !isExist && obj.IsCreate { // 不存在&需要创建
    				dstFile, err := os.Create(willCreateFileName)
    				if err != nil {
    					fmt.Println("[WriteBDir]", obj.FileName, "文件创建失败,err:", err.Error())
    					dstFile.Close()
    					continue
    				}
    
    				_, err = dstFile.WriteString(string(obj.Content))
    				if err != nil {
    					fmt.Println("[WriteBDir]给文件"+willCreateFileName+"写入内容失败：", err.Error())
    					dstFile.Close()
    					continue
    				}
    				fmt.Println("已在"+pathDist+"下创建了新文件：", obj.FileName)
    				dstFile.Close()
    				continue
    			}
    			if isExist && !obj.IsCreate { // 存在&需要更新
    				file, err := os.OpenFile(willCreateFileName, os.O_WRONLY|os.O_TRUNC, 0600)
    				if err != nil {
    					fmt.Println("打开文件"+willCreateFileName+"失败：", err.Error())
    					file.Close()
    					continue
    				}
    				_, err = file.WriteString(string(obj.Content))
    				if err != nil {
    					fmt.Println("[WriteBDir]更新文件"+willCreateFileName+"内容失败：", err.Error())
    					file.Close()
    					continue
    				}
    				fmt.Println("已在"+pathDist+"下更新了文件：", obj.FileName)
    				file.Close()
    			}
    		default:
    
    		}
    	}
    }
    
    func getAllFile(path string, createCh chan *WillCreate) {
    	files, err := ioutil.ReadDir(path)
    	if err != nil {
    		fmt.Println("[getAllFile]读取目录出错：", err.Error())
    		return
    	}
    
    	for _, f := range files {
    		// 判断该文件在B目录是否存在
    		willCreateFileName := pathDist + "\\" + f.Name()
    		isExist, err := IsExist(willCreateFileName)
    		if err != nil {
    			fmt.Println("[getAllFile]识别", willCreateFileName, "失败。")
    			continue
    		}
    
    		aFileName := pathSrc + "\\" + f.Name()
    		bs, err := ioutil.ReadFile(aFileName)
    		if err != nil {
    			fmt.Println("[getAllFile]读取"+willCreateFileName+"出错：", err.Error())
    			continue
    		}
    
    		if !isExist { // 如果B目录下不存在该文件，则创建
    			obj := &WillCreate{FileName: f.Name(), Content: bs, IsCreate: true}
    			fmt.Println("往fileCh写入了文件：", obj.FileName)
    			createCh <- obj
    			continue
    		}
    
    		// 如果文件存在，则判断变动
    		nowTime := time.Now().Unix()
    		fileLastModTime := f.ModTime().Unix()
    		if (nowTime - 1) < fileLastModTime {  // 没想到更好的判断最新变动的办法，先拿时间判断替代
    			obj := &WillCreate{FileName: f.Name(), Content: bs, IsCreate: false}
    			fmt.Println("检测到文件", obj.FileName, "刚刚被修改,已向fileCh写入。")
    			createCh <- obj
    		}
    	}
    }
    
    // 判断文件是否存在
    func IsExist(filePath string) (exist bool, err error) {
    	_, err = os.Stat(filePath)
    	if os.IsNotExist(err) { // 文件不存在
    		exist = false
    		err = nil
    		return
    	} else if err != nil {
    		return
    	}
    	exist = true
    	return
    }

A目录下随便创建几个文件，B下一开始没有，来 speed running：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0hZWlhfOTk4Nw_size_16_color_FFFFFF_t_70][]

不断修改A目录下某文件，打开B中该文件看看是不是已经更新了：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0hZWlhfOTk4Nw_size_16_color_FFFFFF_t_70 1][]

code is cheep , thoughts are precious.

plain  
Exciting  
Wonderful  
Unstoppable

随风随浪飘荡

随着一生里的浪

你我在重叠的那一刻

顷刻各在一方...

[go_redis]: https://lan6193.blog.csdn.net/article/details/108938119
[go_redis 1]: https://lan6193.blog.csdn.net/article/details/108931188
[go_goroutine]: https://lan6193.blog.csdn.net/article/details/105715373
[go]: https://lan6193.blog.csdn.net/article/details/108670428
[go_goroutine 1]: https://lan6193.blog.csdn.net/article/details/105559568
[go_go_channel_goroutine]: https://lan6193.blog.csdn.net/article/details/97528721
[go_channel]: https://lan6193.blog.csdn.net/article/details/101201408
[go_WaitGroup]: https://lan6193.blog.csdn.net/article/details/100887891
[channel]: https://lan6193.blog.csdn.net/article/details/97656420
[go_B_A_B_A]: https://lan6193.blog.csdn.net/article/details/108901339
[go 1]: https://lan6193.blog.csdn.net/article/details/103298105
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0hZWlhfOTk4Nw_size_16_color_FFFFFF_t_70]: /images/20221123/54707a26dd8d478cafd3a97e01b7435a.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0hZWlhfOTk4Nw_size_16_color_FFFFFF_t_70 1]: /images/20221123/7eb925aca0054b9f9b624ad5c36df83c.png