Skip to content

Commit

Permalink
Merge pull request #396 from nishthax/main
Browse files Browse the repository at this point in the history
Added the code for url shortener using java
  • Loading branch information
kishanrajput23 authored Oct 23, 2023
2 parents aa398c0 + 167a885 commit 778065d
Showing 1 changed file with 52 additions and 0 deletions.
52 changes: 52 additions & 0 deletions UrlShortener.java
Original file line number Diff line number Diff line change
@@ -0,0 +1,52 @@
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.HashMap;
import java.util.Map;
import java.util.Random;

public class UrlShortener {
private Map<String, String> urlMap = new HashMap<>();
private static final String BASE_URL = "http://short.url/";

public String shortenUrl(String originalUrl) {
String shortUrl = generateShortUrl(originalUrl);
urlMap.put(shortUrl, originalUrl);
return BASE_URL + shortUrl;
}

public String expandUrl(String shortUrl) {
String shortKey = shortUrl.substring(BASE_URL.length());
return urlMap.get(shortKey);
}

private String generateShortUrl(String originalUrl) {
try {
MessageDigest md = MessageDigest.getInstance("MD5");
byte[] digest = md.digest(originalUrl.getBytes());

StringBuilder shortUrl = new StringBuilder();
Random random = new Random();

for (int i = 0; i < 6; i++) {
int index = random.nextInt(digest.length);
shortUrl.append(String.format("%02x", digest[index]));
}

return shortUrl.toString();
} catch (NoSuchAlgorithmException e) {
e.printStackTrace();
return null;
}
}

public static void main(String[] args) {
UrlShortener urlShortener = new UrlShortener();

String longUrl = "https://www.example.com/some/long/url";
String shortUrl = urlShortener.shortenUrl(longUrl);
System.out.println("Short URL: " + shortUrl);

String expandedUrl = urlShortener.expandUrl(shortUrl);
System.out.println("Expanded URL: " + expandedUrl);
}
}

0 comments on commit 778065d

Please sign in to comment.