Merge pull request #1 from xxxibgdrgnmm/refactor-repository

feat: refactor repository factory, add readme

Author: phuongdnguyen

.gitignore

@@ -0,0 +1,2 @@
+.idea/
+reverse-registry

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,9 +8,10 @@ type MysqlConfig struct {
 }
 
 type Config struct {
-	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 {

config/config.local.yaml

@@ -1,3 +1,4 @@
+db: sqlite
 dbConfig:
   host: localhost
   user: root

constant/const.go

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

driver/mysql_driver.go

@@ -15,9 +15,6 @@ func NewMySQLDB(host string, user string, password string, dbName string) (*gorm
 		Logger: logger.Default.LogMode(logger.Silent),
 	})
 	if err != nil {
-		msg := fmt.Sprintf("cannot connect to database. host: %s, user: %s, db: %s", host, user, dbName)
-		fmt.Println(msg)
-
 		return nil, err
 	}