Merge pull request #51 from moevm/queries_v2

Импорт и экспорт БД

server/api/data/v1/data.proto

@@ -0,0 +1,35 @@
+syntax = "proto3";
+
+package data.v1;
+
+option go_package = "data/v1;data_v1";
+
+import "google/api/annotations.proto";
+
+service DataAPI {
+    rpc ExportDBV1 (ExportDBV1Request) returns (ExportDBV1Response) {
+        option (google.api.http) = {
+            get: "/api/data"
+        };
+    };
+    rpc ImportDBV1 (ImportDBV1Request) returns (ImportDBV1Response) {
+        option (google.api.http) = {
+            post: "/api/data"
+            body: "*"
+        };
+    };
+}
+    message ExportDBV1Request {
+
+    }
+
+
+    message ExportDBV1Response {
+        string db = 1;
+    }
+
+    message ImportDBV1Request {
+        string db = 1;
+    }
+
+    message ImportDBV1Response {}

server/internal/app/app.go

@@ -12,9 +12,11 @@ import (
 	"google.golang.org/grpc"
 
 	"plants/internal/config"
+	"plants/internal/handlers/data"
 	"plants/internal/handlers/plants"
 	"plants/internal/handlers/trades"
 	"plants/internal/handlers/users"
+	dataAPI "plants/internal/pkg/pb/data/v1"
 	plantsAPI "plants/internal/pkg/pb/plants/v1"
 	tradesAPI "plants/internal/pkg/pb/trades/v1"
 	usersAPI "plants/internal/pkg/pb/users/v1"
@@ -47,6 +49,9 @@ func Run(cfg *config.Config) error {
 	tradesHandler := trades.New(repo)
 	tradesAPI.RegisterTradesServer(gRPCServer, tradesHandler)
 
+	dataHandler := data.New(repo)
+	dataAPI.RegisterDataAPIServer(gRPCServer, dataHandler)
+
 	mux := runtime.NewServeMux()
 	if err = plantsAPI.RegisterPlantsAPIHandlerServer(ctx, mux, plantsHandler); err != nil {
 		return err
@@ -59,6 +64,9 @@ func Run(cfg *config.Config) error {
 	if err = tradesAPI.RegisterTradesHandlerServer(ctx, mux, tradesHandler); err != nil {
 		return err
 	}
+	if err = dataAPI.RegisterDataAPIHandlerServer(ctx, mux, dataHandler); err != nil {
+		return err
+	}
 
 	var group errgroup.Group
 	l, err := net.Listen("tcp", fmt.Sprintf(":%v", cfg.GRPC.Port))

server/internal/handlers/data/export_db_v1.go

@@ -0,0 +1,18 @@
+package data
+
+import (
+	"context"
+
+	api "plants/internal/pkg/pb/data/v1"
+)
+
+func (h *Handler) ExportDBV1(
+	ctx context.Context,
+	req *api.ExportDBV1Request,
+) (*api.ExportDBV1Response, error) {
+	res, err := h.storage.ExportDB(ctx)
+	if err != nil {
+		return &api.ExportDBV1Response{}, err
+	}
+	return &api.ExportDBV1Response{Db: res}, nil
+}

server/internal/handlers/data/import_db_v1.go

@@ -0,0 +1,18 @@
+package data
+
+import (
+	"context"
+
+	api "plants/internal/pkg/pb/data/v1"
+)
+
+func (h *Handler) ImportDBV1(
+	ctx context.Context,
+	req *api.ImportDBV1Request,
+) (*api.ImportDBV1Response, error) {
+	err := h.storage.ImportDB(ctx, req.Db)
+	if err != nil {
+		return &api.ImportDBV1Response{}, err
+	}
+	return &api.ImportDBV1Response{}, nil
+}

server/internal/handlers/data/service.go

@@ -0,0 +1,23 @@
+package data
+
+import (
+	"context"
+
+	v1 "plants/internal/pkg/pb/data/v1"
+)
+
+type Storage interface {
+	ExportDB(context.Context) (string, error)
+	ImportDB(context.Context, string) error
+}
+
+type Handler struct {
+	v1.UnimplementedDataAPIServer
+	storage Storage
+}
+
+func New(storage Storage) *Handler {
+	return &Handler{
+		storage: storage,
+	}
+}