DELIMITER //
CREATE FUNCTION broj_prijatelja(id INT)
RETURNS INT
DETERMINISTIC
BEGIN
    DECLARE ret INT;
     SELECT COUNT(*) FROM prijateljstva WHERE PRIJATELJ_1 = id OR PRIJATELJ_2 = id INTO ret;
    RETURN ret;
END; //

DELIMITER ;