在開(kāi)發(fā)一個(gè)運(yùn)維管理系統(tǒng)時(shí),一個(gè)模塊中要實(shí)現(xiàn)一個(gè)批量處理并刪除某個(gè)目錄下的每一個(gè)文件的功能。
想法是用遞歸算法遍歷目錄下的子目錄和每一個(gè)文件,在處理每一個(gè)文件的時(shí)候單開(kāi)協(xié)程來(lái)并發(fā)處理。
自己寫了一個(gè)demo,但發(fā)現(xiàn)每次都還沒(méi)處理完目錄下的文件就開(kāi)始刪目錄而導(dǎo)致死鎖。
因?yàn)閯h除目錄的代碼也是在遞歸函數(shù)里面的,所以我不知道該什么時(shí)候從通道里讀數(shù)據(jù)出來(lái)。
刪除并發(fā)的幾行代碼可以單線程正常工作實(shí)現(xiàn)功能,就是不知道怎么搞多并發(fā),大家給個(gè)建議來(lái)討論下唄。
使用filepath.Glob
+ 帶緩沖的channel:
package main
import (
"fmt"
"path/filepath"
"sync"
)
func main() {
ls, err := filepath.Glob("*") // 獲取頂層文件/文件夾
CheckErr(err)
wg := new(sync.WaitGroup)
ch := make(chan string, 5)
go DeleteFile(ch, wg)
for _, v := range ls {
wg.Add(1)
ch <- v
}
close(ch)
wg.Wait()
}
func DeleteFile(ch chan string, wg *sync.WaitGroup) {
for v := range ch {
fmt.Println("to remove : ", v)
wg.Done()
}
}
func CheckErr(err error) {
if err != nil {
panic(err)
}
}
package segment
import (
"errors"
"fmt"
"io/ioutil"
"os"
"path/filepath"
"sync"
)
func DeleteDir(filedir string) error {
var w sync.WaitGroup
filelist, err := ioutil.ReadDir(filedir)
if err != nil {
return err
}
for _, value := range filelist {
if value.IsDir() {
w.Add(1)
go func(name string) {
defer w.Done()
DeleteDir(filepath.Join(filedir, name))
}(value.Name())
} else {
w.Add(1)
go func(name string) {
defer w.Done()
DeleteFile(filepath.Join(filedir, name))
}(value.Name())
}
}
w.Wait()
err = os.Remove(filedir)
if err != nil {
fmt.Printf("remove file error : %s \n", err.Error())
}
return err
}
func DeleteDirAll(filedir string) error {
if IsDir(filedir) {
err := DeleteDir(filedir)
if err != nil {
fmt.Printf("delete dir failed : %s \n", err.Error())
return err
}
return nil
} else {
return DeleteFile(filedir)
}
}
func DeleteFile(filename string) error {
if IsExist(filename) {
return os.Remove(filename)
}
return errors.New("file not exist")
}
func IsDir(filename string) bool {
fi, err := os.Stat(filename)
if err != nil {
return false
}
if fi == nil {
return false
}
if fi.IsDir() {
return true
}
return false
}
func IsExist(filename string) bool {
fi, err := os.Stat(filename)
if err != nil {
fmt.Printf("file path not exist : %s \n", err.Error())
return false
}
if fi != nil {
return true
}
return false
}
北大青鳥(niǎo)APTECH成立于1999年。依托北京大學(xué)優(yōu)質(zhì)雄厚的教育資源和背景,秉承“教育改變生活”的發(fā)展理念,致力于培養(yǎng)中國(guó)IT技能型緊缺人才,是大數(shù)據(jù)專業(yè)的國(guó)家
北大青鳥(niǎo)中博軟件學(xué)院創(chuàng)立于2003年,作為華東區(qū)著名互聯(lián)網(wǎng)學(xué)院和江蘇省首批服務(wù)外包人才培訓(xùn)基地,中博成功培育了近30000名軟件工程師走向高薪崗位,合作企業(yè)超4
中公教育集團(tuán)創(chuàng)建于1999年,經(jīng)過(guò)二十年潛心發(fā)展,已由一家北大畢業(yè)生自主創(chuàng)業(yè)的信息技術(shù)與教育服務(wù)機(jī)構(gòu),發(fā)展為教育服務(wù)業(yè)的綜合性企業(yè)集團(tuán),成為集合面授教學(xué)培訓(xùn)、網(wǎng)
達(dá)內(nèi)教育集團(tuán)成立于2002年,是一家由留學(xué)海歸創(chuàng)辦的高端職業(yè)教育培訓(xùn)機(jī)構(gòu),是中國(guó)一站式人才培養(yǎng)平臺(tái)、一站式人才輸送平臺(tái)。2014年4月3日在美國(guó)成功上市,融資1
曾工作于聯(lián)想擔(dān)任系統(tǒng)開(kāi)發(fā)工程師,曾在博彥科技股份有限公司擔(dān)任項(xiàng)目經(jīng)理從事移動(dòng)互聯(lián)網(wǎng)管理及研發(fā)工作,曾創(chuàng)辦藍(lán)懿科技有限責(zé)任公司從事總經(jīng)理職務(wù)負(fù)責(zé)iOS教學(xué)及管理工作。
浪潮集團(tuán)項(xiàng)目經(jīng)理。精通Java與.NET 技術(shù), 熟練的跨平臺(tái)面向?qū)ο箝_(kāi)發(fā)經(jīng)驗(yàn),技術(shù)功底深厚。 授課風(fēng)格 授課風(fēng)格清新自然、條理清晰、主次分明、重點(diǎn)難點(diǎn)突出、引人入勝。
精通HTML5和CSS3;Javascript及主流js庫(kù),具有快速界面開(kāi)發(fā)的能力,對(duì)瀏覽器兼容性、前端性能優(yōu)化等有深入理解。精通網(wǎng)頁(yè)制作和網(wǎng)頁(yè)游戲開(kāi)發(fā)。
具有10 年的Java 企業(yè)應(yīng)用開(kāi)發(fā)經(jīng)驗(yàn)。曾經(jīng)歷任德國(guó)Software AG 技術(shù)顧問(wèn),美國(guó)Dachieve 系統(tǒng)架構(gòu)師,美國(guó)AngelEngineers Inc. 系統(tǒng)架構(gòu)師。