add show pos for word

dictionary.go

@@ -11,6 +11,7 @@ import (
 type Dictionary struct {
 	total, logTotal float64
 	freqMap         map[string]float64
+	posMap          map[string]string
 	sync.RWMutex
 }
 
@@ -43,6 +44,8 @@ func (d *Dictionary) addToken(token dictionary.Token) {
 			d.freqMap[frag] = 0.0
 		}
 	}
+
+	d.posMap[token.Text()] = token.Pos()
 }
 
 func (d *Dictionary) updateLogTotal() {
@@ -57,6 +60,13 @@ func (d *Dictionary) Frequency(key string) (float64, bool) {
 	return freq, ok
 }
 
+func (d *Dictionary) Pos(key string) (string, bool) {
+	d.RLock()
+	pos, ok := d.posMap[key]
+	d.RUnlock()
+	return pos, ok
+}
+
 func (d *Dictionary) loadDictionary(fileName string) error {
 	return dictionary.LoadDictionary(d, fileName)
 }

jieba.go

@@ -29,6 +29,11 @@ func (seg *Segmenter) Frequency(word string) (float64, bool) {
 	return seg.dict.Frequency(word)
 }
 
+// Pos returns a word's part_of_speech and existence
+func (seg *Segmenter) Pos(word string) (string, bool) {
+	return seg.dict.Pos(word)
+}
+
 // AddWord adds a new word with frequency to dictionary
 func (seg *Segmenter) AddWord(word string, frequency float64) {
 	seg.dict.AddToken(dictionary.NewToken(word, frequency, ""))
@@ -94,7 +99,7 @@ func (seg *Segmenter) SuggestFrequency(words ...string) float64 {
 // LoadDictionary loads dictionary from given file name. Everytime
 // LoadDictionary is called, previously loaded dictionary will be cleard.
 func (seg *Segmenter) LoadDictionary(fileName string) error {
-	seg.dict = &Dictionary{freqMap: make(map[string]float64)}
+	seg.dict = &Dictionary{freqMap: make(map[string]float64), posMap: make(map[string]string)}
 	return seg.dict.loadDictionary(fileName)
 }