FwCookie.java
3.83 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
package cn.fw.freya.model.data;
import cn.fw.freya.utils.DateUtil;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.ToString;
import org.apache.http.impl.cookie.BasicClientCookie;
import org.hibernate.Hibernate;
import org.openqa.selenium.Cookie;
import javax.persistence.*;
import java.time.LocalDateTime;
import java.util.Objects;
/**
* @author null
* @date 2021-11-12 09:06
* @description cookie
*/
@Data
@Entity
@ToString
@AllArgsConstructor
@NoArgsConstructor
@Table(name = "cookie", indexes = {@Index(columnList = "accountNo")})
public class FwCookie {
@Id
@GeneratedValue(strategy = GenerationType.IDENTITY)
private Long id;
/**
* 账户号
*/
private String accountNo;
/**
* 账户类型 (1:快手, 2:抖音, 3:懂车帝, 4:Bilibili)
*/
private Integer type;
/**
* cookie名称
*/
private String name;
/**
* cookie值
*/
@Column(length = 2000)
private String value;
/**
* cookie的作用域
*/
private String domain;
/**
* 过期时间
*/
private LocalDateTime expiryDate;
/**
* 接口路径
*/
private String path;
/**
* 是否httpOnly
*/
private Boolean httpOnly;
/**
* 是否安全的
*/
private Boolean secure;
/**
* 版本号
*/
private Integer version;
/**
* cookie创建时间
*/
private LocalDateTime createTime;
/**
* 数据库cookies->seleniumCookie
*
* @return
*/
public Cookie seleniumCookie() {
return new Cookie.Builder(this.name, this.value)
.path(this.path)
.expiresOn(DateUtil.localDateTime2Date(Objects.nonNull(this.expiryDate) ? this.expiryDate : LocalDateTime.now().plusDays(1)))
.domain(this.domain)
.isSecure(this.secure)
.isHttpOnly(this.httpOnly)
.build();
}
public BasicClientCookie httpClientCookie() {
BasicClientCookie cookie = new BasicClientCookie(this.name, this.value);
cookie.setDomain(this.domain);
cookie.setSecure(this.secure);
cookie.setExpiryDate(DateUtil.localDateTime2Date(this.expiryDate));
cookie.setPath(this.path);
cookie.setVersion(this.version);
cookie.setAttribute(BasicClientCookie.PATH_ATTR, this.path);
cookie.setAttribute(BasicClientCookie.DOMAIN_ATTR, this.domain);
return cookie;
}
/**
* WebDriver获取到的cookies->DB
*
* @param cookie WebDriver获取到的cookie
* @param accountNo 账户号
* @return cookies
*/
public static FwCookie toDb(Cookie cookie, String accountNo, Integer type) {
FwCookie fwCookie = new FwCookie();
fwCookie.setAccountNo(accountNo);
fwCookie.setType(type);
fwCookie.setDomain(cookie.getDomain());
fwCookie.setExpiryDate(DateUtil.date2LocalDateTime(cookie.getExpiry()));
fwCookie.setName(cookie.getName());
fwCookie.setValue(cookie.getValue());
fwCookie.setPath(cookie.getPath());
fwCookie.setSecure(cookie.isSecure());
fwCookie.setCreateTime(LocalDateTime.now());
fwCookie.setHttpOnly(cookie.isHttpOnly());
return fwCookie;
}
@Override
public boolean equals(Object o) {
if (this == o) {
return true;
}
if (o == null || Hibernate.getClass(this) != Hibernate.getClass(o)) {
return false;
}
FwCookie fwCookie = (FwCookie) o;
return id != null && Objects.equals(id, fwCookie.id);
}
@Override
public int hashCode() {
return getClass().hashCode();
}
}