add sample readme, parse time duration from config

Author: phuongdnguyen

Dockerfile

@@ -0,0 +1,8 @@
+FROM cgr.dev/chainguard/go AS builder
+COPY . /app
+RUN cd /app && CGO_ENABLED=0 GOOS=linux go build -o reverse-registry .
+
+FROM cgr.dev/chainguard/glibc-dynamic
+COPY --from=builder /app/reverse-registry /usr/bin/
+COPY ./config/config.local.yaml /etc/reverse-registry/config.local.yaml
+CMD ["/usr/bin/go-digester", "server", "--config=/etc/reverse-registry/config.local.yaml"]

README.md

@@ -0,0 +1,8 @@
+# Reverse registry
+
+
+This is a simple registry redirect service that redirect our.domain.com/* to cgr.dev/chainguard/*. It also run a background process to periodically hash image index to sha256 checksum and save to a local database (default in-mem sqlite is used. For production purpose, mysql is the recommended choice)
+
+Deploying
+
+[![Run on Google Cloud](https://deploy.cloud.run/button.svg)](https://deploy.cloud.run)

app.json

@@ -0,0 +1,25 @@
+{
+    "name": "reverse-registry",
+    "env": {
+        "WORKER_FETCH_INTERVAL": {
+            "description": "specify interval to fetch manifest from cgr",
+            "value": "30s",
+            "required": true
+        }
+    },
+    "options": {
+        "allow-unauthenticated": false,
+        "memory": "512Mi",
+        "cpu": "1",
+        "port": 443,
+        "http2": false,
+        "concurrency": 80,
+        "max-instances": 1
+    },
+    "build": {
+        "skip": false,
+        "buildpacks": {
+            "builder": "gcr.io/buildpacks/builder:v1"
+        }
+    }
+}

app/app.go

@@ -58,11 +58,15 @@ func RunFetcher(conf config.Config) error {
 	if err != nil {
 		return err
 	}
+	d, err := time.ParseDuration(conf.WorkerFetchInterval)
+	if err != nil {
+		return err
+	}
 	fetcher := digestfetcher.New(digestfetcher.Options{
 		Storage:       storage,
 		Registry:      registryClient,
 		Log:           log,
-		FetchInterval: conf.WorkerFetchInterval * time.Second,
+		FetchInterval: d,
 	})
 	return fetcher.Fetch(conf.Images)
 }

cmd/root.go

@@ -28,6 +28,7 @@ import (
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
 	"github.com/xxxibgdrgnmm/reverse-registry/config"
+	"github.com/xxxibgdrgnmm/reverse-registry/constant"
 )
 
 var cfgFile string
@@ -92,5 +93,13 @@ func initConfig() {
 		if err = viper.Unmarshal(&c); err != nil {
 			panic(fmt.Sprintf("can not marshal config %v", err))
 		}
+		wF := os.Getenv(constant.WorkerFetchIntervalEnv)
+		mP := os.Getenv(constant.MySQLPassWordEnv)
+		if wF != "" {
+			c.WorkerFetchInterval = wF
+		}
+		if mP != "" {
+			c.DBConfig.Password = mP
+		}
 	}
 }

config/config.go

@@ -1,7 +1,5 @@
 package config
 
-import "time"
-
 type MysqlConfig struct {
 	Host     string `mapstructure:"host"`
 	User     string `mapstructure:"user"`
@@ -10,10 +8,10 @@ type MysqlConfig struct {
 }
 
 type Config struct {
-	DB                  string        `mapstructure:"db"`
-	DBConfig            MysqlConfig   `mapstructure:"dbConfig"`
-	Images              []Image       `mapstructure:"images"`
-	WorkerFetchInterval time.Duration `mapstructure:"workerFetchInterval"`
+	DB                  string      `mapstructure:"db"`
+	DBConfig            MysqlConfig `mapstructure:"dbConfig"`
+	Images              []Image     `mapstructure:"images"`
+	WorkerFetchInterval string      `mapstructure:"workerFetchInterval"`
 }
 
 type Image struct {

constant/const.go

@@ -0,0 +1,6 @@
+package constant
+
+const (
+	WorkerFetchIntervalEnv = "WORKER_FETCH_INTERVAL"
+	MySQLPassWordEnv       = "MYSQL_PASSWORD"
+)

services/digest-fetcher/digetst_fetcher_test.go

@@ -18,6 +18,7 @@ func TestFetch(t *testing.T) {
 		Constraint:  "^1.2.*",
 		MainPackage: "nginx",
 	})
+
 	conf := config.Config{
 		DBConfig: config.MysqlConfig{
 			Host:     "localhost",
@@ -26,17 +27,19 @@ func TestFetch(t *testing.T) {
 			DBName:   "test",
 		},
 		Images:              images,
-		WorkerFetchInterval: 10,
+		WorkerFetchInterval: "10s",
 	}
+	d, err := time.ParseDuration(conf.WorkerFetchInterval)
 	storage, err := inject.GetStorage(conf)
 	assert.NoError(t, err)
 	registryClient, err := inject.GetContainerRegistryClient()
 	assert.NoError(t, err)
+	assert.NoError(t, err)
 	fetcher := New(Options{
 		Storage:       storage,
 		Registry:      registryClient,
 		Log:           log,
-		FetchInterval: conf.WorkerFetchInterval * time.Second,
+		FetchInterval: d,
 	})
 	fetcher.Fetch(conf.Images)
 }