mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2025-01-08 12:01:20 +00:00
c68f00045a
* Revert "remove tombstones (#426)"
This reverts commit aa6bab40e8.
* tombstones don't hide storage or account anymore
* auto-format code by prettier (similar to gofmt)
* wow, it works.....
* small simplification, but need make it more clear
* rebase to master
* rebase to master
* rebase to master
* re-run ci
* clean test files
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.RegisterIntermediateHashAPI(root.Group("intermediate-hash"), 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()
|
|
})
|
|
}
|