Method Loop

internal/stats/http_request_stat_manager.go:202–304

Loop 单个循环

()

// Loop 单个循环
func (this *HTTPRequestStatManager) Loop() error {
	select {
	case ipString := <-this.ipChan:
		// serverId@ip@bytes@isAttack
		var pieces = strings.Split(ipString, "@")
		if len(pieces) < 4 {
			return nil
		}
		var serverIdString = pieces[0]
		var ip = pieces[1]

		var result = iplib.LookupIP(ip)
		if result != nil && result.IsOk() {
			this.locker.Lock()
			if result.CountryId() > 0 {
				var key = serverIdString + "@" + types.String(result.CountryId()) + "@" + types.String(result.ProvinceId()) + "@" + types.String(result.CityId())
				stat, ok := this.cityMap[key]
				if !ok {
					// 检查数量
					if this.serverCityCountMap[serverIdString] > 128 { // 限制单个服务的城市数量，防止数量过多
						this.locker.Unlock()
						return nil
					}
					this.serverCityCountMap[serverIdString]++ // 需要放在限制之后，因为使用的是int16

					stat = &StatItem{}
					this.cityMap[key] = stat
				}
				stat.Bytes += types.Int64(pieces[2])
				stat.CountRequests++
				if types.Int8(pieces[3]) == 1 {
					stat.AttackBytes += types.Int64(pieces[2])
					stat.CountAttackRequests++
				}
			}

			if result.ProviderId() > 0 {
				this.providerMap[serverIdString+"@"+types.String(result.ProviderId())]++
			} else if utils.IsLocalIP(ip) { // 局域网IP
				this.providerMap[serverIdString+"@258"]++
			}
			this.locker.Unlock()
		}
	case userAgentString := <-this.userAgentChan:
		var atIndex = strings.Index(userAgentString, "@")
		if atIndex < 0 {
			return nil
		}
		var serverIdString = userAgentString[:atIndex]
		var userAgent = userAgentString[atIndex+1:]

		var result = SharedUserAgentParser.Parse(userAgent)
		var osInfo = result.OS
		if len(osInfo.Name) > 0 {
			dotIndex := strings.Index(osInfo.Version, ".")
			if dotIndex > -1 {
				osInfo.Version = osInfo.Version[:dotIndex]
			}