90 lines
2.5 KiB
Go
90 lines
2.5 KiB
Go
|
package xml
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"io"
|
|||
|
|
"log"
|
|||
|
|
"net/http"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"github.com/ECUST-XX/xml" // 注意：需确保该xml库与示例中使用的一致
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func getxml() ([]byte, error) {
|
|||
|
|
|
|||
|
|
// 读取整个文件内容，返回字节切片和错误
|
|||
|
|
content, err := ReadHTTPFile("http://127.0.0.1:8080/assets/talk_count.xml")
|
|||
|
|
if err != nil {
|
|||
|
|
// 处理错误（文件不存在、权限问题等）
|
|||
|
|
log.Fatalf("无法读取文件: %v", err)
|
|||
|
|
}
|
|||
|
|
return content, nil
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// TalkCount 根节点，对应<talk_count>标签
|
|||
|
|
// 注意：xml.Name需指定命名空间，与XML中的xmlns保持一致
|
|||
|
|
type TalkCount struct {
|
|||
|
|
XMLName xml.Name `xml:"nieo.seer.org.talk-count talk_count"`
|
|||
|
|
Entries []Entry `xml:"entry"` // 包含所有entry节点
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Entry 单个条目配置，对应<entry>标签
|
|||
|
|
type Entry struct {
|
|||
|
|
ID int `xml:"id,attr"` // 条目ID
|
|||
|
|
ItemID int `xml:"item_id,attr"` // 物品ID
|
|||
|
|
ItemMinCount int `xml:"item_min_count,attr"` // 物品最小数量
|
|||
|
|
ItemMaxCount int `xml:"item_max_count,attr"` // 物品最大数量
|
|||
|
|
Desc string `xml:"desc,attr"` // 描述信息
|
|||
|
|
Count int `xml:"count,attr"` // 计数
|
|||
|
|
Policy string `xml:"policy,attr,omitempty"` // 策略（可选，如week）
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 复用示例中的ReadHTTPFile函数（读取远程文件）
|
|||
|
|
func ReadHTTPFile(url string) ([]byte, error) {
|
|||
|
|
client := &http.Client{
|
|||
|
|
Timeout: 30 * time.Second,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
resp, err := client.Get(url)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("请求失败: %w", err)
|
|||
|
|
}
|
|||
|
|
defer resp.Body.Close()
|
|||
|
|
|
|||
|
|
if resp.StatusCode != http.StatusOK {
|
|||
|
|
return nil, fmt.Errorf("请求返回非成功状态码: %d", resp.StatusCode)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
content, err := io.ReadAll(resp.Body)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("读取内容失败: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return content, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取XML内容（可根据实际URL修改）
|
|||
|
|
func getTalkCountXML() ([]byte, error) {
|
|||
|
|
// 注意：此处URL需替换为实际的talk_count.xml文件地址
|
|||
|
|
content, err := ReadHTTPFile("http://127.0.0.1:8080/assets/talk_count.xml")
|
|||
|
|
if err != nil {
|
|||
|
|
log.Fatalf("无法读取文件: %v", err)
|
|||
|
|
}
|
|||
|
|
return content, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 解析XML到结构体
|
|||
|
|
func getTalkCount() TalkCount {
|
|||
|
|
var talkCount TalkCount
|
|||
|
|
content, _ := getTalkCountXML()
|
|||
|
|
// 解析XML时需注意命名空间匹配
|
|||
|
|
if err := xml.Unmarshal(content, &talkCount); err != nil {
|
|||
|
|
log.Fatalf("XML解析失败: %v", err)
|
|||
|
|
}
|
|||
|
|
return talkCount
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 全局配置变量，用于存储解析后的结果
|
|||
|
|
var TalkCountConfig = getTalkCount()
|