// 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
}
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)
}
}
