stackexchange-dnscontrol/models/recorddb.go

package models

// Functions that make it easier to deal with a group of records.

type RecordDB struct {
	labelAndTypeMap map[RecordKey]struct{}
}

// NewRecordDBFromRecords creates a RecordDB from a list of RecordConfig.
func NewRecordDBFromRecords(recs Records, zone string) *RecordDB {
	result := &RecordDB{}

	//fmt.Printf("DEBUG: BUILDING RecordDB: zone=%v\n", zone)
	result.labelAndTypeMap = make(map[RecordKey]struct{}, len(recs))
	for _, rec := range recs {
		//fmt.Printf("    DEBUG: Adding %+v\n", rec.Key())
		result.labelAndTypeMap[rec.Key()] = struct{}{}
	}
	//fmt.Printf("DEBUG: BUILDING RecordDB: DONE!\n")

	return result
}

// ContainsLT returns true if recdb contains rec. Matching is done
// on the record's label and type (i.e. the RecordKey)
func (recdb *RecordDB) ContainsLT(rec *RecordConfig) bool {
	_, ok := recdb.labelAndTypeMap[rec.Key()]
	//fmt.Printf("DEBUG: ContainsLT(%q) = %v (%v)\n", rec.Key(), ok, recdb)
	return ok
}