// 将十进制转换为62进制   0-9a-zA-Z 六十二进制
func transTo62(id int64)string{// 1 -- > 1// 10-- > a// 61-- > Zcharset := "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"var shortUrl []bytefor{var result bytenumber := id % 62result = charset[number]var tmp []bytetmp = append(tmp,result)shortUrl = append(tmp,shortUrl...)id = id / 62if id == 0{break}}fmt.Println(string(shortUrl))return string(shortUrl)
}

|____logic
| |____logic.go
|____model
| |____data.go
|____api
| |____api.go
|____client
| |____client.go

package modeltype Long2ShortRequest struct {OriginUrl string `json:"origin_url"`
}type ResponseHeader struct {Code int `json:"code"`Message string `json:"message"`
}type Long2ShortResponse struct {ResponseHeaderShortUrl string `json:"short_url"`
}type Short2LongRequest struct {ShortUrl string `json:"short_url"`
}type Short2LongResponse struct {ResponseHeaderOriginUrl string `json:"origin_url"`
}

package logicimport("go_dev/11/short_url/model""github.com/jmoiron/sqlx""fmt""crypto/md5""database/sql"
)var (Db *sqlx.DB
)type ShortUrl struct {Id int64 `db:"id"`ShortUrl string `db:"short_url"`OriginUrl string `db:"origin_url"`HashCode string `db:"hash_code"`
}func InitDb(dsn string)(err error) {// 数据库初始化Db, err = sqlx.Open("mysql",dsn)if err != nil{fmt.Println("connect to mysql failed:",err)return}return
}func Long2Short(req *model.Long2ShortRequest) (response *model.Long2ShortResponse, err error) {response = &model.Long2ShortResponse{}urlMd5 := fmt.Sprintf("%x",md5.Sum([]byte(req.OriginUrl)))var short ShortUrlerr = Db.Get(&short,"select id,short_url,origin_url,hash_code from short_url where hash_code=?",urlMd5)if err == sql.ErrNoRows{err = nil// 数据库中没有记录，重新生成一个新的短urlshortUrl,errRet := generateShortUrl(req,urlMd5)if errRet != nil{err = errRetreturn}response.ShortUrl = shortUrlreturn}if err != nil{return}response.ShortUrl = short.ShortUrlreturn
}func generateShortUrl(req *model.Long2ShortRequest,hashcode string)(shortUrl string,err error){result,err := Db.Exec("insert INTO short_url(origin_url,hash_code)VALUES (?,?)",req.OriginUrl,hashcode)if err != nil{return}// 0-9a-zA-Z 六十二进制insertId,_:= result.LastInsertId()shortUrl = transTo62(insertId)_,err = Db.Exec("update short_url set short_url=? where id=?",shortUrl,insertId)if err != nil{fmt.Println(err)return}return
}// 将十进制转换为62进制   0-9a-zA-Z 六十二进制
func transTo62(id int64)string{// 1 -- > 1// 10-- > a// 61-- > Zcharset := "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"var shortUrl []bytefor{var result bytenumber := id % 62result = charset[number]var tmp []bytetmp = append(tmp,result)shortUrl = append(tmp,shortUrl...)id = id / 62if id == 0{break}}fmt.Println(string(shortUrl))return string(shortUrl)
}func Short2Long(req *model.Short2LongRequest) (response *model.Short2LongResponse, err error) {response = &model.Short2LongResponse{}var short ShortUrlerr = Db.Get(&short,"select id,short_url,origin_url,hash_code from short_url where short_url=?",req.ShortUrl)if err == sql.ErrNoRows{response.Code = 404return}if err != nil{response.Code = 500return}response.OriginUrl = short.OriginUrlreturn
}

package mainimport ("io/ioutil""net/http""fmt""encoding/json""go_dev/11/short_url/logic""go_dev/11/short_url/model"_ "github.com/go-sql-driver/mysql"
)const (ErrSuccess = 0ErrInvalidParameter = 1001ErrServerBusy = 1002
)func getMessage(code int) (msg string){switch code {case ErrSuccess:msg = "success"case ErrInvalidParameter:msg = "invalid parameter"case ErrServerBusy:msg = "server busy"default:msg = "unknown error"}return
}// 用于将返回序列化数据，失败的返回
func responseError(w http.ResponseWriter, code int) {var response model.ResponseHeaderresponse.Code = coderesponse.Message = getMessage(code)data, err := json.Marshal(response)if err != nil {w.Write([]byte("{\"code\":500, \"message\": \"server busy\"}"))return}w.Write(data)
}// 用于将返回序列化数据，成功的返回
func responseSuccess(w http.ResponseWriter, data interface{}) {dataByte, err := json.Marshal(data)if err != nil {w.Write([]byte("{\"code\":500, \"message\": \"server busy\"}"))return}w.Write(dataByte)
}// 长地址到短地址
func Long2Short(w http.ResponseWriter, r *http.Request) {// 这里需要说明的是发来的数据是通过post发过来一个json格式的数据data, err := ioutil.ReadAll(r.Body)if err != nil {fmt.Println("read all failded, ", err)responseError(w, 1001)return}var req model.Long2ShortRequest// 将反序列化的数据保存在结构体中err = json.Unmarshal(data, &req)if err != nil {fmt.Println("Unmarshal failded, ", err)responseError(w, 1002)return}resp, err := logic.Long2Short(&req)if err != nil {fmt.Println("Long2Short failded, ", err)responseError(w, 1003)return}responseSuccess(w, resp)
}// 短地址到长地址
func Short2Long(w http.ResponseWriter, r *http.Request) {// 这里需要说明的是发来的数据是通过post发过来一个json格式的数据data, err := ioutil.ReadAll(r.Body)if err != nil {fmt.Println("read all failded, ", err)responseError(w, 1001)return}var req model.Short2LongRequest// 将反序列化的数据保存在结构体中err = json.Unmarshal(data, &req)if err != nil {fmt.Println("Unmarshal failded, ", err)responseError(w, 1002)return}resp, err := logic.Short2Long(&req)if err != nil {fmt.Println("Long2Short failded, ", err)responseError(w, 1003)return}responseSuccess(w, resp)
}func main(){err := logic.InitDb("root:123456@tcp(192.168.50.145:3306)/short_url?parseTime=true")if err != nil{fmt.Printf("init db failed,err:%v\n",err)return}http.HandleFunc("/trans/long2short", Long2Short)http.HandleFunc("/trans/short2long", Short2Long)http.ListenAndServe(":18888", nil)
}