main.go

package main

import (
	"net"
	"log"
	"os/signal"
	"os"
	pb "upload-service/proto"
	
	"google.golang.org/grpc"
	"github.com/aws/aws-sdk-go/aws"
	"github.com/aws/aws-sdk-go/aws/credentials"
	"github.com/aws/aws-sdk-go/aws/session"
	"github.com/aws/aws-sdk-go/service/s3"
)

func main() {
	s3AccessKey := os.Getenv("S3_ACCESS_KEY")
    s3SecretKey := os.Getenv("S3_SECRET_KEY")
    s3Endpoint  := os.Getenv("S3_ENDPOINT")
	tcpPort		:= os.Getenv("TCP_PORT")
	s3Token     := ""

	// Configure to use S3 Server
	s3Config := &aws.Config{
		Credentials:      credentials.NewStaticCredentials(s3AccessKey, s3SecretKey, s3Token),
		Endpoint:         aws.String(s3Endpoint),
		Region:           aws.String("eu-east-1"),
		DisableSSL:       aws.Bool(true),
		S3ForcePathStyle: aws.Bool(true),
	}
	newSession := session.New(s3Config)
	s3Client := s3.New(newSession)
	lis, err := net.Listen("tcp", ":" + tcpPort)
	if err != nil {
		log.Fatalf("failed to listen: %v", err)
	}
	
	grpcServer := grpc.NewServer()
	server := &UploadHandler{ UploadService: UploadService{ s3Client: s3Client }}
	pb.RegisterUploadServer(grpcServer, server)
	grpcServer.Serve(lis)
	signalChan := make(chan os.Signal, 1)
    signal.Notify(signalChan, os.Interrupt)
    <-signalChan
}