Filter API中对象池的

filter.go

// Filter 注册的filter，用于在server端过滤数据并提供高性能查询
type Filter struct {
	KeyFileds []string
	FilterStr string      // 更新无极数据时用到的，注册的filter条件
	Template  interface{} // 返回数据类型模板
	// objStore 用于Filter后数据的存储
	objStore      map[string]interface{}
	objStoreMutex *sync.RWMutex
	dataVersion   int // 当前内存中无极数据的版本
	stat          *Stat
}

// Get 查询Filter中指定Key的数据
//
// "key" 用于查询数据。例如注册时keyFields为
//   []string{"video_type"}
// 当我们需要查询 video_type 取值为 1 的数据，可以填
//   "video_type=1"
func (filter *Filter) Get(key string) interface{} {
	filter.objStoreMutex.RLock()
	obj, ok := filter.objStore[key]
	filter.objStoreMutex.RUnlock()
	filter.stat.Gets.Add(1)
	if !ok {
		return nil
	}
	filter.stat.Hits.Add(1)
	return obj
}

// 解析filter后无极数据到存储，供 Filter.Get 函数查询
func (filter *Filter) unmarshalToCache(data []map[string]interface{}) {
	filter.objStoreMutex.Lock()
	defer filter.objStoreMutex.Unlock()
	for _, item := range data {
		filter.stat.Reloads.Add(1)
		bs, _ := json.Marshal(item)
		// 还原出json数据后，再按照用户期望的sturct结构进行 Unmarshal
		obj := reflect.New(reflect.TypeOf(filter.Template)).Interface()
		decoder := json.NewDecoder(bytes.NewReader(bs))
		if err := decoder.Decode(obj); err != nil {
			log.Errorf("polling(%s): json.Decode() error: %s", filter.FilterStr, err.Error())
			filter.stat.ReloadErrs.Add(1)
			continue
		}
		key := ""
		hasError := false
		for _, fieldName := range filter.KeyFileds {
			if value, ok := item[fieldName]; !ok {
				log.Errorf("polling(%s): item[%s] not exist", filter.FilterStr, fieldName)
				hasError = true
				filter.stat.ReloadErrs.Add(1)
				break
			} else {
				// 按顺序拼接成查询key
				key += fmt.Sprintf("%s=%v&", fieldName, value)
			}
		}
		if hasError {
			// key找不到，跳过整行数据
			continue
		}
		// 截断末尾的 & 并保存obj供直接查询
		key = key[0 : len(key)-1]
		filter.objStore[key] = obj
	}
}

// AddFilter 注册新的Filter
//
// "keyFields" 用于指定filter查询时的名称和顺序，如
//   []string{"field1", "field2"}
// 则Get时key为
//   field1=%s&field2=%s
// "filterStr" 为过滤无极表的方法，具体写法参见 http://v.oa.com/open-wuji/documents/api-query.html 数据过滤一节
// 参数无需转义，不筛选可以留空
//   video_type=1
// "template" 无极数据的数据结构模板
//
// 注: 必须设置 `EnableFilter=true` 此API才可用
func (client *Client) AddFilter(keyFields []string, filterStr string, template interface{}) (*Filter, error) {
	if !client.opts.EnableFilter {
		return nil, ErrFilterNotEnable
	}
	filter := &Filter{
		KeyFileds:     keyFields,
		FilterStr:     filterStr,
		Template:      template,
		objStore:      make(map[string]interface{}, client.opts.EntriesCapacity),
		objStoreMutex: new(sync.RWMutex),
		stat:          client.stat,
	}
	client.filters[filterStr] = filter
	_ = client.initFromCache(filter) // 从Cache读取数据 (如果有)
	go client.polling(filter)        // 注册后台更新
	return filter, nil
}