make it compile

lib/netdb/std.go

@@ -1,10 +1,10 @@
 package netdb
 
 import (
-  "github.com/golang/glog"
-  "github.com/majestrate/goi2p/lib/common"
+  log "github.com/golang/glog"
+  "github.com/bounce-chat/go-i2p/lib/common"
+  "io"
   "os"
-  "time"
 )
 
 // standard network database implementation
@@ -28,9 +28,67 @@ func (db StdNetDB) Exists() bool {
   return common.FileExists(db.Path())
 }
 
+func (db StdNetDB) SaveEntry(e *Entry) (err error) {
+  var f io.WriteCloser 
+  f, err = os.OpenFile(e.FilePath(db), os.O_WRONLY, 0600)
+  if err == nil {
+    err = e.WriteTo(f)
+    if err != nil {
+      log.Errorf("failed to write netdb entry: %s", err.Error())
+    }
+    f.Close()
+  } else {
+    log.Errorf("failed to save netdb entry: %s", err.Error())
+  }
+  return
+}
+
+
+// reseed if we have less than minRouters known routers
+// returns error if reseed failed
+func (db StdNetDB) Reseed(minRouters int) (err error) {
+  current := db.KnownPeerCount()
+  if current <= minRouters {
+    // we need to reseed
+    rs := GetRandomReseed()
+    log.Infof("Reseeding from %s", rs)
+    chnl := make(chan *Entry)
+    // receive entries from reseed
+    go func(c chan *Entry) {
+      count := 0
+      for {
+        e, ok := <- c
+        if ok {
+          // got an entry
+          // save it to our netdb
+          err := db.SaveEntry(e)
+          if err == nil {
+            count ++
+          }
+        }
+      }
+    }(chnl) // call
+    err = rs.Reseed(chnl)
+  }
+  return
+}
+
+// ensure that the network database exists and is seeded with a minimum number of routers
+func (db StdNetDB) Ensure(minRouters int) (err error) {
+  if ! db.Exists() {
+    err = db.Create()
+  }
+  if err == nil {
+    // database directory ensured
+    // try to reseed
+    err = db.Reseed(minRouters)
+  }
+  return
+}
+
 // create base network database directory
 func (db StdNetDB) Create() (err error) {
-  glog.Infof("Create network database in %s", db.Path())
+  log.Infof("Create network database in %s", db.Path())
   err = os.Mkdir(db.Path(), 0600)
   return
 }