golang 一键生成 form,json,gorm标签

最新推荐文章于 2024-08-12 18:11:14 发布

fwhezfwhez

最新推荐文章于 2024-08-12 18:11:14 发布

阅读量1.2w

点赞数 3

分类专栏： go

本文链接：https://blog.csdn.net/fwhezfwhez/article/details/83622657

版权

go 专栏收录该内容

128 篇文章 10 订阅

订阅专栏

每次有新表产生时，就会有新的model，model编写完后，繁杂的标签注解令人烦躁，所以写了这样的工具。


func AddJsonFormGormTag(in string) (string) {
	var result string
	scanner := bufio.NewScanner(strings.NewReader(in))
	var oldLineTmp = ""
	var lineTmp = ""
	var propertyTmp = ""
	var seperateArr []string
	for scanner.Scan() {
		oldLineTmp = scanner.Text()
		lineTmp = strings.Trim(scanner.Text(), " ")
		if strings.Contains(lineTmp, "{") || strings.Contains(lineTmp, "}") {
			result = result + oldLineTmp + "\n"
			continue
		}
		seperateArr = Split(lineTmp, " ")
		// 接口或者父类声明不参与tag, 自带tag不参与tag
		if len(seperateArr) == 1 || len(seperateArr) == 3 {
			continue
		}
		propertyTmp = HumpToUnderLine(seperateArr[0])
		oldLineTmp = oldLineTmp + fmt.Sprintf("    `gorm:\"column:%s\" json:\"%s\" form:\"%s\"`", propertyTmp, propertyTmp, propertyTmp)
		result = result + oldLineTmp + "\n"
	}
	return result
}


// 增强型split，对  a,,,,,,,b,,c     以","进行切割成[a,b,c]
func Split(s string, sub string) []string {
	var rs = make([]string, 0, 20)
	tmp := ""
	Split2(s, sub, &tmp, &rs)
	return rs
}

// 附属于Split，可独立使用
func Split2(s string, sub string, tmp *string, rs *[]string) {
	s = strings.Trim(s, sub)
	if !strings.Contains(s, sub) {
		*tmp = s
		*rs = append(*rs, *tmp)
		return
	}
	for i, _ := range s {
		if string(s[i]) == sub {
			*tmp = s[:i]
			*rs = append(*rs, *tmp)
			s = s[i+1:]
			Split2(s, sub, tmp, rs)
			return
		}
	}
}

// 驼峰转下划线
func HumpToUnderLine(s string) string{
	if s =="ID" {
		return "id"
	}
	var rs string
	elements:=FindUpperElement(s)
	for _,e :=range elements{
		s =strings.Replace(s, e, "_"+strings.ToLower(e),-1)
	}
	rs = strings.Trim(s," ")
	rs = strings.Trim(rs,"\t")
	return strings.Trim(rs,"_")
}

// 找到字符串中大写字母的列表,附属于HumpToUnderLine
func FindUpperElement(s string) []string {
	var rs = make([]string, 0, 10)
	for i := range s {
		if s[i] >= 65 && s[i] <= 90 {
			rs = append(rs, string(s[i]))
		}
	}
	return rs
}

如何使用

func TestAddJsonFormTag(t *testing.T) {
	rs := AddJsonFormGormTag(`
	type User struct{
		Id        uint64
		NameSpace  string
		DailyRoutine int
	}
	`)
	fmt.Println(rs)
}

结果

	type User struct{
		Id        uint64    `gorm:"column:id" json:"id" form:"id"`
		NameSpace  string    `gorm:"column:name_space" json:"name_space" form:"name_space"`
		DailyRoutine int    `gorm:"column:daily_routine" json:"daily_routine" form:"daily_routine"`
	}

具体对齐，使用go fmt xxx.go 命令
goland 快捷键是 ctrl alt L