resolve url by id

micro · Sep 2, 2022 · 63be799 · 63be799
1 parent 3604044
commit 63be799
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 7 deletions.
diff --git a/url/handler/url.go b/url/handler/url.go
@@ -292,6 +292,10 @@ func (e *Url) List(ctx context.Context, req *url.ListRequest, rsp *url.ListRespo
 func (e *Url) Resolve(ctx context.Context, req *url.ResolveRequest, rsp *url.ResolveResponse) error {
 	id := strings.Replace(req.ShortURL, e.hostPrefix, "", -1)
 
+	if len(req.Id) > 0 {
+		id = req.Id
+	}
+
 	records, err := store.Read("url/" + id)
 	if err != nil {
 		return err

diff --git a/url/proto/url.pb.go b/url/proto/url.pb.go
diff --git a/url/proto/url.proto b/url/proto/url.proto
@@ -84,8 +84,10 @@ message UpdateResponse {}
 
 // Resolve returns the destination URL of a short URL.
 message ResolveRequest {
+	// resolve by id
+	string id = 1;
 	// short url to resolve
-	string shortURL = 1;
+	string shortURL = 2;
 }
 
 message ResolveResponse {