-
Notifications
You must be signed in to change notification settings - Fork 0
/
shortener_controller.go
67 lines (58 loc) · 1.86 KB
/
shortener_controller.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
package controllers
import (
"net/http"
"time"
"github.com/gin-gonic/gin"
"github.com/zerodot618/go-huang/api/models"
)
// ShortenerController is a struct that represents a controller for shortener-related operations“
type ShortenerController struct{}
// CreateShortURl creates a short URL from a long URL and stores it in the database
// c *gin.Context is a pointer to the Gin context
func (ctrl *ShortenerController) CreateShortURL(c *gin.Context) {
// Create a URL struct
var url models.URL
// Bind the JSON body to the URL struct
c.BindJSON(&url)
// Generate a short URL
url.GenerateShortURL()
// Create the URL in the database
models.CreateURL(&url)
// Return the short URL in the response
c.JSON(http.StatusOK, gin.H{"short_url": url.ShortURL})
}
// RedirectShortURL redirects the user to the long URL associated with the short URL
// c *gin.Context is a pointer to the Gin context
func (ctrl *ShortenerController) RedirectShortURL(c *gin.Context) {
shortURL := c.Param("short_url")
url, err := models.GetURLByShortURL(shortURL)
if err != nil {
c.JSON(http.StatusNotFound, gin.H{"error": "URL not found"})
c.Abort()
return
}
// Update statistics
url.AccessCount++
now := time.Now()
url.LastAccessed = &now
url.AccessPlace = c.ClientIP()
models.UpdateURL(&url)
c.Redirect(http.StatusMovedPermanently, url.LongURL)
}
// GetURLStatistics returns the statistics for a specific short URL
// c *gin.Context is a pointer to the Gin context
func (ctrl *ShortenerController) GetURLStatistics(c *gin.Context) {
shortURL := c.Param("short_url")
url, err := models.GetURLByShortURL(shortURL)
if err != nil {
c.JSON(404, gin.H{"error": "URL not found"})
return
}
c.JSON(200, gin.H{
"short_url": url.ShortURL,
"long_url": url.LongURL,
"access_count": url.AccessCount,
"last_accessed": url.LastAccessed,
"access_place": url.AccessPlace,
})
}