diff options
Diffstat (limited to 'main_server.go')
| -rw-r--r-- | main_server.go | 6 |
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))); |