1 files changed, 121 insertions, 0 deletions
diff --git a/benchmarks/s3concurrent/main.go b/benchmarks/s3concurrent/main.go
new file mode 100644
index 0000000..a67fc5b
--- /dev/null
+++ b/benchmarks/s3concurrent/main.go
@@ -0,0 +1,121 @@
+package main
+
+import (
+    "context"
+    "crypto/tls"
+    "io"
+    "log"
+    "math/rand"
+    "net/http"
+    "os"
+
+	"github.com/minio/minio-go/v7"
+	"github.com/minio/minio-go/v7/pkg/credentials"
+    "github.com/google/uuid"
+)
+
+func buildMc() (*minio.Client, error) {
+	_, isSSL := os.LookupEnv("SSL")
+	opts := minio.Options{
+		Creds:  credentials.NewStaticV4(os.Getenv("AWS_ACCESS_KEY_ID"), os.Getenv("AWS_SECRET_ACCESS_KEY"), ""),
+		Secure: isSSL,
+	}
+
+	if region, ok := os.LookupEnv("REGION"); ok {
+		opts.Region = region
+	}
+
+	if _, ok := os.LookupEnv("SSL_INSECURE"); ok {
+		opts.Transport = &http.Transport{TLSClientConfig: &tls.Config{InsecureSkipVerify: true}}
+	}
+
+	mc, err := minio.New(os.Getenv("ENDPOINT"), &opts)
+    return mc, err
+}
+
+type PRNG struct { 
+    rem int64
+}
+func (r *PRNG)  Read(p []byte) (n int, err error) {
+    //log.Printf("rem=%d, buf=%d\n", r.rem, len(p))
+    if int64(len(p)) > r.rem {
+        p = p[:r.rem]
+    }
+
+    if int64(len(p)) > r.rem {
+        log.Fatal("LOGIC ERROR")
+    }
+
+    n, err = rand.Read(p)
+    if err != nil {
+        return
+    }
+    r.rem -= int64(n)
+    if r.rem <= 0 {
+        err = io.EOF
+        //log.Printf("PRNG file has been fully read. rem=%d,n=%d,err=%s\n", r.rem, n, err)
+    }
+    return
+}
+
+func putObj(buck string, size int64) error {
+    mc, err := buildMc()
+    if err != nil {
+        return err
+    }
+
+    prng := new(PRNG)
+    prng.rem = size
+
+    key := uuid.New().String()
+
+    _, err = mc.PutObject(
+        context.Background(), 
+        buck, 
+        key, 
+        prng, 
+        size, 
+        minio.PutObjectOptions{ContentType:"application/octet-stream"},
+    )
+
+    return err
+}
+
+func main() {
+    minio.MaxRetry = 1
+    mc, err := buildMc()
+	if err != nil {
+		log.Fatal("failed connect", err)
+		return
+	}
+
+    // Create Bucket
+    buck := uuid.New().String()
+    err = mc.MakeBucket(context.Background(), buck, minio.MakeBucketOptions{ })
+    if err != nil {
+        log.Fatal(err)
+        return
+    }
+    log.Printf("created bucket %s\n", buck)
+
+    // Send to bucket
+    for i := 1; i <= 16; i++ {
+        log.Printf("start concurrent loop with %d coroutines\n", i)
+        syn := make(chan error)
+        for j := 1; j <= i; j++ {
+            go func() {
+                syn <- putObj(buck, 1024 * 1024)
+            }()
+        }
+
+        for j := 1; j <= i; j++ {
+            cerr := <-syn
+            if cerr != nil {
+                log.Printf("%d/%d failed with %s\n", j, i, cerr)
+            }
+        }
+        log.Printf("done, %d coroutines returned\n", i)
+    }
+
+    log.Println("done")
+}