|
1 | 1 | package main |
2 | 2 |
|
3 | 3 | import ( |
4 | | - "context" |
5 | | - "log/slog" |
6 | | - "os" |
7 | | - "os/signal" |
| 4 | + "database/sql" |
| 5 | + "fmt" |
| 6 | + "log" |
| 7 | + "net/http" |
| 8 | + "time" |
8 | 9 |
|
9 | | - "github.com/talgat-ruby/exercises-go/exercise7/blogging-platform/internal/api" |
10 | | - "github.com/talgat-ruby/exercises-go/exercise7/blogging-platform/internal/db" |
| 10 | + "github.com/gin-gonic/gin" |
| 11 | + "github.com/lib/pq" |
| 12 | + _ "github.com/lib/pq" |
11 | 13 | ) |
12 | 14 |
|
| 15 | +type BlogPost struct { |
| 16 | + ID int `json:"id"` |
| 17 | + Title string `json:"title" binding:"required"` |
| 18 | + Content string `json:"content" binding:"required"` |
| 19 | + Category string `json:"category" binding:"required"` |
| 20 | + Tags []string `json:"tags" binding:"required"` |
| 21 | + CreatedAt time.Time `json:"createdAt"` |
| 22 | + UpdatedAt time.Time `json:"updatedAt"` |
| 23 | +} |
| 24 | + |
| 25 | +var db *sql.DB |
| 26 | + |
13 | 27 | func main() { |
14 | | - ctx, cancel := context.WithCancel(context.Background()) |
| 28 | + var err error |
| 29 | + connStr := "postgres://astanamac:adlet@localhost:5432/blog?sslmode=disable" |
| 30 | + db, err = sql.Open("postgres", connStr) |
| 31 | + if err != nil { |
| 32 | + log.Fatal("Ошибка при подключении к PostgreSQL:", err) |
| 33 | + } |
15 | 34 |
|
16 | | - // db |
17 | | - _, err := db.New() |
| 35 | + // Проверка соединения |
| 36 | + err = db.Ping() |
18 | 37 | if err != nil { |
19 | | - slog.ErrorContext( |
20 | | - ctx, |
21 | | - "initialize service error", |
22 | | - "service", "db", |
23 | | - "error", err, |
24 | | - ) |
25 | | - panic(err) |
26 | | - } |
27 | | - |
28 | | - // api |
29 | | - a := api.New() |
30 | | - if err := a.Start(ctx); err != nil { |
31 | | - slog.ErrorContext( |
32 | | - ctx, |
33 | | - "initialize service error", |
34 | | - "service", "api", |
35 | | - "error", err, |
36 | | - ) |
37 | | - panic(err) |
38 | | - } |
39 | | - |
40 | | - go func() { |
41 | | - shutdown := make(chan os.Signal, 1) // Create channel to signify s signal being sent |
42 | | - signal.Notify(shutdown, os.Interrupt) // When an interrupt is sent, notify the channel |
43 | | - |
44 | | - sig := <-shutdown |
45 | | - slog.WarnContext(ctx, "signal received - shutting down...", "signal", sig) |
46 | | - |
47 | | - cancel() |
48 | | - }() |
| 38 | + log.Fatal("Не удалось подключиться к базе данных:", err) |
| 39 | + } else { |
| 40 | + fmt.Println("✅ Успешное подключение к базе данных!") |
| 41 | + } |
| 42 | + |
| 43 | + r := gin.Default() |
| 44 | + r.POST("/posts", createPost) |
| 45 | + r.PUT("/posts/:id", updatePost) |
| 46 | + r.DELETE("/posts/:id", deletePost) |
| 47 | + r.GET("/posts/:id", getPost) |
| 48 | + r.GET("/posts", getAllPosts) |
| 49 | + |
| 50 | + fmt.Println("🚀 Сервер запущен на порту 8080") |
| 51 | + r.Run(":8080") |
| 52 | +} |
| 53 | + |
| 54 | +func createPost(c *gin.Context) { |
| 55 | + var post BlogPost |
| 56 | + if err := c.ShouldBindJSON(&post); err != nil { |
| 57 | + c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) |
| 58 | + return |
| 59 | + } |
| 60 | + timeNow := time.Now() |
| 61 | + post.CreatedAt, post.UpdatedAt = timeNow, timeNow |
| 62 | + |
| 63 | + query := `INSERT INTO posts (title, content, category, tags, created_at, updated_at) VALUES ($1, $2, $3, $4, $5, $6) RETURNING id` |
| 64 | + err := db.QueryRow(query, post.Title, post.Content, post.Category, pq.Array(post.Tags), post.CreatedAt, post.UpdatedAt).Scan(&post.ID) |
| 65 | + if err != nil { |
| 66 | + c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) |
| 67 | + return |
| 68 | + } |
| 69 | + |
| 70 | + c.JSON(http.StatusCreated, post) |
| 71 | +} |
| 72 | + |
| 73 | +func updatePost(c *gin.Context) { |
| 74 | + id := c.Param("id") |
| 75 | + var post BlogPost |
| 76 | + if err := c.ShouldBindJSON(&post); err != nil { |
| 77 | + c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) |
| 78 | + return |
| 79 | + } |
| 80 | + |
| 81 | + post.UpdatedAt = time.Now() |
| 82 | + query := `UPDATE posts SET title=$1, content=$2, category=$3, tags=$4, updated_at=$5 WHERE id=$6` |
| 83 | + result, err := db.Exec(query, post.Title, post.Content, post.Category, pq.Array(post.Tags), post.UpdatedAt, id) |
| 84 | + if err != nil { |
| 85 | + c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) |
| 86 | + return |
| 87 | + } |
| 88 | + |
| 89 | + rowsAffected, _ := result.RowsAffected() |
| 90 | + if rowsAffected == 0 { |
| 91 | + c.JSON(http.StatusNotFound, gin.H{"error": "Post not found"}) |
| 92 | + return |
| 93 | + } |
| 94 | + |
| 95 | + c.JSON(http.StatusOK, post) |
| 96 | +} |
| 97 | + |
| 98 | +func deletePost(c *gin.Context) { |
| 99 | + id := c.Param("id") |
| 100 | + query := `DELETE FROM posts WHERE id=$1` |
| 101 | + result, err := db.Exec(query, id) |
| 102 | + if err != nil { |
| 103 | + c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) |
| 104 | + return |
| 105 | + } |
| 106 | + |
| 107 | + rowsAffected, _ := result.RowsAffected() |
| 108 | + if rowsAffected == 0 { |
| 109 | + c.JSON(http.StatusNotFound, gin.H{"error": "Post not found"}) |
| 110 | + return |
| 111 | + } |
| 112 | + |
| 113 | + c.Status(http.StatusNoContent) |
| 114 | +} |
| 115 | + |
| 116 | +func getPost(c *gin.Context) { |
| 117 | + id := c.Param("id") |
| 118 | + var post BlogPost |
| 119 | + var tags pq.StringArray |
| 120 | + |
| 121 | + query := `SELECT id, title, content, category, tags, created_at, updated_at FROM posts WHERE id=$1` |
| 122 | + err := db.QueryRow(query, id).Scan(&post.ID, &post.Title, &post.Content, &post.Category, (*pq.StringArray)(&tags), &post.CreatedAt, &post.UpdatedAt) |
| 123 | + if err != nil { |
| 124 | + c.JSON(http.StatusNotFound, gin.H{"error": "Post not found"}) |
| 125 | + return |
| 126 | + } |
| 127 | + |
| 128 | + post.Tags = tags |
| 129 | + c.JSON(http.StatusOK, post) |
| 130 | +} |
| 131 | + |
| 132 | +func getAllPosts(c *gin.Context) { |
| 133 | + searchTerm := c.Query("term") |
| 134 | + query := `SELECT id, title, content, category, tags, created_at, updated_at FROM posts WHERE title ILIKE $1 OR content ILIKE $1 OR category ILIKE $1` |
| 135 | + rows, err := db.Query(query, "%"+searchTerm+"%") |
| 136 | + if err != nil { |
| 137 | + c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) |
| 138 | + return |
| 139 | + } |
| 140 | + defer rows.Close() |
| 141 | + |
| 142 | + var posts []BlogPost |
| 143 | + for rows.Next() { |
| 144 | + var post BlogPost |
| 145 | + var tags pq.StringArray |
| 146 | + |
| 147 | + if err := rows.Scan(&post.ID, &post.Title, &post.Content, &post.Category, (*pq.StringArray)(&tags), &post.CreatedAt, &post.UpdatedAt); err != nil { |
| 148 | + c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) |
| 149 | + return |
| 150 | + } |
| 151 | + post.Tags = tags |
| 152 | + posts = append(posts, post) |
| 153 | + } |
| 154 | + |
| 155 | + c.JSON(http.StatusOK, posts) |
49 | 156 | } |
0 commit comments