.
This commit is contained in:
parent
897286b0fe
commit
a204032fe4
|
@ -0,0 +1,9 @@
|
||||||
|
package com.loafle.bridge.discoveryport.constant;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by root on 16. 11. 15.
|
||||||
|
*/
|
||||||
|
public enum DirectionType {
|
||||||
|
S, // Send
|
||||||
|
R // Recv
|
||||||
|
}
|
|
@ -0,0 +1,9 @@
|
||||||
|
package com.loafle.bridge.discoveryport.constant;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by root on 16. 11. 15.
|
||||||
|
*/
|
||||||
|
|
||||||
|
public enum PortType {
|
||||||
|
TCP,UDP
|
||||||
|
}
|
|
@ -1,32 +1,60 @@
|
||||||
package com.loafle.bridge.discoveryport.entity;
|
package com.loafle.bridge.discoveryport.entity;
|
||||||
|
|
||||||
|
import com.loafle.bridge.discoveryhost.entity.DiscoveryHost;
|
||||||
|
import com.loafle.bridge.discoveryport.constant.PortType;
|
||||||
|
|
||||||
import javax.persistence.*;
|
import javax.persistence.*;
|
||||||
import java.util.Date;
|
import java.util.Date;
|
||||||
|
import java.util.List;
|
||||||
/**
|
|
||||||
* Created by root on 16. 11. 15.
|
|
||||||
*/
|
|
||||||
|
|
||||||
@Entity
|
@Entity
|
||||||
public class DiscoveryPort {
|
public class DiscoveryPort {
|
||||||
|
|
||||||
@Id
|
@Id
|
||||||
|
@Column(name = "ID")
|
||||||
@GeneratedValue(strategy = GenerationType.AUTO)
|
@GeneratedValue(strategy = GenerationType.AUTO)
|
||||||
private long id;
|
private long id;
|
||||||
|
|
||||||
@Column(nullable = false)
|
@ManyToOne
|
||||||
private String type;
|
@JoinColumn(name = "HOST_ID", nullable = false)
|
||||||
|
private DiscoveryHost host;
|
||||||
|
|
||||||
@Column(nullable = false)
|
@Column(name = "PORT_TYPE",nullable = false)
|
||||||
|
@Enumerated(EnumType.STRING)
|
||||||
|
private PortType type;
|
||||||
|
|
||||||
|
@Column(name = "PORT_NUMBER",nullable = false)
|
||||||
private short portNumber;
|
private short portNumber;
|
||||||
|
|
||||||
@Temporal(value = TemporalType.TIMESTAMP)
|
@OneToMany(mappedBy = "port" , fetch = FetchType.LAZY)
|
||||||
private Date createDate;
|
private List<ServiceScanHistory> histories;
|
||||||
|
|
||||||
@Temporal(value = TemporalType.TIMESTAMP)
|
@Column(name = "CREATE_AT",nullable = false,updatable = false)
|
||||||
private Date updateDate;
|
@Temporal(TemporalType.TIMESTAMP)
|
||||||
|
private Date createAt;
|
||||||
|
|
||||||
public DiscoveryPort() {
|
@Column(name = "UPDATE_AT",nullable = false)
|
||||||
|
@Temporal(TemporalType.TIMESTAMP)
|
||||||
|
private Date updateAt;
|
||||||
|
|
||||||
|
|
||||||
|
public DiscoveryPort() {}
|
||||||
|
|
||||||
|
public DiscoveryHost getHost() {
|
||||||
|
return host;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setHost(DiscoveryHost host) {
|
||||||
|
this.host = host;
|
||||||
|
}
|
||||||
|
|
||||||
|
public DiscoveryPort(PortType type, DiscoveryHost host, short port, Date createAt, Date updateAt) {
|
||||||
|
this.host = host;
|
||||||
|
|
||||||
|
this.type = type;
|
||||||
|
this.portNumber = port;
|
||||||
|
this.createAt = createAt;
|
||||||
|
this.updateAt = updateAt;
|
||||||
}
|
}
|
||||||
|
|
||||||
public long getId() {
|
public long getId() {
|
||||||
|
@ -37,11 +65,11 @@ public class DiscoveryPort {
|
||||||
this.id = id;
|
this.id = id;
|
||||||
}
|
}
|
||||||
|
|
||||||
public String getType() {
|
public PortType getType() {
|
||||||
return type;
|
return type;
|
||||||
}
|
}
|
||||||
|
|
||||||
public void setType(String type) {
|
public void setType(PortType type) {
|
||||||
this.type = type;
|
this.type = type;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -53,19 +81,27 @@ public class DiscoveryPort {
|
||||||
this.portNumber = portNumber;
|
this.portNumber = portNumber;
|
||||||
}
|
}
|
||||||
|
|
||||||
public Date getCreateDate() {
|
public Date getCreateAt() {
|
||||||
return createDate;
|
return createAt;
|
||||||
}
|
}
|
||||||
|
|
||||||
public void setCreateDate(Date createDate) {
|
public void setCreateAt(Date createAt) {
|
||||||
this.createDate = createDate;
|
this.createAt = createAt;
|
||||||
}
|
}
|
||||||
|
|
||||||
public Date getUpdateDate() {
|
public Date getUpdateAt() {
|
||||||
return updateDate;
|
return updateAt;
|
||||||
}
|
}
|
||||||
|
|
||||||
public void setUpdateDate(Date updateDate) {
|
public void setUpdateAt(Date updateAt) {
|
||||||
this.updateDate = updateDate;
|
this.updateAt = updateAt;
|
||||||
|
}
|
||||||
|
|
||||||
|
public List<ServiceScanHistory> getHistories() {
|
||||||
|
return histories;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setHistories(List<ServiceScanHistory> histories) {
|
||||||
|
this.histories = histories;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,7 +1,96 @@
|
||||||
package com.loafle.bridge.discoveryport.entity;
|
package com.loafle.bridge.discoveryport.entity;
|
||||||
|
|
||||||
|
import com.loafle.bridge.discoveryport.constant.DirectionType;
|
||||||
|
|
||||||
|
import javax.persistence.*;
|
||||||
|
import java.util.Date;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Created by root on 16. 11. 15.
|
* Created by root on 16. 11. 15.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
@Entity
|
||||||
public class ServiceScanHistory {
|
public class ServiceScanHistory {
|
||||||
|
|
||||||
|
@Id
|
||||||
|
@Column(name = "ID")
|
||||||
|
@GeneratedValue(strategy = GenerationType.AUTO)
|
||||||
|
private long id;
|
||||||
|
|
||||||
|
@ManyToOne(fetch = FetchType.LAZY)
|
||||||
|
@JoinColumn(name = "PORT_ID")
|
||||||
|
private DiscoveryPort port;
|
||||||
|
|
||||||
|
@Column(name = "SERVICE_NAME", nullable = false)
|
||||||
|
private String serviceName;
|
||||||
|
|
||||||
|
@Column(name = "DIRECTION", nullable = false)
|
||||||
|
@Enumerated(EnumType.STRING)
|
||||||
|
private DirectionType direction;
|
||||||
|
|
||||||
|
@Column(name = "CREATE_AT",nullable = false,updatable = false)
|
||||||
|
@Temporal(TemporalType.TIMESTAMP)
|
||||||
|
private Date createAt;
|
||||||
|
|
||||||
|
@Lob
|
||||||
|
@Column(name = "PACKET", nullable = false)
|
||||||
|
private byte[] packet;
|
||||||
|
|
||||||
|
public ServiceScanHistory() {}
|
||||||
|
|
||||||
|
public ServiceScanHistory(DiscoveryPort port, String serviceName, DirectionType direction, byte[] packet, Date createAt) {
|
||||||
|
this.port = port;
|
||||||
|
this.serviceName = serviceName;
|
||||||
|
this.direction = direction;
|
||||||
|
this.packet = packet;
|
||||||
|
this.createAt = createAt;
|
||||||
|
}
|
||||||
|
|
||||||
|
public long getId() {
|
||||||
|
return id;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setId(long id) {
|
||||||
|
this.id = id;
|
||||||
|
}
|
||||||
|
|
||||||
|
public DiscoveryPort getPort() {
|
||||||
|
return port;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setPort(DiscoveryPort port) {
|
||||||
|
this.port = port;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getServiceName() {
|
||||||
|
return serviceName;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setServiceName(String serviceName) {
|
||||||
|
this.serviceName = serviceName;
|
||||||
|
}
|
||||||
|
|
||||||
|
public DirectionType getDirection() {
|
||||||
|
return direction;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setDirection(DirectionType direction) {
|
||||||
|
this.direction = direction;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Date getCreateAt() {
|
||||||
|
return createAt;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setCreateAt(Date createAt) {
|
||||||
|
this.createAt = createAt;
|
||||||
|
}
|
||||||
|
|
||||||
|
public byte[] getPacket() {
|
||||||
|
return packet;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setPacket(byte[] packet) {
|
||||||
|
this.packet = packet;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -10,6 +10,6 @@ spring.datasource.driver-class-name=org.h2.Driver
|
||||||
|
|
||||||
## Hibernate configuration
|
## Hibernate configuration
|
||||||
spring.jpa.database=h2
|
spring.jpa.database=h2
|
||||||
spring.jpa.hibernate.ddl-auto=update
|
spring.jpa.hibernate.ddl-auto=create-drop
|
||||||
spring.jpa.hibernate.naming-strategy=org.hibernate.cfg.ImprovedNamingStrategy
|
spring.jpa.hibernate.naming-strategy=org.hibernate.cfg.ImprovedNamingStrategy
|
||||||
spring.jpa.show-sql=true
|
spring.jpa.show-sql=true
|
|
@ -0,0 +1,97 @@
|
||||||
|
package com.loafle.bridge.discoveryport.repository;
|
||||||
|
|
||||||
|
import com.loafle.bridge.Application;
|
||||||
|
import com.loafle.bridge.discoveryport.constant.DirectionType;
|
||||||
|
import com.loafle.bridge.discoveryport.constant.PortType;
|
||||||
|
import com.loafle.bridge.discoveryport.entity.DiscoveryPort;
|
||||||
|
import com.loafle.bridge.discoveryport.entity.ServiceScanHistory;
|
||||||
|
import org.apache.log4j.Logger;
|
||||||
|
import org.junit.After;
|
||||||
|
import org.junit.Before;
|
||||||
|
import org.junit.Test;
|
||||||
|
import org.junit.runner.RunWith;
|
||||||
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
|
import org.springframework.test.context.ContextConfiguration;
|
||||||
|
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
|
||||||
|
import org.springframework.transaction.annotation.Transactional;
|
||||||
|
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.Date;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
import static org.junit.Assert.assertEquals;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Created by root on 16. 11. 15.
|
||||||
|
*/
|
||||||
|
|
||||||
|
@RunWith(SpringJUnit4ClassRunner.class)
|
||||||
|
@ContextConfiguration(classes = Application.class)
|
||||||
|
public class DiscoveryPortRepositoryTest {
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private DiscoveryPortRepository repo;
|
||||||
|
|
||||||
|
Logger l = Logger.getLogger(this.getClass());
|
||||||
|
|
||||||
|
|
||||||
|
@Before
|
||||||
|
public void Before() {
|
||||||
|
repo.save(new DiscoveryPort(PortType.TCP,null ,(short)1, new Date(), new Date()));
|
||||||
|
repo.save(new DiscoveryPort(PortType.UDP,null , (short)2, new Date(), new Date()));
|
||||||
|
repo.save(new DiscoveryPort(PortType.TCP,null , (short)3, new Date(), new Date()));
|
||||||
|
repo.save(new DiscoveryPort(PortType.UDP,null , (short)4, new Date(), new Date()));
|
||||||
|
repo.save(new DiscoveryPort(PortType.TCP,null , (short)5, new Date(), new Date()));
|
||||||
|
}
|
||||||
|
|
||||||
|
@After
|
||||||
|
public void After() {
|
||||||
|
repo.deleteAll();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void TestInsertPort() {
|
||||||
|
DiscoveryPort p = new DiscoveryPort(PortType.TCP, null ,(short)1, new Date(), new Date());
|
||||||
|
|
||||||
|
repo.save(p);
|
||||||
|
DiscoveryPort compare = repo.findOne(p.getId());
|
||||||
|
|
||||||
|
assertEquals(p.getId(),compare.getId());
|
||||||
|
|
||||||
|
l.debug(p.getId());
|
||||||
|
l.debug(p.getType());
|
||||||
|
l.debug(p.getPortNumber());
|
||||||
|
l.debug(p.getCreateAt());
|
||||||
|
l.debug(p.getUpdateAt());
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
@Transactional
|
||||||
|
public void TestInsertServiceHistory() {
|
||||||
|
List<DiscoveryPort> list = repo.findAll();
|
||||||
|
|
||||||
|
DiscoveryPort p = list.get(0);
|
||||||
|
|
||||||
|
List<ServiceScanHistory> h = new ArrayList<ServiceScanHistory>();
|
||||||
|
byte [] a = {10,20,30,40,50};
|
||||||
|
h.add(new ServiceScanHistory(p,"testservice1", DirectionType.S,a, new Date()));
|
||||||
|
h.add(new ServiceScanHistory(p,"testservice2", DirectionType.R,a, new Date()));
|
||||||
|
h.add(new ServiceScanHistory(p,"testservice3", DirectionType.S,a, new Date()));
|
||||||
|
h.add(new ServiceScanHistory(p,"testservice4", DirectionType.R,a, new Date()));
|
||||||
|
h.add(new ServiceScanHistory(p,"testservice5", DirectionType.S,a, new Date()));
|
||||||
|
|
||||||
|
p.setHistories(h);
|
||||||
|
repo.save(p);
|
||||||
|
|
||||||
|
repo.flush();
|
||||||
|
DiscoveryPort find = repo.findOne(p.getId());
|
||||||
|
|
||||||
|
List<ServiceScanHistory> l1 = p.getHistories();
|
||||||
|
List<ServiceScanHistory> l2 = find.getHistories();
|
||||||
|
|
||||||
|
assertEquals(l1.size(), l2.size());
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user