一、在这之前先写一点关于事务的小知识点如下:
引用:import org.geotools.data.DefaultTransaction;
import org.geotools.data.Transaction;
//追加数据的事务
1、Transaction session = new DefaultTransaction(“Adding”);
//事务的设置
featureStore.setTransaction( session );
//创建文件存储事务
2、Transaction transaction = new DefaultTransaction(“create”)
//自动提交事务
3、Transaction.AUTO_COMMIT
二、将数据添加到数据库demo
package org.geotools.WPS;
import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.geotools.data.DataStore;
import org.geotools.data.DataStoreFinder;
import org.geotools.data.DefaultTransaction;
import org.geotools.data.Transaction;
import org.geotools.data.collection.ListFeatureCollection;
import org.geotools.data.postgis.PostgisNGDataStoreFactory;
import org.geotools.data.simple.SimpleFeatureCollection;
import org.geotools.data.simple.SimpleFeatureSource;
import org.geotools.data.simple.SimpleFeatureStore;
import org.geotools.feature.simple.SimpleFeatureBuilder;
import org.opengis.feature.simple.SimpleFeature;
import org.opengis.feature.simple.SimpleFeatureType;
import org.opengis.filter.identity.FeatureId;
import org.locationtech.jts.geom.Coordinate;
import org.locationtech.jts.geom.GeometryFactory;
import org.locationtech.jts.geom.Point;
public class FeatureAddToDataBase {
/**
* @param dbtype: 数据库类型,postgis or mysql
* @param host: ip地址
* @param port: 端口号
* @param database: 需要连接的数据库
* @param userName: 用户名
* @param password: 密码
* @param tableName: 需要连接的表名
* @return: 返回为SimpleFeatureSource类型
*/
public static SimpleFeatureSource connAndgetCollection(String dbtype, String host, String port,
String database, String userName, String password,String tableName) {
Map<String, Object> params = new HashMap<String, Object>();
DataStore pgDatastore=null;
params.put(PostgisNGDataStoreFactory.DBTYPE.key, dbtype); //需要连接何种数据库,postgis or mysql
params.put(PostgisNGDataStoreFactory.HOST.key, host);//ip地址
params.put(PostgisNGDataStoreFactory.PORT.key, new Integer(port));//端口号
params.put(PostgisNGDataStoreFactory.DATABASE.key, database);//需要连接的数据库
params.put(PostgisNGDataStoreFactory.SCHEMA.key, "public");//架构
params.put(PostgisNGDataStoreFactory.USER.key, userName);//需要连接数据库的名称
params.put(PostgisNGDataStoreFactory.PASSWD.key, password);//数据库的密码
SimpleFeatureSource fSource=null;
try {
//获取存储空间
pgDatastore = DataStoreFinder.getDataStore(params);
//根据表名获取source
fSource=pgDatastore.getFeatureSource(tableName);
if (pgDatastore != null) {
System.out.println("系统连接到位于:" + host + "的空间数据库" + database
+ "成功!");
} else {
System.out.println("系统连接到位于:" + host + "的空间数据库" + database
+ "失败!请检查相关参数");
}
} catch (IOException e) {
e.printStackTrace();
System.out.println("系统连接到位于:" + host + "的空间数据库" + database
+ "失败!请检查相关参数");
}
return fSource;
}
/**
* @param featureSource:要素源
* @return SimpleFeatureCollection类型
*/
public static SimpleFeatureCollection GetFeatureCollection (SimpleFeatureSource featureSource) {
GeometryFactory geometryFactory = new GeometryFactory();
SimpleFeatureCollection collection =null;
//获取类型
SimpleFeatureType TYPE = featureSource.getSchema();
System.out.println(TYPE);
//创建要素集合
List<SimpleFeature> features = new ArrayList<>();
//创建要素模板
SimpleFeatureBuilder featureBuilder = new SimpleFeatureBuilder(TYPE);
//创建要素并添加道集合
double latitude = Double.parseDouble("39.9");
double longitude = Double.parseDouble("116.3");
String name ="beijing";
int number = Integer.parseInt("16");
//创建一个点geometry
Point point = geometryFactory.createPoint(new Coordinate(longitude, latitude));
//添加的数据一定按照SimpleFeatureType给的字段顺序进行赋值
//添加name属性
featureBuilder.add(name);
//添加number属性
featureBuilder.add(number);
//添加geometry属性
featureBuilder.add(point);
//构建要素
SimpleFeature feature = featureBuilder.buildFeature(null);
System.out.println(feature);
//将要素添加到要素几何中
features.add(feature);
collection = new ListFeatureCollection(TYPE, features);
return collection;
}
public static void main(String[] args) {
//调用方法
SimpleFeatureSource featureSource=FeatureAddToDataBase.connAndgetCollection("postgis", "localhost", "5432", "sqlView", "postgres", "postgres","7");
if( featureSource instanceof SimpleFeatureStore ){
SimpleFeatureStore featureStore = (SimpleFeatureStore) featureSource;
SimpleFeatureCollection featureCollection=FeatureAddToDataBase.GetFeatureCollection(featureSource);
//创建事务
Transaction session = new DefaultTransaction("Adding");
featureStore.setTransaction( session );
try {
List<FeatureId> added = featureStore.addFeatures( featureCollection );
System.out.println( "Added "+added );
//提交事务
session.commit();
}
catch (Throwable t){
System.out.println( "Failed to add features: "+t );
try {
//事务回归
session.rollback();
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}
}
}