前言

可能是自己對yaml檔案不夠了解吧,想做一個yaml檔案的讀寫,在網上搜到的例項都是使用yaml.dump(),我也這麼跟著做,但使用該方法的yaml檔案始終在開頭有兩個!!,導致沒辦法使用yaml.load()進行讀取.最後看到一個yaml.dumpAsMap()的方法,靈光一現,乾脆自己寫一個得了.以下是實現程式碼,就是一些簡單的IO;

程式碼

除了基礎的jdk外,還引入了兩個包,分別是:fastjson-1.2.2.jar，snakeyaml-1.17.jar

package com.xiangpeng.IM.test;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.UUID;

import org.junit.jupiter.api.Test;
import org.yaml.snakeyaml.Yaml;

import com.alibaba.fastjson.JSON;

public class YamlTest {
	
	@Test
	public void yamlWrite() throws IOException {
		// 準備寫入物件
		User user = new User();
		user.setUid(UUID.randomUUID().toString().replace("-", ""));
		user.setName("彰顯忠");
		user.setSex("男");
		user.setAge(37);

		// 例項化一個Yaml,並準備寫入資料
		Yaml yaml = new Yaml();
		String dumpAsMap = yaml.dumpAsMap(user);
		
		// 使用流寫入yaml檔案
		FileWriter fileWriter = new FileWriter(YamlTest.class.getClassLoader().getResource("").getPath()+"../config/UserInfo.yml");
		PrintWriter writer = new PrintWriter(fileWriter);
		writer.write(dumpAsMap);
		
		// 收尾
		fileWriter.close();
		writer.close();
	}
	
	@Test
	public void yamlRead() throws FileNotFoundException {
		Yaml yaml = new Yaml();
		File file = new File(YamlTest.class.getClassLoader().getResource("").getPath()+"../config/UserInfo.yml");
		User user = JSON.parseObject(JSON.toJSONString(yaml.load(new FileInputStream(file))), User.class);
		System.out.println(user);
	}
}

class User {
	private String uid;
	private String name;
	private String sex;
	private int age;
	@Override
	public String toString() {
		return "User [uid=" + uid + ", name=" + name + ", sex=" + sex + ", age=" + age + "]";
	}
	public String getUid() {
		return uid;
	}
	public void setUid(String uid) {
		this.uid = uid;
	}
	public String getName() {
		return name;
	}
	public void setName(String name) {
		this.name = name;
	}
	public String getSex() {
		return sex;
	}
	public void setSex(String sex) {
		this.sex = sex;
	}
	public int getAge() {
		return age;
	}
	public void setAge(int age) {
		this.age = age;
	}
}