Erectile dysfunctions are psychological in 90% cases. It may be situational or due other illness that affect nearby area of genital organs specifically prostate issues. So don’t worry, once prostatitis will be cure, your erection will be ok. But please consult properly with concerned urologist and take proper treatment as this problem will require long term antibiotics for 2-4 weeks depending up on response.