Refactor geo resources

This commit is contained in:
世界
2022-07-05 13:23:47 +08:00
parent 8392567962
commit 2d9203ee74
12 changed files with 273 additions and 198 deletions

View File

@@ -14,10 +14,10 @@ type Reader struct {
domainLength map[string]int
}
func Open(path string) (*Reader, error) {
func Open(path string) (*Reader, []string, error) {
content, err := os.Open(path)
if err != nil {
return nil, err
return nil, nil, err
}
reader := &Reader{
reader: content,
@@ -25,9 +25,13 @@ func Open(path string) (*Reader, error) {
err = reader.readMetadata()
if err != nil {
content.Close()
return nil, err
return nil, nil, err
}
return reader, nil
codes := make([]string, 0, len(reader.domainIndex))
for code := range reader.domainIndex {
codes = append(codes, code)
}
return reader, codes, nil
}
func (r *Reader) readMetadata() error {