Exemplo n.º 1
0
// LoadMapper - Loads a data into the Mapper
func (storage *Record) LoadMapper(query string) (*mapper.Record, error) {

	var err error

	mapp := new(mapper.Record)
	mapp.Fields = make(map[string]mapper.Field)
	baseRecord := make(map[string][]string)
	relatedRecord := make(map[string]map[string][]string)

	// Loads prearranged values
	for index, record := range storage.Mapper.Fields {
		if len(record.Value) != 0 && len(record.Related) == 0 &&
			len(record.RelatedBy) == 0 && len(record.RelatedTo) == 0 {
			mapp.Fields[index] = mapper.Field{
				Key:      record.Key,
				Value:    record.Value,
				Format:   record.Format,
				Multiple: record.Multiple,
				Hide:     record.Hide,
			}
		}
	}

	// Loads base record
	for index, record := range storage.Mapper.Fields {
		// Check for base record
		if len(record.Value) == 0 && len(record.Related) != 0 &&
			(len(record.RelatedBy) == 0 || len(record.RelatedTo) == 0) {
			// if not cached, do it
			if len(baseRecord) == 0 {
				baseRecord, err = storage.CurrentStorage.Search(record.Related, query)
				if err != nil {
					return nil, err
				}
				if len(baseRecord) == 0 {
					return nil, nil
				}
			}
			answer := []string{}

			// collects all fields into answer
			for _, item := range record.Name {
				if result, ok := baseRecord[item]; ok {
					answer = append(answer, result...)
				}
			}

			mapp.Fields[index] = mapper.Field{
				Key:      record.Key,
				Value:    answer,
				Format:   record.Format,
				Multiple: record.Multiple,
				Hide:     record.Hide,
			}
		}
	}

	// Loads related records
	for index, record := range storage.Mapper.Fields {
		// Check for related record
		if len(record.RelatedBy) != 0 && len(record.RelatedTo) != 0 && len(record.Related) != 0 {
			answer := []string{}
			nameToAsk := record.RelatedBy
			queryRelated := strings.Join(baseRecord[record.Related], "")

			// if non-related record from specified type/table
			if len(record.Value) != 0 {
				queryRelated = record.Value[0]
			}

			// if record not cached, do it
			if _, ok := relatedRecord[record.Related]; ok == false {
				if record.Multiple {
					relatedRecord[record.Related], err = storage.CurrentStorage.SearchMultiple(
						record.RelatedTo,
						nameToAsk,
						queryRelated,
					)
				} else {
					relatedRecord[record.Related], err = storage.CurrentStorage.SearchRelated(
						record.RelatedTo,
						nameToAsk,
						queryRelated,
					)
				}
			}
			// collects all fields into answer
			for _, item := range record.Name {
				if result, ok := relatedRecord[record.Related][item]; ok {
					answer = append(answer, result...)
				}
			}
			mapp.Fields[index] = mapper.Field{
				Key:      record.Key,
				Value:    answer,
				Format:   record.Format,
				Multiple: record.Multiple,
				Hide:     record.Hide,
			}
		}
	}

	return mapp, nil
}
Exemplo n.º 2
0
func TestStorage(t *testing.T) {

	conf := config.New()
	flag.Parse()
	mapp := new(mapper.Record)
	storage := New(conf, mapp)
	answer, ok := storage.Search("")
	if ok != false {
		t.Error("Expected ok is false, got", ok)
	}
	if answer != "not found\n" {
		t.Error("Expected answer is not found, got", answer)
	}
	answer, ok = storage.Search("aaa")
	if ok != false {
		t.Error("Expected ok is false, got", ok)
	}
	if answer != "not found\n" {
		t.Error("Expected answer is not found, got", answer)
	}
	mapp.Fields = make(map[string]mapper.Field)
	mapp.Fields["01"] = mapper.Field{
		Key:     "Domain Name: ",
		Name:    []string{"name"},
		Related: "name",
	}
	mapp.Fields["03"] = mapper.Field{
		Key:   "Registrar WHOIS Server: ",
		Value: []string{"whois.markmonitor.com"},
	}
	mapp.Fields["05"] = mapper.Field{
		Key:     "Updated Date: ",
		Name:    []string{"updatedDate"},
		Format:  "{date}",
		Related: "name",
	}
	mapp.Fields["12"] = mapper.Field{
		Key:      "Domain Status: ",
		Name:     []string{"domainStatus"},
		Multiple: true,
		Related:  "name",
	}
	mapp.Fields["13"] = mapper.Field{
		Key:       "Registry Registrant ID: ",
		Name:      []string{"handle"},
		Hide:      true,
		Related:   "ownerHandle",
		RelatedBy: "handle",
		RelatedTo: "customer",
	}
	mapp.Fields["21"] = mapper.Field{
		Key: "Registrant Phone: ",
		Name: []string{
			"phone.countryCode",
			"phone.areaCode",
			"phone.subscriberNumber",
		},
		Format:    "{string}.{string}{string}",
		Related:   "ownerHandle",
		RelatedBy: "handle",
		RelatedTo: "customer",
	}
	mapp.Fields["52"] = mapper.Field{
		Key:       "Name Server: ",
		Name:      []string{"name"},
		Multiple:  true,
		Related:   "nsgroupId",
		RelatedBy: "nsgroupId",
		RelatedTo: "nameserver",
	}
	mapp.Fields["55"] = mapper.Field{
		Key:       "",
		Value:     []string{"1"},
		Name:      []string{"updatedDate"},
		Format:    ">>> Last update of WHOIS database: {date} <<<",
		Related:   "whois",
		RelatedBy: "id",
		RelatedTo: "whois",
	}
	storage = New(conf, mapp)
	answer, ok = storage.Search("google.com")
	if ok != true {
		t.Error("Expected ok is true, got", ok)
	}
	expected := `Domain Name: google.com
Registrar WHOIS Server: whois.markmonitor.com
Updated Date: 2014-05-19T04:00:17Z
Domain Status: clientUpdateProhibited
Domain Status: clientTransferProhibited
Domain Status: clientDeleteProhibited
Registry Registrant ID: 
Registrant Phone: +1.6502530000
Name Server: ns1.google.com
Name Server: ns2.google.com
Name Server: ns3.google.com
Name Server: ns4.google.com
>>> Last update of WHOIS database: 2014-06-01T11:00:07Z <<<
`
	if answer != expected {
		t.Error("Expected answer:\n", expected, "\n, got:\n", answer)
	}
	conf.Storage.StorageType = "mysql"
	storage = New(conf, mapp)
	answer, ok = storage.Search("mmm")
	if ok != false {
		t.Error("Expected ok is false, got", ok)
	}
	if answer != "not found\n" {
		t.Error("Expected answer is not found, got", answer)
	}
	conf.Storage.StorageType = "elasticsearch"
	storage = New(conf, mapp)
	answer, ok = storage.Search("eee")
	if ok != false {
		t.Error("Expected ok is false, got", ok)
	}
	if answer != "not found\n" {
		t.Error("Expected answer is not found, got", answer)
	}
}