feat: grpc 関連の api を実装

api/members.go

@@ -165,3 +165,22 @@ func queryToColumnName(query string) util.SortKey {
 		return util.MemberID
 	}
 }
+
+func (server *Server) downloadMembers(w http.ResponseWriter, r *http.Request) {
+	ctx := r.Context()
+	span, ctx := tracer.StartSpanFromContext(ctx, "api.downloadMembers")
+	defer span.Finish()
+
+	w.Header().Set("Content-Type", "applicaiton/zip")
+	// ファイル名はここで指定する。
+	w.Header().Set("Content-Disposition", "attachment; filename=saka-members.zip")
+
+	err := server.grpcClient.DownloadMembers(ctx, w)
+	if err != nil {
+		server.logger.Warnf(ctx, "failed do download members: %v", err)
+		w.WriteHeader(http.StatusBadRequest)
+		return
+	}
+
+	w.WriteHeader(http.StatusOK)
+}

api/other.go

@@ -9,7 +9,6 @@ import (
 )
 
 func (server *Server) Health(w http.ResponseWriter, r *http.Request) {
-
 	ctx := r.Context()
 	span, ctx := tracer.StartSpanFromContext(ctx, "api.Health")
 	defer span.Finish()
@@ -23,3 +22,21 @@ func (server *Server) Health(w http.ResponseWriter, r *http.Request) {
 	fmt.Fprint(w, string(jsonRes))
 	server.logger.Debug(ctx, "Health checked")
 }
+
+func (server *Server) healthGrpc(w http.ResponseWriter, r *http.Request) {
+	ctx := r.Context()
+	span, ctx := tracer.StartSpanFromContext(ctx, "api.healthGrpc")
+	defer span.Finish()
+
+	msg, err := server.grpcClient.Health(ctx)
+	if err != nil {
+		server.logger.Warn(ctx, "failed do health check")
+
+		w.Write([]byte(err.Error()))
+		w.WriteHeader(http.StatusInternalServerError)
+		return
+	}
+
+	w.Write([]byte(msg))
+	w.WriteHeader(http.StatusOK)
+}

api/server.go

@@ -10,26 +10,32 @@ import (
 	muxtrace "gopkg.in/DataDog/dd-trace-go.v1/contrib/gorilla/mux"
 
 	"github.com/android-project-46group/api-server/db"
+	"github.com/android-project-46group/api-server/repository/grpc"
 	"github.com/android-project-46group/api-server/util"
 )
 
 // Server serves HTTP requests for this service.
 type Server struct {
-	config  util.Config
-	querier db.Querier
-	router  *muxtrace.Router
-	matcher language.Matcher
-	logger  util.Logger
+	config     util.Config
+	querier    db.Querier
+	router     *muxtrace.Router
+	matcher    language.Matcher
+	logger     util.Logger
+	grpcClient grpc.GrpcClient
 }
 
 // Create a new server from the given config file.
-func NewServer(config util.Config, querier db.Querier, matcher language.Matcher, logger util.Logger) (*Server, error) {
+func NewServer(
+	config util.Config, querier db.Querier, matcher language.Matcher,
+	logger util.Logger, grpcClient grpc.GrpcClient,
+) (*Server, error) {
 
 	server := &Server{
-		config:  config,
-		querier: querier,
-		matcher: matcher,
-		logger:  logger,
+		config:     config,
+		querier:    querier,
+		matcher:    matcher,
+		logger:     logger,
+		grpcClient: grpcClient,
 	}
 
 	err := server.setupRouter()
@@ -84,6 +90,14 @@ func (server *Server) setupRouter() error {
 		HandlerFunc(server.getAllFormations).
 		Methods("GET")
 
+	r.Path(rootPath + "/members/download").
+		HandlerFunc(server.downloadMembers).
+		Methods("GET")
+
+	r.Path(rootPath + "/health/grpc").
+		HandlerFunc(server.healthGrpc).
+		Methods("GET")
+
 	server.router = r
 	return nil
 }

app.env

@@ -6,3 +6,4 @@ DB_SOURCE=postgresql://ubuntu:sakamichi@localhost:5432/sakamichi?sslmode=disable
 LOG_PATH=/usr/log/api/saka-debug.log
 HOST=ubuntu
 SERVICE=saka-api
+GRPC_URL=localhost:9876

main.go

@@ -26,7 +26,7 @@ func main() {
 
 	logger, closeFunc, err := util.NewZapLogger(config.LogPath, config.Host, config.Service, util.Degub)
 	if err != nil {
-		log.Fatal("cannot create logger")
+		log.Fatal("cannot create logger: ", err)
 	}
 	defer closeFunc()
 
@@ -38,8 +38,10 @@ func main() {
 
 	matcher := util.NewMatcher()
 
+	grpcClient := grpc.NewClient(logger, config)
+
 	// DI to server
-	server, err := api.NewServer(config, querier, matcher, logger)
+	server, err := api.NewServer(config, querier, matcher, logger, grpcClient)
 	if err != nil {
 		logger.Criticalf(context.Background(), "cannot create server:", err)
 	}