92 lines
1.7 KiB
Go
92 lines
1.7 KiB
Go
|
package drugs
|
||
|
|
||
|
import (
|
||
|
"encoding/json"
|
||
|
"github.com/sahilm/fuzzy"
|
||
|
"io/ioutil"
|
||
|
"strings"
|
||
|
)
|
||
|
|
||
|
type DrugInteraction struct {
|
||
|
ID string
|
||
|
}
|
||
|
|
||
|
type PathWay struct {
|
||
|
ID string
|
||
|
Drugs []string
|
||
|
}
|
||
|
|
||
|
type Salt struct {
|
||
|
ID string
|
||
|
Name string
|
||
|
}
|
||
|
|
||
|
type DrugDatabase struct {
|
||
|
drugs []DrugInfo
|
||
|
index map[string]int
|
||
|
keys []string
|
||
|
}
|
||
|
|
||
|
type Classification struct {
|
||
|
Description string
|
||
|
DirectParent string
|
||
|
Kingdom string
|
||
|
SuperClass string
|
||
|
Class string
|
||
|
SubClass string
|
||
|
}
|
||
|
|
||
|
type DrugInfo struct {
|
||
|
ID string
|
||
|
Name string
|
||
|
DrugType string
|
||
|
HalfLife string
|
||
|
Description string
|
||
|
Classification Classification
|
||
|
Indication string
|
||
|
Toxicity string
|
||
|
MechanismOfAction string
|
||
|
Pharmacodynamics string
|
||
|
Absorption string
|
||
|
Interactions []DrugInteraction
|
||
|
Salts []Salt
|
||
|
Groups map[string]bool
|
||
|
Pathways []PathWay
|
||
|
}
|
||
|
|
||
|
func (dd * DrugDatabase) generateIndex() {
|
||
|
dd.index = make(map[string]int)
|
||
|
for i,drug := range dd.drugs {
|
||
|
dd.index[strings.ToLower(drug.ID)] = i
|
||
|
dd.index[strings.ToLower(drug.Name)] = i
|
||
|
dd.keys = append(dd.keys, strings.ToLower(drug.ID))
|
||
|
dd.keys = append(dd.keys, strings.ToLower(drug.Name))
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (dd * DrugDatabase) Lookup(query string) (DrugInfo) {
|
||
|
index,ok := dd.index[strings.ToLower(query)]
|
||
|
if ok {
|
||
|
return dd.drugs[index]
|
||
|
}
|
||
|
matches := fuzzy.Find(query, dd.keys)
|
||
|
for _,match := range matches {
|
||
|
return dd.drugs[dd.index[match.Str]]
|
||
|
}
|
||
|
return DrugInfo{}
|
||
|
}
|
||
|
|
||
|
func LoadDatabaseJSON(path string) DrugDatabase {
|
||
|
data,err := ioutil.ReadFile(path)
|
||
|
if err == nil {
|
||
|
var db []DrugInfo
|
||
|
err := json.Unmarshal(data, &db)
|
||
|
if err == nil {
|
||
|
var ddb DrugDatabase
|
||
|
ddb.drugs = db
|
||
|
ddb.generateIndex()
|
||
|
return ddb
|
||
|
}
|
||
|
}
|
||
|
return DrugDatabase{}
|
||
|
}
|