func (t *TiedotEngine) AddIndex(collection string, path Path) {
c := t.tiedot.Use(collection)
tdPath := strings.Join(path, tiedot.INDEX_PATH_SEP)
if _, ok := c.SecIndexes[tdPath]; ok {
log.Infof("Index on path:%v already exists for collection:%s", tdPath, collection)
return
}
log.V(3).Infof("Adding index on path:%v to collection:%s", tdPath, collection)
err := c.Index(path)
log.FatalIfErr(err, "Failure creating index on collection:"+collection)
}
func newShort(w http.ResponseWriter, r *http.Request, tde *kv.TiedotEngine) {
defer r.Body.Close()
raw, err := ioutil.ReadAll(r.Body)
log.FatalIfErr(err, "Failure reading request err:")
var v M
err = json.Unmarshal(raw, &v)
log.FatalIfErr(err, "Failure decoding JSON json:"+string(raw)+" err:")
if dest, ok := v["url"]; ok {
count := incrCount(tde)
shortSlug := base62.EncodeInt(count)
parsed, err := url.Parse(dest.(string))
if err != nil {
log.Warning("Malformed URL:" + dest.(string) + " err:" + err.Error())
}
if parsed.Scheme == "" {
dest = "http://" + dest.(string)
}
s := Shortened{
Original: dest.(string),
Short: count,
}
err = saveShortened(s, tde)
log.FatalIfErr(err, "Failure saving URL err:")
out, _ := json.Marshal(map[string]interface{}{
"Short": s.Short,
"Original": s.Original,
"Full": *base + shortSlug,
"HitCount": s.HitCount,
})
w.Write(out)
} else {
log.Info("No url field included in JSON")
w.WriteHeader(http.StatusBadRequest)
w.Write([]byte("Require 'url' field in request JSON"))
}
}
// Create a new LevelDBEngine with the given file and options
func NewTiedotEngine(directory string, collections []string, dropPref DropPreference) *TiedotEngine {
db, err := tiedot.OpenDB(directory)
log.FatalIfErr(err, "Failure opening tiedot basedir err:")
for _, c := range collections {
if _, ok := db.StrCol[c]; ok {
log.V(4).Info("Collection %s already exists")
if dropPref == DropIfExist {
log.Info("Dropping collection %s due to dropIfExist option")
err = db.Drop(c)
log.FatalIfErr(err, "Failure dropping collection with name:%s err:", c)
err = db.Create(c, 1) // partition DB for use by up to 1 goroutines at a time
log.FatalIfErr(err, "Failure creating collection with name:%s err:", c)
}
} else {
log.V(4).Info("Creating collection %s")
err = db.Create(c, 1) // partition DB for use by up to 1 goroutines at a time
log.FatalIfErr(err, "Failure creating collection with name:%s err:", c)
}
}
tde := &TiedotEngine{
tiedot: db,
}
return tde
}
