go語言遍歷文件夾並解析json文件

package main

import (
    "github.com/buger/jsonparser"
	"fmt"
	"log"
	"os"
	"io"
	"io/ioutil"
	"path"
)

var (
	files []string
)

//獲取指定目錄下的所有文件和目錄
func TraverseDir(dirPth string) ( err error) {
	dir, err := ioutil.ReadDir(dirPth)
	if err != nil {
	return  err
	}
	PthSep := string(os.PathSeparator)
	//suffix = strings.ToUpper(suffix) //忽略後綴匹配的大小寫
	for _, fi := range dir {
		
		if fi.IsDir() { // 忽略目錄
			TraverseDir(dirPth + PthSep + fi.Name())
		}else{
			files = append(files, dirPth+PthSep+fi.Name())
		}
	}
	return  nil
}

//解析json文件並檢測需要字段用於計數
func ParserIWant(filePath string)(ret int64) {
	f, err := os.Open(filePath)
 	if err != nil {
		fmt.Println("File Open Error")
	 }else{
		 fmt.Println("File Open OK")
	 }
	line, err := ioutil.ReadAll(f)
	if err == io.EOF {
		fmt.Println("File Is Commplete")
	} else if err != nil {
		log.Fatal(err)
	}

	result, err := jsonparser.GetInt(line, "NocarInfo", "Tribike", "tribikeCount")
    if err != nil {
        fmt.Println(err)
	}
	if result == 1{
		return result
	}else{
		result, err := jsonparser.GetInt(line, "NocarInfo", "Motobike", "motobikeCount")
		if err != nil {
			fmt.Println(err)
		}
		if result == 1{
			return result
		}else{
			result, err := jsonparser.GetInt(line, "NocarInfo", "Bike", "bikeCount")
			if err != nil {
				fmt.Println(err)
			}
			if result == 1{
				return result
			}
		}
	}
	return 0
}

func main() {
	_ = TraverseDir("E:\\mvcode\\v2bufshool\\")
	var nVehicle int64
	for _, table1 := range files {
		fileSuffix := path.Ext(table1)
		fmt.Println(fileSuffix)
		if fileSuffix==".json"{
			ret := ParserIWant(table1)
			if ret == 1{
				nVehicle++
			}
		}
	}
	fmt.Printf("Has Vehicle:%d", nVehicle)
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章