113 lines
2.9 KiB
Go
113 lines
2.9 KiB
Go
package main
|
||
|
||
import (
|
||
"encoding/base64"
|
||
"time"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
const defaultTTL = time.Hour * 24 * 365 // 默认过期时间，1年
|
||
const defaultRenewTime = time.Hour * 48 // 默认续命时间，2天
|
||
const defaultShortKeyLength = 7 // 默认短链接长度，7位
|
||
|
||
// ShortToLongHandler gets the long URL from a short URL
|
||
func ShortToLongHandler() gin.HandlerFunc {
|
||
return func(c *gin.Context) {
|
||
resp := Response{}
|
||
shortKey := c.Param("shortKey")
|
||
longURL := ShortToLong(c, shortKey)
|
||
if longURL == "" {
|
||
resp.Code = ResponseCodeServerError
|
||
resp.Msg = "failed to get long URL, please check the short URL if exists or expired"
|
||
|
||
c.JSON(404, resp)
|
||
return
|
||
}
|
||
|
||
// todo
|
||
// check whether need renew expiration time
|
||
// only renew once per day
|
||
// if err := Renew(c, shortKey, defaultRenewTime); err != nil {
|
||
// logger.Warn("failed to renew short URL: ", err.Error())
|
||
// }
|
||
|
||
c.Redirect(301, longURL)
|
||
}
|
||
}
|
||
|
||
type LongToShortParams struct {
|
||
LongUrl string `form:"longUrl" json:"longUrl" binding:"required"`
|
||
ShortKey string `form:"shortKey" json:"shortKey" binding:"omitempty"`
|
||
}
|
||
|
||
// LongToShortHandler creates a short URL from a long URL
|
||
func LongToShortHandler() gin.HandlerFunc {
|
||
return func(c *gin.Context) {
|
||
resp := Response{}
|
||
|
||
// check parameters
|
||
req := LongToShortParams{}
|
||
if err := c.ShouldBind(&req); err != nil {
|
||
resp.Code = ResponseCodeParamsCheckError
|
||
resp.Msg = "invalid parameters"
|
||
logger.Warn("invalid parameters: ", err.Error())
|
||
|
||
c.JSON(200, resp)
|
||
return
|
||
}
|
||
|
||
// 兼容以前的实现，这里如果是 base64 编码的字符串，进行解码
|
||
_longUrl, err := base64.StdEncoding.DecodeString(req.LongUrl)
|
||
if err == nil {
|
||
req.LongUrl = string(_longUrl)
|
||
}
|
||
|
||
// generate short key
|
||
if req.ShortKey == "" {
|
||
req.ShortKey = GenerateRandomString(defaultShortKeyLength)
|
||
}
|
||
// check whether short key exists
|
||
exists, err := CheckRedisKeyIfExist(c, req.ShortKey)
|
||
if err != nil {
|
||
resp.Code = ResponseCodeServerError
|
||
resp.Msg = "failed to check short key"
|
||
logger.Error("failed to check short key: ", err.Error())
|
||
|
||
c.JSON(200, resp)
|
||
return
|
||
}
|
||
if exists {
|
||
resp.Code = ResponseCodeParamsCheckError
|
||
resp.Msg = "short key already exists, please use another one or leave it empty to generate automatically"
|
||
|
||
logger.Info("short key already exists: ", req.ShortKey)
|
||
c.JSON(200, resp)
|
||
return
|
||
}
|
||
|
||
options := &LongToShortOptions{
|
||
ShortKey: req.ShortKey,
|
||
URL: req.LongUrl,
|
||
expiration: defaultTTL,
|
||
}
|
||
if err := LongToShort(c, options); err != nil {
|
||
resp.Code = ResponseCodeServerError
|
||
resp.Msg = "failed to create short URL"
|
||
logger.Warn("failed to create short URL: ", err.Error())
|
||
|
||
c.JSON(200, resp)
|
||
return
|
||
}
|
||
|
||
shortURL := proto + "://" + domain + "/" + options.ShortKey
|
||
|
||
// 兼容以前的返回结构体
|
||
respDataLegacy := gin.H{
|
||
"Code": ResponseCodeSuccessLegacy,
|
||
"ShortUrl": shortURL,
|
||
}
|
||
c.JSON(200, respDataLegacy)
|
||
}
|
||
}
|