2. 程式人生 > 實用技巧 >我的go練手專案--使用go實現“在檔案中替換”功能

我的go練手專案--使用go實現“在檔案中替換”功能

阿新 • • 發佈：2020-07-26

寫這個專案的需求比較簡單，就是想批量替換一批檔案裡面的關鍵字（實際場景是為了遷移到達夢，需要把php程式碼裡面使用oci的函式名全替換了）

不過由於替換規則比較複雜，現有的文字編輯器不好操作，所以寫了一個工具輔助

package main

import (
	"bytes"
	"fmt"
	"io/ioutil"
	"log"
	"os"
	"regexp"
	"strings"
)

//獲取指定目錄下的所有檔案和目錄
func GetFilesAndDirs(dirPth string) (files []string, dirs []string, err error) {
	dir, err := ioutil.ReadDir(dirPth)
	if err != nil {
		return nil, nil, err
	}

	PthSep := string(os.PathSeparator)
	//suffix = strings.ToUpper(suffix) //忽略字尾匹配的大小寫

	for _, fi := range dir {
		if fi.IsDir() { // 目錄, 遞迴遍歷
			dirs = append(dirs, dirPth+PthSep+fi.Name())
			GetFilesAndDirs(dirPth + PthSep + fi.Name())
		} else {
			// 過濾指定格式
			ok := strings.HasSuffix(fi.Name(), ".go")
			if ok {
				files = append(files, dirPth+PthSep+fi.Name())
			}
		}
	}

	return files, dirs, nil
}


//獲取指定目錄下的所有檔案,包含子目錄下的檔案
func GetAllFiles(dirPth string) (files []string, err error) {
	var dirs []string
	dir, err := ioutil.ReadDir(dirPth)
	if err != nil {
		return nil, err
	}

	PthSep := string(os.PathSeparator)
	//suffix = strings.ToUpper(suffix) //忽略字尾匹配的大小寫

	for _, fi := range dir {
		if fi.IsDir() { // 目錄, 遞迴遍歷
			dirs = append(dirs, dirPth+PthSep+fi.Name())
			GetAllFiles(dirPth + PthSep + fi.Name())
		} else {
			files = append(files, dirPth+PthSep+fi.Name())
			/*
			// 過濾指定格式
			ok := strings.HasSuffix(fi.Name(), ".go")
			if ok {
				files = append(files, dirPth+PthSep+fi.Name())
			}

			 */
		}
	}

	// 讀取子目錄下檔案
	for _, table := range dirs {
		temp, _ := GetAllFiles(table)
		for _, temp1 := range temp {
			files = append(files, temp1)
		}
	}

	return files, nil
}

//獲取目錄裡面的檔名與expr正則匹配的檔案
func GetAllFilesMatch(dirPth string, expr string) (files []string, err error) {
	var reg *regexp.Regexp = nil

	if expr != "" {
		//先編譯正則 提高匹配效率
		reg, err = regexp.Compile(expr)
		if err != nil {
			return nil, err
		}
	}

	return getAllFilesMatchInter(reg, dirPth)
}

//獲取指定目錄下的所有檔案,包含子目錄下的檔案
func getAllFilesMatchInter(reg *regexp.Regexp, dirPth string) (files []string, err error) {
	var dirs []string

	dir, err := ioutil.ReadDir(dirPth)
	if err != nil {
		return nil, err
	}

	PthSep := string(os.PathSeparator)
	//suffix = strings.ToUpper(suffix) //忽略字尾匹配的大小寫

	for _, fi := range dir {
		if fi.IsDir() { // 目錄, 遞迴遍歷
			dirs = append(dirs, dirPth+PthSep+fi.Name())
		} else {
			if (reg != nil) {
				// 過濾指定格式
				ok := reg.MatchString(fi.Name())
				if ok {
					files = append(files, dirPth+PthSep+fi.Name())
				}
			} else {
				files = append(files, dirPth+PthSep+fi.Name())
			}
		}
	}

	// 讀取子目錄下檔案
	for _, table := range dirs {
		temp, err := getAllFilesMatchInter(reg, table)
		if err != nil {
			return nil, err
		}

		for _, temp1 := range temp {
			files = append(files, temp1)
		}
	}

	return files, nil
}

