lo

package main

import (
	"fmt"

	"github.com/samber/lo" // 确保安装了 "github.com/samber/lo" 包
)

// 定义 FundCalendar 结构体
type FundCalendar struct {
	Symbol   string    `json:"symbol"`
	Calendar []Holiday `json:"calendar"`
}

// 定义 FundCalendars 类型
type FundCalendars []*FundCalendar

// 定义 Holiday 结构体
type Holiday struct {
	Date string `json:"date"`
	Name string `json:"name"`
}

// FundCalendars 类型的 ToSymbolMap 方法
func (f FundCalendars) ToSymbolMap() map[string]*FundCalendar {
	return lo.SliceToMap(f, func(item *FundCalendar) (string, *FundCalendar) {
		return item.Symbol, item
	})
}

func main() {
	// 创建一些 Holiday 数据
	holidays1 := []Holiday{
		{Date: "2024-01-01", Name: "New Year's Day"},
		{Date: "2024-12-25", Name: "Christmas Day"},
	}

	holidays2 := []Holiday{
		{Date: "2024-07-04", Name: "Independence Day"},
		{Date: "2024-11-28", Name: "Thanksgiving Day"},
	}

	// 创建 FundCalendar 实例
	fund1 := &FundCalendar{Symbol: "FUND1", Calendar: holidays1}
	fund2 := &FundCalendar{Symbol: "FUND2", Calendar: holidays2}

	// 创建 FundCalendars 切片
	fundCalendars := FundCalendars{fund1, fund2}

	// 转换为 Symbol 映射
	symbolMap := fundCalendars.ToSymbolMap()

	// 打印映射结果
	for symbol, fundCalendar := range symbolMap {
		fmt.Printf("Symbol: %s, Calendar: %+v\n", symbol, fundCalendar.Calendar)
	}
}