invidx.go

package main

type InvIdx struct {
	inv_word_map map[string]map[string]uint
	href_map     map[string]struct {
		title      string
		word_count uint
	}
}

func NewInvIdx() *InvIdx {
	return &InvIdx{
		make(map[string]map[string]uint),
		make(map[string]struct {
			title      string
			word_count uint
		}),
	}
}

func (ii *InvIdx) GetAllDocs() (map[string]struct{}, error) {
	docs := make(map[string]struct{})
	for url := range ii.href_map {
		docs[url] = struct{}{}
	}
	return docs, nil
}

func (ii *InvIdx) GetAllFreq(doc string) (map[string]uint, error) {
	return ii.inv_word_map[doc], nil
}

func (ii *InvIdx) GetDocsWithWord(word string) (map[string]struct {
	title      string
	word_count uint
	freq       uint
}, error) {
	doc_map := make(map[string]struct {
		title      string
		word_count uint
		freq       uint
	})
	for url, word_map := range ii.inv_word_map {
		if freq, ok := word_map[word]; ok {
			doc_map[url] = struct {
				title      string
				word_count uint
				freq       uint
			}{ii.href_map[url].title, ii.href_map[url].word_count, freq}
		}
	}
	return doc_map, nil
}

func (ii *InvIdx) AddDocWithWords(title string, url string, wordcount uint, words map[string]uint) error {
	ii.href_map[url] = struct {
		title      string
		word_count uint
	}{title, wordcount}
	ii.inv_word_map[url] = words
	return nil
}

func (ii *InvIdx) ContainsDoc(doc string) (bool, error) {
	_, ok := ii.href_map[doc]
	return ok, nil
}

func (ii *InvIdx) Size() (uint, error) {
	return uint(len(ii.href_map)), nil
}

func (ii *InvIdx) DeleteDataBase() {
	ii.inv_word_map = nil
	ii.href_map = nil
}