View
package com.zhongruan.View;
import com.zhongruan.Dao.MusicDao;
import com.zhongruan.Dao.UserDao;
import com.zhongruan.LoginException.LoginExcption;
import com.zhongruan.model.Music;
import com.zhongruan.model.User;
import java.sql.SQLException;
import java.util.List;
import java.util.Scanner;
public class User_View {
public static void main(String[] args) throws LoginExcption, SQLException, ClassNotFoundException {
boolean flag=true;
while (flag){
System.out.println("-----您尚未登录,请先登录-----");
System.out.println("请输入用户名:");
Scanner input=new Scanner(System.in);
String username=input.next();
System.out.println("请输入密码:");
String pw=input.next();
UserDao userDao=new UserDao();
userDao.add(username,pw);
User user=userDao.findUserByUsername(username);
if (user==null){
System.out.println("-----该用户尚未注册-----");
System.out.println("请输入注册名:");
String name=input.next();
System.out.println("请输入注册密码:");
String password=input.next();
userDao.add(name,password);
System.out.println("-----注册成功!-----");
}else if(user.getPassword().equals(pw)){
System.out.println("-----欢迎来到音乐管理系统-----");
musicSystem(user);
flag=false;
}else {
// System.out.println("-----密码错误,请重新输入-----");
throw new LoginExcption("登录失败");
}
}
}
public static void musicSystem(User user) throws SQLException, ClassNotFoundException {
Scanner input=new Scanner(System.in);
System.out.println("1、音乐查询 2、音乐添加 3、音乐修改 4、音乐删除 5、推出");
int choice=input.nextInt();
MusicDao musicDao=new MusicDao();
switch (choice){
case 1:
List<Music> musics=musicDao.findMusics();
System.out.println(musics);
break;
case 2:
if (user.getType()!=1){
System.out.println("权限不足,请联系管理员。");
break;
}
System.out.println("请输入要添加的音乐名:");
String musicname=input.next();
System.out.println("请输入要添加的音乐作者名:");
String musicauthor=input.next();
musicDao.add(musicname,musicauthor);
break;
case 3:
if (user.getType()!=0){
System.out.println("权限不足,请联系管理员。");
break;
}
System.out.println("");
System.out.println("请输入音乐名:");
String musicname1=input.next();
System.out.println("请输入作者名:");
String author=input.next();
MusicDao.update(musicname1,author);
case 4:
if (user.getType()!=0){
System.out.println("权限不足,请联系管理员。");
break;
}
System.out.println("请输入要删除的歌曲id");
int id2=input.nextInt();
musicDao.delete(id2);
case 5:System.exit(0);
}
}
}
LoginException
package com.zhongruan.LoginException;
public class LoginExcption extends Exception{
public LoginExcption(String message){
super(message);
}
}
User
package com.zhongruan.model;
public class User {
private int id;
private String name;
private String password;
private int type;
public int getId() {
return id;
}
public void setId(int id) {
this.id = id;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public String getPassword() {
return password;
}
public void setPassword(String password) {
this.password = password;
}
public int getType() {
return type;
}
public void setType(int type) {
this.type = type;
}
@Override
public String toString() {
return "User{" +
"id=" + id +
", name='" + name + '\'' +
", password='" + password + '\'' +
", type=" + type +
'}';
}
}
Userdao
package com.zhongruan.Dao;
import com.zhongruan.Util.DBUtil;
import com.zhongruan.model.Music;
import com.zhongruan.model.User;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.util.ArrayList;
import java.util.List;
public class UserDao {
public User findUserByUsername(String username){
ResultSet resultSet=null;
PreparedStatement statement=null;
Connection connection=null;
User user=null;
try{
connection= DBUtil.getConnection();
String sql="select * from tb_user where username=?";
statement=connection.prepareStatement(sql);
statement.setString(1,username);
resultSet=statement.executeQuery();
while (resultSet.next()){
user=new User();
user.setName(resultSet.getString(2));
user.setPassword(resultSet.getString(3));
user.setType(resultSet.getInt(4));
}
}catch (Exception e){
e.printStackTrace();
}finally {
DBUtil.CloseAll(resultSet,statement,connection);
}
return user;
}
public void add(String username,String password){
ResultSet resultSet=null;
PreparedStatement statement = null;
Connection connection = null;
try{
connection = DBUtil.getConnection();
String sql="INSERT INTO stu (username ,password) VALUES(?,?)";
statement = connection.prepareStatement(sql);
statement.setString(1,username);
statement.setString(2,password);
}catch (Exception e) {
e.printStackTrace();
} finally {
DBUtil.CloseAll(null, statement, connection);
}
}
}