mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2025-01-09 04:21:20 +00:00
9324d83cb2
* - handle cursor.Prefix on server - move state reports to KV interface * add CmdCursorSeekKey * tests for abstract_kv * avoid reading configs of databases * avoid reading configs of databases * make linter happy * make linter happy * cleanup * port badger features from original implementation * try to fix test * try to fix test * .Close() don't return error anymore - defer friendly * try to enable badger now * try to enable badger now * badger can't run on CI yet * badger can't run on CI yet * re-run ci * skip ctx cancelation for badger
86 lines
2.0 KiB
Go
86 lines
2.0 KiB
Go
package rest
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/ledgerwatch/turbo-geth/cmd/restapi/apis"
|
|
"github.com/ledgerwatch/turbo-geth/ethdb"
|
|
)
|
|
|
|
func printError(name string, err error) {
|
|
if err != nil {
|
|
log.Printf("%v: SUCCESS", name)
|
|
} else {
|
|
log.Printf("%v: FAIL (err=%v)", name, err)
|
|
}
|
|
}
|
|
|
|
func ServeREST(ctx context.Context, localAddress, remoteDBAddress string) error {
|
|
r := gin.Default()
|
|
root := r.Group("api/v1")
|
|
allowCORS(root)
|
|
root.Use(func(c *gin.Context) {
|
|
c.Next()
|
|
if len(c.Errors) > 0 {
|
|
c.AbortWithStatusJSON(http.StatusInternalServerError, c.Errors)
|
|
}
|
|
})
|
|
|
|
db, err := ethdb.NewRemote().Path(remoteDBAddress).Open(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer db.Close()
|
|
e := &apis.Env{
|
|
DB: db,
|
|
RemoteDBAddress: remoteDBAddress,
|
|
}
|
|
|
|
if err = apis.RegisterRemoteDBAPI(root.Group("remote-db"), e); err != nil {
|
|
return err
|
|
}
|
|
if err = apis.RegisterAccountAPI(root.Group("accounts"), e); err != nil {
|
|
return err
|
|
}
|
|
if err = apis.RegisterStorageAPI(root.Group("storage"), e); err != nil {
|
|
return err
|
|
}
|
|
if err = apis.RegisterStorageTombstonesAPI(root.Group("storage-tombstones"), e); err != nil {
|
|
return err
|
|
}
|
|
|
|
log.Printf("serving on %v... press ctrl+C to abort\n", localAddress)
|
|
|
|
srv := &http.Server{Addr: localAddress, Handler: r}
|
|
|
|
// Initializing the server in a goroutine so that
|
|
// it won't block the graceful shutdown handling below
|
|
go func() {
|
|
<-ctx.Done()
|
|
shutdownCtx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
if err := srv.Shutdown(shutdownCtx); err != nil {
|
|
log.Fatal("Server forced to shutdown:", err)
|
|
}
|
|
}()
|
|
|
|
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
|
log.Fatalf("listen: %s\n", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func allowCORS(r *gin.RouterGroup) {
|
|
r.Use(func(c *gin.Context) {
|
|
c.Header("Access-Control-Allow-Origin", "*")
|
|
c.Header("Access-Control-Allow-Headers", "Content-Type")
|
|
c.Header("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS")
|
|
c.Next()
|
|
})
|
|
}
|