Create & Init Project...

app/admin/main/macross/dao/oss/BUILD

@@ -0,0 +1,33 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["dao.go"],
+    importpath = "go-common/app/admin/main/macross/dao/oss",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/admin/main/macross/conf:go_default_library",
+        "//library/log:go_default_library",
+        "//vendor/github.com/aliyun/aliyun-oss-go-sdk/oss:go_default_library",
+    ],
+)
+
+filegroup(
+    name = "package-srcs",
+    srcs = glob(["**"]),
+    tags = ["automanaged"],
+    visibility = ["//visibility:private"],
+)
+
+filegroup(
+    name = "all-srcs",
+    srcs = [":package-srcs"],
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+)

app/admin/main/macross/dao/oss/dao.go

@@ -0,0 +1,61 @@
+package oss
+
+import (
+	"context"
+	"io"
+	"path"
+
+	"go-common/app/admin/main/macross/conf"
+	"go-common/library/log"
+
+	"github.com/aliyun/aliyun-oss-go-sdk/oss"
+)
+
+// Dao macross dao
+type Dao struct {
+	// conf
+	c      *conf.Config
+	client *oss.Client
+}
+
+// New dao.
+func New(c *conf.Config) (d *Dao) {
+	d = &Dao{
+		c: c,
+	}
+	client, err := oss.New(d.c.Oss.Endpoint, d.c.Oss.AccessKeyID, d.c.Oss.AccessKeySecret)
+	if err != nil {
+		panic(err)
+	}
+	d.client = client
+	return
+}
+
+// Put put object into oss.
+func (d *Dao) Put(c context.Context, rd io.Reader, apkName string) (uri string, err error) {
+	bucket, _ := d.client.Bucket(d.c.Oss.Bucket)
+	uri = path.Join(d.c.Oss.OriginDir, apkName)
+	// put
+	if err = bucket.PutObject(uri, rd); err != nil {
+		log.Error("bucket.PutObject(%s) error(%v)", uri, err)
+		return
+	}
+	uri = path.Join(d.c.Oss.Bucket, uri) // NOTE: begin with '/' when upload success
+	return
+}
+
+// Publish publish object into oss.
+func (d *Dao) Publish(objectKey string, destKey string) (uri string, err error) {
+	bucket, _ := d.client.Bucket(d.c.Oss.Bucket)
+	_, err = bucket.CopyObject(d.c.Oss.PublishDir+objectKey, d.c.Oss.PublishDir+destKey)
+	if err != nil {
+		log.Error("bucket.CopyObject(%s, %s) error(%v)", d.c.Oss.PublishDir+objectKey, d.c.Oss.PublishDir+destKey, err)
+		return
+	}
+	uri = path.Join("/", d.c.Oss.PublishDir+destKey)
+	return
+}
+
+// Close close kafka connection.
+func (d *Dao) Close() {
+}