Skip to content

Commit f06ff8d

Browse files
committed
更新
1 parent fe498e9 commit f06ff8d

File tree

3 files changed

+18
-13
lines changed

3 files changed

+18
-13
lines changed

codemsg/stringer.go

Lines changed: 15 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -464,19 +464,22 @@ func (f *File) genDecl(node ast.Node) bool {
464464
mapName := ""
465465
var err error
466466
v.Name, mapName, err = parseTakeCodeToMap(v.Name)
467-
if err != nil {
468-
panic(err)
467+
if len(mapName) > 0 {
468+
469+
if err != nil {
470+
panic(err)
471+
}
472+
if f.c.SaveCodeToMap == nil {
473+
f.c.SaveCodeToMap = make(map[string]map[int]bool)
474+
}
475+
476+
saveMap := f.c.SaveCodeToMap[mapName]
477+
if saveMap == nil {
478+
saveMap = make(map[int]bool)
479+
}
480+
saveMap[int(v.value)] = true
481+
f.c.SaveCodeToMap[mapName] = saveMap
469482
}
470-
if f.c.SaveCodeToMap == nil {
471-
f.c.SaveCodeToMap = make(map[string]map[int]bool)
472-
}
473-
474-
saveMap := f.c.SaveCodeToMap[mapName]
475-
if saveMap == nil {
476-
saveMap = make(map[int]bool)
477-
}
478-
saveMap[int(v.value)] = true
479-
f.c.SaveCodeToMap[mapName] = saveMap
480483
}
481484

482485
f.values = append(f.values, v)

codemsg/take_to_map.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ func saveTakeFileMap(dir, packageName, types string, mapName string, m map[int]b
6363
}
6464
sourceCode, err := format.Source(buf.Bytes())
6565
if err != nil {
66-
panic(err)
66+
panic(string(buf.Bytes()))
6767
}
6868

6969
err = os.WriteFile(fileName, sourceCode, 0o644)

testdata/err.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,4 +6,6 @@ const (
66
ENo ErrNo = 1003 // 号码出错 @TakeCodeToMap(InfoMap)
77

88
ENotFound ErrNo = 1004 // 找不到 @TakeCodeToMap(InfoMap)
9+
10+
ENotLogin ErrNo = 1005 // 未登录
911
)

0 commit comments

Comments
 (0)