pagina

1 files changed, 4 insertions, 2 deletions

diff --git a/main_server.go b/main_server.go
index ea41ce8..00d3938 100644
--- a/main_server.go
+++ b/main_server.go
@@ -59,7 +59,7 @@ func correct_url(url string) string {
 	return url;
 }
 
-func add_url(url string) {
+func add_url(url string) string{
 	hurl := correct_url(url);
 	hsh  := gen_hash(hurl);
 	fmt.Printf("adding: %s -> %s \n", hurl, hsh);
@@ -90,6 +90,7 @@ func add_url(url string) {
 			log.Fatal(err);
 		}
 	}
+	return hsh;	
 }
 
 func get_hash_from_db(hash string) string{
@@ -122,7 +123,8 @@ func read_json(cont []byte) (Data, error) {
 
 func handle_request(conn net.Conn, data Data) {
 	if data.Type == 1 {
-		add_url(data.URL);
+		ret := add_url(data.URL);
+		conn.Write([]byte(fmt.Sprintf("%s\n", ret)));
 	}else if data.Type == 2 {
 		ret := get_hash_from_db(data.URL);
 		conn.Write([]byte(fmt.Sprintf("%s\n", ret)));