mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2025-01-03 17:44:29 +00:00
43 lines
831 B
Go
43 lines
831 B
Go
package apis
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/ledgerwatch/turbo-geth/ethdb"
|
|
)
|
|
|
|
func RegisterRemoteDBAPI(router *gin.RouterGroup, e *Env) error {
|
|
router.GET("/", e.GetDB)
|
|
router.POST("/", e.PostDB)
|
|
return nil
|
|
}
|
|
|
|
func (e *Env) GetDB(c *gin.Context) {
|
|
var host, port string
|
|
|
|
split := strings.Split(e.RemoteDBAddress, ":")
|
|
if len(split) == 2 {
|
|
host, port = split[0], split[1]
|
|
}
|
|
c.JSON(http.StatusOK, map[string]string{"host": host, "port": port})
|
|
}
|
|
|
|
func (e *Env) PostDB(c *gin.Context) {
|
|
newAddr := c.Query("host") + ":" + c.Query("port")
|
|
kv, err := ethdb.NewRemote().Path(newAddr).Open()
|
|
if err != nil {
|
|
c.Error(err) //nolint:errcheck
|
|
return
|
|
}
|
|
e.RemoteDBAddress = newAddr
|
|
|
|
e.KV.Close()
|
|
|
|
e.KV = kv
|
|
db := ethdb.NewObjectDatabase(kv)
|
|
e.DB = db
|
|
c.Status(http.StatusOK)
|
|
}
|