func changeKey(fileName string) {
	buf, _ := ioutil.ReadFile(fileName)
	//log.Printf("%v", string(buf))
	if !bytes.Contains(buf, []byte("OCI")) {
		return
	}

	ok, _ := regexp.Match("OCI[a-zA-Z]", buf)
	if !ok {
		return
	}

	fmt.Printf("修改檔案[%s]\n", fileName)
	buf = bytes.ReplaceAll(buf, []byte("\nOCI"), []byte("\nHT_DPI"))
	buf = bytes.ReplaceAll(buf, []byte("\rOCI"), []byte("\rHT_DPI"))
	/*
	buf = bytes.ReplaceAll(buf, []byte(";OCI"), []byte(";HT_DPI"))
	buf = bytes.ReplaceAll(buf, []byte("\"OCI"), []byte("\"HT_DPI"))
	buf = bytes.ReplaceAll(buf, []byte("/OCI"), []byte("/HT_DPI"))
	buf = bytes.ReplaceAll(buf, []byte("@OCI"), []byte("@HT_DPI"))
	buf = bytes.ReplaceAll(buf, []byte("\nOCI"), []byte("\nHT_DPI"))
	buf = bytes.ReplaceAll(buf, []byte("!OCI"), []byte("!HT_DPI"))
	buf = bytes.ReplaceAll(buf, []byte("(OCI"), []byte("(HT_DPI"))
	buf = bytes.ReplaceAll(buf, []byte("=OCI"), []byte("=HT_DPI"))
	buf = bytes.ReplaceAll(buf, []byte(" OCI"), []byte(" HT_DPI"))
	buf = bytes.ReplaceAll(buf, []byte("\tOCI"), []byte("\tHT_DPI"))
	*/
	ioutil.WriteFile(fileName, buf, os.ModePerm)
}

func changeKey1(fileName string) {
	buf, _ := ioutil.ReadFile(fileName)
	//log.Printf("%v", string(buf))
	if !bytes.Contains(buf, []byte("ocifetchinto")) {
		return
	}

	fmt.Printf("修改檔案[%s]\n", fileName)
	buf = bytes.ReplaceAll(buf, []byte("ocifetchinto"), []byte("HT_DPIFetchInto"))

	ioutil.WriteFile(fileName, buf, os.ModePerm)
}


func changeKey2(fileName string) {
	buf, _ := ioutil.ReadFile(fileName)
	//log.Printf("%v", string(buf))
	if !bytes.Contains(buf, []byte("HT_DPIFetchinto")) {
		return
	}

	fmt.Printf("修改檔案[%s]\n", fileName)
	buf = bytes.ReplaceAll(buf, []byte("HT_DPIFetchinto"), []byte("HT_DPIFetchInto"))

	ioutil.WriteFile(fileName, buf, os.ModePerm)
}



func changeKey3(fileName string) {
	buf, _ := ioutil.ReadFile(fileName)
	//log.Printf("%v", string(buf))
	if !bytes.Contains(buf, []byte("HT_DPIexecute")) {
		return
	}

	fmt.Printf("修改檔案[%s]\n", fileName)
	buf = bytes.ReplaceAll(buf, []byte("HT_DPIexecute"), []byte("HT_DPIExecute"))

	ioutil.WriteFile(fileName, buf, os.ModePerm)
}

/*
ocibindbyname(
ociBindByName(
OCIBindByName(
OCICommit(
ocidefinebyname(
OCIExecute(
ocifetch(
ocifetchstatement(
OCIFreeStatement(
ocilogoff(
OCILogoff(
OCIRollback(
ocisetprefetch(
*/
func changeKey4(fileName string) {
	buf, _ := ioutil.ReadFile(fileName)
	//log.Printf("%v", string(buf))
	if !bytes.Contains(buf, []byte("oci")) {
		return
	}

	fmt.Printf("修改檔案[%s]\n", fileName)
	buf = bytes.ReplaceAll(buf, []byte("ocibindbyname"), []byte("HT_DPIBindByName"))
	buf = bytes.ReplaceAll(buf, []byte("ociBindByName"), []byte("HT_DPIBindByName"))
	buf = bytes.ReplaceAll(buf, []byte("ocidefinebyname"), []byte("HT_DPIDefineByName"))
	buf = bytes.ReplaceAll(buf, []byte("ocifetch"), []byte("HT_DPIFetch"))
	buf = bytes.ReplaceAll(buf, []byte("ocifetchstatement"), []byte("HT_DPIFetchStatement"))
	buf = bytes.ReplaceAll(buf, []byte("ocilogoff"), []byte("HT_DPILogoff"))
	buf = bytes.ReplaceAll(buf, []byte("ocisetprefetch"), []byte("HT_DPISetPreFetch"))

	ioutil.WriteFile(fileName, buf, os.ModePerm)
}


func main() {
	files, dirs, _ := GetFilesAndDirs(`C:\inc_chk`)

	for _, dir := range dirs {
		fmt.Printf("獲取的資料夾為[%s]\n", dir)
	}


	for _, dir := range files {
		fmt.Printf("獲取的檔案為[%s]\n", dir)
	}

	xfiles, err := GetAllFilesMatch(`C:\inc_chk_dpi`, `.*\.php$`)
	if err != nil {
		log.Printf("err = %v", err)
		os.Exit(1)
	}

	for _, file := range xfiles {
		//fmt.Printf("獲取的檔案為[%s]\n", file)
		changeKey(file)
		/*
		changeKey4(file)
		changeKey1(file)
		changeKey2(file)
		changeKey3(file)

		 */
	}
}

