Google Cloud Datastore Export Handler

export.go

package datastorebackup

import (
	"fmt"
	"net/http"
	"time"

	"google.golang.org/appengine"
	"google.golang.org/appengine/log"
	"google.golang.org/appengine/user"

	datastore "google.golang.org/api/datastore/v1beta1"

	"golang.org/x/net/context"
	"golang.org/x/oauth2/google"
)

func Export(c context.Context, w http.ResponseWriter, r *http.Request) {
	for k, v := range r.Header {
		log.Infof(c, "%s = %s", k, v)
	}

	if r.Header.Get("X-Appengine-Cron") != "true" {
		if user.IsAdmin(c) == false {
			l, err := user.LoginURL(c, r.URL.Path)
			if err != nil {
				http.Error(w, err.Error(), http.StatusInternalServerError)
				return
			}
			http.Redirect(w, r, l, http.StatusFound)
			return
		}
	}

	ctxWithDeadline, cancel := context.WithTimeout(c, 9*time.Minute)
	defer cancel()
	client, err := google.DefaultClient(ctxWithDeadline, datastore.DatastoreScope)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}

	service, err := datastore.New(client)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}

	p := appengine.AppID(c)
	op, err := service.Projects.Export(p, &datastore.GoogleDatastoreAdminV1beta1ExportEntitiesRequest{
		EntityFilter: &datastore.GoogleDatastoreAdminV1beta1EntityFilter{
			NamespaceIds: []string{},
      Kinds:        []string{"hoge","fuga"},
		},
		OutputUrlPrefix: fmt.Sprintf("gs://%s-datastore-backup", p),
	}).Do()
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}

	w.WriteHeader(op.HTTPStatusCode)
	b, err := op.Response.MarshalJSON()
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}
	w.Write(b)
}

This is really useful. Thank you.

Thank you