feat: use alloydb omni instead of crdb

Signed-off-by: STRRL <[email protected]>

Author: STRRL

pkg/cmd/cmd.go

@@ -45,9 +45,11 @@ Then, with a single command, you create and start all the services from your con
 			}
 		}
 
+		patchFileSetDatabaseConnectionURI(file, "postgres://postgres:password@rss3_node_alloydb:5432/postgres")
+
 		composeFile := compose.NewCompose(
 			compose.WithWorkers(cfg.Component.Decentralized),
-			compose.SetDependsOnCRDB(),
+			compose.SetDependsOnAlloyDB(),
 			compose.SetNodeVersion(version),
 			compose.SetNodeVolume(),
 			compose.SetRestartPolicy(),
@@ -84,6 +86,69 @@ func randomString(n int) string {
 	return string(b)
 }
 
+func patchFileSetDatabaseConnectionURI(file string, newConnectionURI string) error {
+	discovered, err := discoverConfigFile(file)
+	if err != nil {
+		return fmt.Errorf("patch config file with new database connection uri, discover config file, %w", err)
+	}
+
+	f, err := os.Open(discovered)
+	if err != nil {
+		return fmt.Errorf("patch config file with new database connection uri, open config file, %w", err)
+	}
+
+	// we do not unmashal DSL node config.File directly because it does not have yaml struct tags and it does not play well with yaml encoder
+	// as a workaround, we unmarshal the file into yaml.Node then manually patch the access token
+	var root yaml.Node
+	if err = yaml.NewDecoder(f).Decode(&root); err != nil {
+		return fmt.Errorf("patch config file with new database connection uri, decode config file, %w", err)
+	}
+
+	if len(root.Content) > 0 {
+		databaseNode, err := findYamlNode("database", root.Content[0])
+		if err != nil {
+			return fmt.Errorf("patch config file with new database connection uri, find database node, %w", err)
+		}
+
+		if databaseNode == nil {
+			return fmt.Errorf("patch config file with new database connection uri, database node not found")
+		}
+
+		uriNode, err := findYamlNode("uri", databaseNode)
+		if err != nil {
+			return fmt.Errorf("patch config file with new database connection uri, find uri node, %w", err)
+		}
+
+		if uriNode == nil {
+			return fmt.Errorf("patch config file with new database connection uri, uri node not found")
+		}
+
+		uriNode.Kind = yaml.ScalarNode
+		uriNode.Tag = "!!str"
+		uriNode.Value = newConnectionURI
+
+	}
+
+	// dump patched yaml node to file
+	if err = f.Close(); err != nil {
+		return fmt.Errorf("patch config file with generated access token, close config file, %w", err)
+	}
+
+	f, err = os.OpenFile(discovered, os.O_WRONLY|os.O_TRUNC, 0644)
+	if err != nil {
+		return fmt.Errorf("patch config file with generated access token, open config file, %w", err)
+	}
+
+	err = yaml.NewEncoder(f).Encode(&root)
+	if err != nil {
+		return fmt.Errorf("patch config file with generated access token, encode config file, %w", err)
+	}
+
+	f.Close()
+
+	return nil
+}
+
 func patchConfigFileWithAccessToken(file string, accessToken string) error {
 	discovered, err := discoverConfigFile(file)
 	if err != nil {

pkg/compose/compose.go

@@ -43,7 +43,7 @@ type Option func(*Compose)
 const dockerComposeContainerNamePrefix = "rss3_node"
 
 func NewCompose(options ...Option) *Compose {
-	cockroachdbVolume := "cockroachdb"
+	alloydbVolume := "alloydb"
 
 	compose := &Compose{
 		Services: map[string]Service{
@@ -58,19 +58,21 @@ func NewCompose(options ...Option) *Compose {
 					Retries:  3,
 				},
 			},
-			fmt.Sprintf("%s_cockroachdb", dockerComposeContainerNamePrefix): {
-				Command:       "start-single-node --cluster-name=node --insecure",
-				ContainerName: fmt.Sprintf("%s_cockroachdb", dockerComposeContainerNamePrefix),
-				Expose:        []string{"26257", "8080"},
-				Image:         "cockroachdb/cockroach:v23.2.5",
-				Volumes:       []string{fmt.Sprintf("%s:/cockroach/cockroach-data", cockroachdbVolume)},
-				// we use similar healthcheck as the official cockroachdb operator
-				// ref: https://github.com/cockroachdb/cockroach-operator/blob/28d139cb0c19d3c7984b2b2da1b25c5ba388d814/pkg/resource/testdata/TestStatefulSetBuilder/default_secure.golden#L76-L83
+			fmt.Sprintf("%s_alloydb", dockerComposeContainerNamePrefix): {
+				ContainerName: fmt.Sprintf("%s_alloydb", dockerComposeContainerNamePrefix),
+				Expose:        []string{"5432"},
+				Image:         "google/alloydbomni:latest",
+				Volumes:       []string{fmt.Sprintf("%s:/alloydb/alloydb-data", alloydbVolume)},
+				Environment: map[string]string{
+					"DATA_DIR":          "/var/lib/postgresql/data",
+					"HOST_PORT":         "5432",
+					"POSTGRES_PASSWORD": "password",
+				},
 				Healthcheck: Healthcheck{
-					Test:     []string{"CMD", "curl", "-f", "http://localhost:8080/health?ready=1"},
+					Test:     []string{"CMD-SHELL", "pg_isready -U postgres"},
 					Interval: 5 * time.Second,
-					Timeout:  1 * time.Second,
-					Retries:  3,
+					Timeout:  5 * time.Second,
+					Retries:  5,
 				},
 			},
 			fmt.Sprintf("%s_core", dockerComposeContainerNamePrefix): {
@@ -91,7 +93,7 @@ func NewCompose(options ...Option) *Compose {
 			},
 		},
 		Volumes: map[string]*string{
-			cockroachdbVolume: nil,
+			alloydbVolume: nil,
 		},
 	}
 
@@ -159,14 +161,14 @@ func WithWorkers(workers []*config.Module) Option {
 	}
 }
 
-// SetDependsOnCRDB would set all the rss3 node service to depend on the cockroachdb service
-func SetDependsOnCRDB() Option {
+// SetDependsOnAlloyDB would set all the rss3 node service to depend on the AlloyDB service
+func SetDependsOnAlloyDB() Option {
 	return func(c *Compose) {
 		services := c.Services
 		for k, v := range services {
 			if strings.Contains(v.Image, "rss3/node") {
 				v.DependsOn = map[string]DependsOn{
-					fmt.Sprintf("%s_cockroachdb", dockerComposeContainerNamePrefix): {
+					fmt.Sprintf("%s_alloydb", dockerComposeContainerNamePrefix): {
 						Condition: "service_healthy",
 					},
 					fmt.Sprintf("%s_redis", dockerComposeContainerNamePrefix): {