Method Init

internal/iplibrary/manager_ip_list.go:115–166

()

}

func (this *IPListManager) Init() {
	// 从数据库中当中读取数据
	// 检查sqlite文件是否存在，以便决定使用sqlite还是kv
	var sqlitePath = Tea.Root + "/data/ip_list.db"
	_, sqliteErr := os.Stat(sqlitePath)

	var db IPListDB
	var err error
	if sqliteErr == nil || !teaconst.EnableKVCacheStore {
		db, err = NewSQLiteIPList()
	} else {
		db, err = NewKVIPList()
	}

	if err != nil {
		remotelogs.Error("IP_LIST_MANAGER", "create ip list local database failed: "+err.Error())
	} else {
		this.db = db

		// 删除本地数据库中过期的条目
		_ = db.DeleteExpiredItems()

		// 本地数据库中最大版本号
		this.lastVersion, err = db.ReadMaxVersion()
		if err != nil {
			remotelogs.Error("IP_LIST_MANAGER", "find max version failed: "+err.Error())
			this.lastVersion = 0
		}
		remotelogs.Println("IP_LIST_MANAGER", "starting from '"+db.Name()+"' version '"+types.String(this.lastVersion)+"' ...")

		// 从本地数据库中加载
		var offset int64 = 0
		var size int64 = 2_000

		var tr = trackers.Begin("IP_LIST_MANAGER:load")
		defer tr.End()

		for {
			items, goNext, readErr := db.ReadItems(offset, size)
			var l = len(items)
			if readErr != nil {
				remotelogs.Error("IP_LIST_MANAGER", "read ip list from local database failed: "+readErr.Error())
			} else {
				this.processItems(items, false)
				if !goNext {
					break
				}
			}
			offset += int64(l)
		}
	}
}

func (this *IPListManager) Loop() error {
	// 是否同步IP名单