fix a bug

zero/lstate/state1/main.go

@@ -61,6 +61,8 @@ func SetLastNum(putter serodb.Putter, num uint64) {
 const delay_block_count = 6
 
 func (self *State1) Parse(last_chose uint64) (chose uint64) {
+	self.MakesureEnv()
+
 	bc := light_ref.Ref_inst.Bc
 	tks := bc.GetTks()
 	next_num := GetLastNum(&self.db)
@@ -157,6 +159,6 @@ func (self *State1) Parse(last_chose uint64) (chose uint64) {
 		log.Info("STATE1 PARSE", "t", chose, "c", next_num-1)
 	}
 
-	return
+	return chose
 
 }

zero/lstate/state1/state1.go

@@ -47,12 +47,31 @@ type State1 struct {
 
 func NewState1(bc lstate.BlockChain) (ret State1) {
 	ret.bc = bc
-	if db, err := leveldb.OpenFile(zconfig.State1_db_dir(), nil); err != nil {
-		panic(err)
+	return
+}
+
+func (self *State1) MakesureEnv() {
+	var need_open_db bool
+	if !zconfig.IsDirExists(zconfig.State1_dir()) {
+		zconfig.Init_State1()
+		need_open_db = true
 	} else {
-		ret.db = SeroDB{db}
+		if !zconfig.IsDirExists(zconfig.State1_db_dir()) {
+			zconfig.Init_State1()
+			need_open_db = true
+		} else {
+			if self.db.db == nil {
+				need_open_db = true
+			}
+		}
+	}
+	if need_open_db {
+		if db, err := leveldb.OpenFile(zconfig.State1_db_dir(), nil); err != nil {
+			panic(err)
+		} else {
+			self.db.db = db
+		}
 	}
-	return
 }
 
 func (self *State1) valid() bool {

zero/zconfig/config.go

@@ -16,6 +16,8 @@
 
 package zconfig
 
+import "io/ioutil"
+
 var is_dev = false
 
 func Init_Dev(dev bool) {
@@ -40,3 +42,16 @@ var VP0 = uint64(788888)
 var MAX_O_INS_LENGTH = 2500
 
 var MAX_TX_OUT_COUNT_LENGTH = 256
+
+func IsDirExists(path string) bool {
+	files, err := ioutil.ReadDir(path)
+	if err != nil {
+		return false
+	} else {
+		if len(files) > 0 {
+			return true
+		} else {
+			return false
+		}
+	}
+}

zero/zconfig/config_test.go

@@ -0,0 +1,14 @@
+package zconfig
+
+import (
+	"fmt"
+	"io/ioutil"
+	"testing"
+)
+
+func TestFile(t *testing.T) {
+
+	files, err := ioutil.ReadDir("/Users")
+	fmt.Print(files, err)
+
+}