相關推薦

我的go練手專案--使用go實現“在檔案中替換”功能

寫這個專案的需求比較簡單，就是想批量替換一批檔案裡面的關鍵字（實際場景是為了遷移到達夢，需要把php程式碼裡面使用oci的函式名全替換了）

我的go練手專案--使用go實現“刪除sql裡面的註釋和字串”功能

專案裡面有一個需求，要對sql進行簡單的語法分析 為了避免sql裡面的字串和註釋對語法分析做干擾，我寫了一個java函式，對sql進行修剪，刪除裡面字串和註釋，用空格代替

我的go練手專案--使用go獲取東方財富網站基金淨值資料

程式碼我放在https://github.com/kingstarer/kingstarer/tree/master/go/getFundValue 這裡簡單介紹一下思路和關鍵程式碼

檔案程式設計練手 [2] —— 編碼實現修改配置檔案的數值

技術標籤：怪獸系統Linux（ubuntu）linuxubuntu 文章轉載請註明出處，加上原文連結，謝謝！https://blog.csdn.net/weixin_46959681/article/details/112784678

Python例項練手專案彙總（附原始碼）

Python學習基礎後，是不是不知道怎麼加固練習？今天分享幾個小遊戲，讓你在遊戲中把知識給鞏固了。

5個趣味Python練手專案

給大家整理幾個具有一定趣味性的專案練練手，一來增加程式設計樂趣，二來能提高你的程式設計技能。

新手練手專案：10行Python程式碼寫一個程式（1）

今天是萬聖節前夕，然後，看著室友看LPL的，戀愛的。我就只能隨便寫寫程式碼，如果你也剛學Python，可以加一下我的交流群，一起交流學習：456926667

Spring Boot專案中實現檔案上傳功能的示例

在實際專案中，檔案上傳是很多專案必不可少的一個功能。那麼在 Spring Boot 專案中又是如何來實現檔案上傳功能的呢？一般來說，上傳的檔案可以儲存到專案根目錄下的某一資料夾中，但這樣做顯然是不太合適的。因此我們

Django入門練手專案——部落格系統

blog/views.py from django.core.paginator import Paginator from django.http import HttpResponse from django.shortcuts import render

【Java練手專案】Java外賣點餐系統

傻瓜式外賣點餐系統（無資料庫） tips： 菜品類(菜品id，菜品名，菜品型別，上架時間，單價，月銷售，總數量)

最適合Java初學者的練手專案！圖書管理系統（資料原始碼分享！）

專案簡介 圖書館管理小專案的主要目的是讓學完JavaSE的同學對之前掌握的知識做一些運用，該專案應用的知識點包括下面內容：

Java專案-王者榮耀專案原始碼分享！java練手專案_java入門

眾所周知，想要學好Java光看視訊或看書是不行的，一定要動手實踐才可以，而且在面試中，面試官也會問你做過些什麼專案？所以在學習Java中，多做專案是很重要的一個步驟！那麼本篇文章就給大家簡單剖析一下Java版本的

【提測平臺】測試平臺開發練手專案原始碼和教程彙總

TPM提測平臺 背景源於多年前的團隊的一個需求，那時候Vue這類前端框架還沒流行起來，如今重新利用最新開箱即用的前後端框架重新實現出來，並配有系列開發分享文章，宗旨是為做想測試開發和或自己想實現個小工具平臺，

spring boot sa-token練手專案

rbac 基於SpringBoot、sa-token 、MyBatis-plus 的許可權後臺管理系統 專案說明 一個練手的專案，具有RBAC的功能許可權系統

使用fileupload元件實現檔案上傳功能

FileUpload檔案上傳 fileUpload是apache的commons元件提供的上傳元件，它最主要的工作就是幫我們解析request.getInpustream()。

python基於FTP實現檔案傳輸相關功能程式碼例項

這篇文章主要介紹了python基於FTP實現檔案傳輸相關功能程式碼例項,文中通過示例程式碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下

Ajax實現檔案上傳功能(Spring MVC)

本文例項為大家分享了Ajax實現檔案上傳的具體程式碼，供大家參考，具體內容如下

java實現檔案重新命名功能

本文例項為大家分享了java實現檔案重新命名的具體程式碼，供大家參考，具體內容如下

node+vue實現檔案上傳功能

本文例項為大家分享了node+vue實現檔案上傳的具體程式碼，供大家參考，具體內容如下

Android 通過騰訊TBS實現檔案預覽功能

1.整合騰訊TBS 使用騰訊TBS來預覽pdf,word,excel,ppt等多種型別的檔案，去 騰訊瀏覽服務官網下載SDK，按照官方文件文件整合SDK。