Merge pull request #1 from devsapp/support-run-local

Support run local

Author: rsonghuster

src/code/const.go

@@ -20,7 +20,9 @@ const (
 	fcAccountID = "x-fc-account-id"
 	fcQualifier = "x-fc-qualifier"
 	fcVersionID = "x-fc-version-id"
+)
 
+var (
 	CertPEM_PATH       = "cert/test-cert.pem"
 	PrivateKeyPEM_PATH = "cert/test-priv.pem"
 	WORK_DIR_BASE      = "/mnt/auto"

src/code/handler.go

@@ -33,6 +33,10 @@ func parseRange(r string) (int64, int64, error) {
 		return 0, 0, err
 	}
 
+	if sz := endPos + 1 - beginPos; sz <= 0 || sz > 50*1024*1024 {
+		return 0, 0, fmt.Errorf("invalid range: %s", r)
+	}
+
 	return beginPos, endPos + 1, nil
 }
 
@@ -41,6 +45,7 @@ func handler(w http.ResponseWriter, r *http.Request) {
 	log.Printf("fcContext=%v", fcCtx)
 	if err != nil {
 		handleError(w, fmt.Errorf("fail to NewFromContext due to  %v", err))
+		return
 	}
 	switch r.Method {
 	case "HEAD":

src/code/main.go

@@ -1,10 +1,89 @@
 package main
 
 import (
+	"io"
+	"log"
 	"net/http"
+	"os"
+	fcoss "repack/oss"
+
+	"github.com/aliyun/aliyun-oss-go-sdk/oss"
 )
 
+func repackLocal() {
+	CertPEM_PATH = "/tmp/cert/test-cert.pem"
+	PrivateKeyPEM_PATH = "/tmp/cert/test-priv.pem"
+	WORK_DIR_BASE = "/tmp"
+
+	fcCtx := &FCContext{
+		SourceObject: os.Getenv("SOURCE_OBJECT"),
+		ChannelID:    os.Getenv("CHANNEL_ID"),
+		OSSEndpoint:  os.Getenv("OSS_ENDPOINT"),
+		Credentials: Credentials{
+			AccessKeyID:     os.Getenv("ACCESS_KEY_ID"),
+			AccessKeySecret: os.Getenv("ACCESS_KEY_SECRET"),
+		},
+		WorkDir: "/tmp",
+	}
+
+	f, res, err := repackAPK(fcCtx)
+	if err != nil {
+		log.Printf("repack error: %v", err)
+		return
+	}
+	defer f.Close()
+	log.Printf("res: %+v", res)
+
+	ossReader, err := fcoss.NewReader(
+		fcoss.OSSConfig{
+			Endpoint:        fcCtx.OSSEndpoint,
+			AccessKeyID:     fcCtx.Credentials.AccessKeyID,
+			AccessKeySecret: fcCtx.Credentials.AccessKeySecret,
+			SecurityToken:   fcCtx.Credentials.SecurityToken,
+		}, fcCtx.SourceObject)
+	if err != nil {
+		log.Printf("read oss: %v", err)
+		return
+	}
+	resp, err := ossReader.Client.GetObject(
+		ossReader.Object, oss.Range(0, res.Offset-1))
+	if err != nil {
+		log.Printf("get object: %v", err)
+		return
+	}
+	defer resp.Close()
+
+	resFile, err := os.Create("/tmp/res.apk")
+	if err != nil {
+		log.Printf("get object: %v", err)
+		return
+	}
+	defer resFile.Close()
+	n, err := io.Copy(resFile, resp)
+	if err != nil {
+		log.Printf("copy: %v", err)
+		return
+	}
+	log.Printf("copied %d bytes", n)
+
+	if _, err := f.Seek(0, 0); err != nil {
+		log.Printf("seek error: %v", err)
+		return
+	}
+	n, err = io.Copy(resFile, f)
+	if err != nil {
+		log.Printf("copy: %v", err)
+		return
+	}
+	log.Printf("copied %d bytes", n)
+}
+
 func main() {
+	if os.Getenv("RUN_LOCAL") == "true" {
+		repackLocal()
+		return
+	}
+
 	http.HandleFunc("/", handler)
 	http.ListenAndServe(":80", nil)
 }

src/code/oss/oss.go

@@ -36,12 +36,33 @@ type OSSConfig struct {
 	SecurityToken   string
 }
 
-// NewReader ...
-func NewReader(config OSSConfig, location string) (*Reader, error) {
+var mu sync.Mutex
+var ossClient *oss.Client
+
+func getOSSClient(config OSSConfig) (*oss.Client, error) {
+	mu.Lock()
+	defer mu.Unlock()
+
+	if ossClient != nil {
+		return ossClient, nil
+	}
+
 	client, err := oss.New(
 		config.Endpoint, config.AccessKeyID, config.AccessKeySecret,
 		oss.SecurityToken(config.SecurityToken))
 
+	if err != nil {
+		return nil, err
+	}
+	ossClient = client
+
+	return client, nil
+}
+
+// NewReader ...
+func NewReader(config OSSConfig, location string) (*Reader, error) {
+	client, err := getOSSClient(config)
+
 	if err != nil {
 		return nil, err
 	}