add upgrade option when blockchain is start

Author: kingwang

blockchain/restore.go

@@ -20,10 +20,25 @@ func (chain *BlockChain) Upgrade() {
 	chain.UpgradeChain()
 	chainlog.Info("storedb upgrade start")
 	chain.UpgradeStore()
+	chainlog.Info("upgrade all dapp")
+	chain.upgradePlugin()
 	chainlog.Info("chain reduce start")
 	chain.ReduceChain()
 }
 
+func (chain *BlockChain) upgradePlugin() {
+	client := chain.client
+	msg := client.NewMessage("execs", types.EventUpgrade, nil)
+	err := client.Send(msg, true)
+	if err != nil {
+		panic(err)
+	}
+	_, err = client.Wait(msg)
+	if err != nil {
+		panic(err)
+	}
+}
+
 //UpgradeStore 升级storedb
 func (chain *BlockChain) UpgradeStore() {
 	meta, err := chain.blockStore.GetStoreUpgradeMeta()

executor/executor.go

@@ -120,11 +120,64 @@ func (exec *Executor) SetQueueClient(qcli queue.Client) {
 				go exec.procExecCheckTx(msg)
 			} else if msg.Ty == types.EventBlockChainQuery {
 				go exec.procExecQuery(msg)
+			} else if msg.Ty == types.EventUpgrade {
+				//执行升级过程中不允许执行其他的事件，这个事件直接不采用异步执行
+				exec.procUpgrade(msg)
 			}
 		}
 	}()
 }
 
+func (exec *Executor) procUpgrade(msg *queue.Message) {
+	for _, plugin := range pluginmgr.GetExecList() {
+		elog.Info("begin upgrade plugin ", "name", plugin)
+		err := exec.upgradePlugin(msg, plugin)
+		if err != nil {
+			panic(err)
+		}
+	}
+}
+
+func (exec *Executor) upgradePlugin(msg *queue.Message, plugin string) error {
+	header, err := exec.qclient.GetLastHeader()
+	if err != nil {
+		msg.Reply(exec.client.NewMessage("", types.EventUpgrade, err))
+		return err
+	}
+	driver, err := drivers.LoadDriverWithClient(exec.qclient, plugin, header.GetHeight())
+	if err == types.ErrUnknowDriver { //已经注册插件，但是没有启动
+		elog.Info("upgrade ignore ", "name", plugin)
+		return nil
+	}
+	if err != nil {
+		msg.Reply(exec.client.NewMessage("", types.EventUpgrade, err))
+		return err
+	}
+	var localdb dbm.KVDB
+	if !exec.disableLocal {
+		localdb = NewLocalDB(exec.client)
+		defer localdb.(*LocalDB).Close()
+		driver.SetLocalDB(localdb)
+	}
+	//目前升级不允许访问statedb
+	driver.SetStateDB(nil)
+	driver.SetAPI(exec.qclient)
+	driver.SetExecutorAPI(exec.qclient, exec.grpccli)
+	driver.SetEnv(header.GetHeight(), header.GetBlockTime(), uint64(header.GetDifficulty()))
+	localdb.Begin()
+	err = driver.Upgrade()
+	if err != nil {
+		localdb.Rollback()
+		msg.Reply(exec.client.NewMessage("", types.EventUpgrade, err))
+		return err
+	}
+	localdb.Commit()
+	msg.Reply(exec.client.NewMessage("", types.EventUpgrade, &types.Reply{
+		IsOk: true,
+	}))
+	return nil
+}
+
 func (exec *Executor) procExecQuery(msg *queue.Message) {
 	//panic 处理
 	defer func() {

pluginmgr/manager.go

@@ -26,6 +26,14 @@ func InitExec(cfg *typ.Chain33Config) {
 	})
 }
 
+//GetExecList 获取插件名字列表
+func GetExecList() (datas []string) {
+	for _, plugin := range pluginItems {
+		datas = append(datas, plugin.GetExecutorName())
+	}
+	return
+}
+
 // InitWallet init wallet plugin
 func InitWallet(wallet wcom.WalletOperate, sub map[string][]byte) {
 	once.Do(func() {

system/dapp/driver.go

@@ -73,6 +73,7 @@ type Driver interface {
 	GetExecutorType() types.ExecutorType
 	CheckReceiptExecOk() bool
 	ExecutorOrder() int64
+	Upgrade() error
 }
 
 // DriverBase defines driverbase type
@@ -97,6 +98,12 @@ type DriverBase struct {
 	ety                  types.ExecutorType
 }
 
+//Upgrade default upgrade only print a message
+func (d *DriverBase) Upgrade() error {
+	blog.Info("upgrade ", "dapp", d.GetName())
+	return nil
+}
+
 // GetPayloadValue define get payload func
 func (d *DriverBase) GetPayloadValue() types.Message {
 	if d.ety == nil {

types/event.go

@@ -156,6 +156,7 @@ const (
 	//exec
 	EventBlockChainQuery = 212
 	EventConsensusQuery  = 213
+	EventUpgrade         = 214
 
 	// BlockChain 接收的事件
 	EventGetLastBlockMainSequence   = 300
@@ -336,4 +337,5 @@ var eventName = map[int]string{
 	EventReplyHeightByTitle:         "EventReplyHeightByTitle",
 	EventGetParaTxByTitleAndHeight:  "EventGetParaTxByTitleAndHeight",
 	EventCmpBestBlock:               "EventCmpBestBlock",
+	EventUpgrade:                    "EventUpgrade",
 }