Go工作流與容器化數(shù)據(jù)庫(kù)集成

Go 語(yǔ)言是一種高性能的編程語(yǔ)言，適用于構(gòu)建高并發(fā)、分布式和微服務(wù)架構(gòu)的應(yīng)用程序

1. 使用 Go 語(yǔ)言創(chuàng)建一個(gè)簡(jiǎn)單的工作流引擎：

package main

import (
	"fmt"
)

type Task func() error

type Workflow struct {
	tasks []Task
}

func NewWorkflow() *Workflow {
	return &Workflow{
		tasks: make([]Task, 0),
	}
}

func (w *Workflow) AddTask(task Task) {
	w.tasks = append(w.tasks, task)
}

func (w *Workflow) Run() error {
	for _, task := range w.tasks {
		if err := task(); err != nil {
			return err
		}
	}
	return nil
}

func main() {
	workflow := NewWorkflow()

	workflow.AddTask(func() error {
		fmt.Println("Task 1: Initialize database connection")
		return nil
	})

	workflow.AddTask(func() error {
		fmt.Println("Task 2: Create tables")
		return nil
	})

	workflow.AddTask(func() error {
		fmt.Println("Task 3: Insert data")
		return nil
	})

	workflow.AddTask(func() error {
		fmt.Println("Task 4: Query data")
		return nil
	})

	if err := workflow.Run(); err != nil {
		fmt.Printf("Error running workflow: %v\n", err)
	} else {
		fmt.Println("Workflow completed successfully")
	}
}

2. 使用 Docker 容器化數(shù)據(jù)庫(kù)：

首先，創(chuàng)建一個(gè) docker-compose.yml 文件，定義數(shù)據(jù)庫(kù)服務(wù)和應(yīng)用程序服務(wù)：

version: '3'

services:
  db:
    image: "postgres:latest"
    environment:
      POSTGRES_USER: "user"
      POSTGRES_PASSWORD: "password"
      POSTGRES_DB: "mydb"
    volumes:
      - "./data:/var/lib/postgresql/data"
    ports:
      - "5432:5432"

  app:
    build: .
    depends_on:
      - db
    environment:
      DATABASE_URL: "postgres://user:password@db:5432/mydb?sslmode=disable"
    ports:
      - "8080:8080"

3. 在 Go 應(yīng)用程序中使用容器化數(shù)據(jù)庫(kù)：

package main

import (
	"database/sql"
	"fmt"
	"log"
	"os"

	_ "github.com/lib/pq"
)

func main() {
	// 獲取數(shù)據(jù)庫(kù)連接字符串
	databaseURL := os.Getenv("DATABASE_URL")
	if databaseURL == "" {
		log.Fatal("DATABASE_URL is not set")
	}

	// 連接到數(shù)據(jù)庫(kù)
	db, err := sql.Open("postgres", databaseURL)
	if err != nil {
		log.Fatalf("Error connecting to database: %v", err)
	}
	defer db.Close()

	// 創(chuàng)建表
	_, err = db.Exec(`CREATE TABLE IF NOT EXISTS users (id SERIAL PRIMARY KEY, name TEXT)`)
	if err != nil {
		log.Fatalf("Error creating table: %v", err)
	}

	// 插入數(shù)據(jù)
	_, err = db.Exec(`INSERT INTO users (name) VALUES ('John Doe')`)
	if err != nil {
		log.Fatalf("Error inserting data: %v", err)
	}

	// 查詢數(shù)據(jù)
	rows, err := db.Query(`SELECT id, name FROM users`)
	if err != nil {
		log.Fatalf("Error querying data: %v", err)
	}
	defer rows.Close()

	for rows.Next() {
		var id int
		var name string
		err = rows.Scan(&id, &name)
		if err != nil {
			log.Fatalf("Error scanning row: %v", err)
		}
		fmt.Printf("User: %d, %s\n", id, name)
	}

	if err = rows.Err(); err != nil {
		log.Fatalf("Error iterating through rows: %v", err)
	}
}

4. 構(gòu)建和運(yùn)行應(yīng)用程序：

在項(xiàng)目根目錄下運(yùn)行以下命令：

docker-compose up --build

這將啟動(dòng)數(shù)據(jù)庫(kù)容器和應(yīng)用程序容器，并在應(yīng)用程序容器中運(yùn)行工作流。應(yīng)用程序?qū)⑦B接到數(shù)據(jù)庫(kù)，創(chuàng)建表、插入數(shù)據(jù)并查詢數(shù)據(jù)。