Merge test folder of test branch
[iotivity.git] / test / src / automation / robot / device_lib / src / org / iotivity / test / rfw / common / devicecontroller / linux / LinuxProcessObserver.java
1 /******************************************************************
2  *
3  * Copyright 2018 Samsung Electronics All Rights Reserved.
4  *
5  *
6  *
7  * Licensed under the Apache License, Version 2.0 (the "License");
8  * you may not use this file except in compliance with the License.
9  * You may obtain a copy of the License at
10  *
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  *
13  * Unless required by applicable law or agreed to in writing, software
14  * distributed under the License is distributed on an "AS IS" BASIS,
15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16  * See the License for the specific language governing permissions and
17  * limitations under the License.
18  *
19  *
20  ******************************************************************/
21
22 package org.iotivity.test.rfw.common.devicecontroller.linux;
23
24 import java.io.File;
25 import java.io.FileNotFoundException;
26 import java.io.FileOutputStream;
27 import java.io.IOException;
28 import java.io.OutputStream;
29 import java.util.logging.Logger;
30
31 import org.iotivity.test.rfw.common.devicecontroller.interfaces.IDeviceController;
32 import org.iotivity.test.rfw.common.devicecontroller.interfaces.ILogObserver;
33 import org.iotivity.test.rfw.common.devicecontroller.tizen.TizenConsoleManager;
34 import org.iotivity.test.rfw.common.util.IoTivityLogger;
35
36 public class LinuxProcessObserver implements ILogObserver {
37     private File          logFile;
38     private OutputStream  logOutStream;
39     private String        logObserverID;
40     private StringBuilder logBuf;
41     private boolean       isEnabled = true;
42     private boolean       makeFile  = false;
43     private String        pattern;
44     private Logger        logger    = IoTivityLogger.GetLogger();
45
46     public LinuxProcessObserver(String logObserverID, String pattern,
47             boolean makeFile, String logDirPath) {
48         this.logObserverID = logObserverID;
49         this.logBuf = new StringBuilder();
50         this.pattern = pattern;
51         this.makeFile = makeFile;
52         logFile = new File(logDirPath + "/" + logObserverID + ".log");
53
54         try {
55             logOutStream = new FileOutputStream(logFile);
56         } catch (FileNotFoundException e) {
57             e.printStackTrace();
58         }
59     }
60
61     @Override
62     public boolean startLogging() {
63         isEnabled = true;
64
65         return true;
66     }
67
68     @Override
69     public boolean stopLogging() {
70         isEnabled = false;
71
72         return true;
73     }
74
75     @Override
76     public boolean cleanLog() {
77         logBuf.delete(0, logBuf.length());
78         return true;
79     }
80
81     @Override
82     public String getLogs(String filter) {
83         //synchronized (this) {
84             StringBuilder filteredBuf = new StringBuilder();
85             String[] copyofLogBuf = logBuf.toString().split("\n");
86             for (String line : copyofLogBuf) {
87                 if (line.contains(filter)) {
88                     filteredBuf.append(line);
89                 }
90             }
91
92             return filteredBuf.toString();
93         //}
94     }
95
96     @Override
97     public String getLogs() {
98         //synchronized (this) {
99             return logBuf.toString();
100         //}
101     }
102
103     @Override
104     public boolean registerObserverable(IDeviceController controller) {
105         logger.info("registerObserverable");
106
107         controller.getLogManager().addObserver(logObserverID, this);
108
109         return true;
110     }
111
112     /*
113      * This function will register the observable only for Console to device
114      * controller This function added to give support Tizen Console
115      */
116     public boolean registerConsoleObserverable(IDeviceController controller) {
117         logger.info("registerConsoleObserverable");
118
119         TizenConsoleManager tizenConsoleManger = (TizenConsoleManager) controller
120                 .getConsoleManager();
121         tizenConsoleManger.getConsoleLogManager().addObserver(logObserverID,
122                 this);
123
124         logger.info("registerConsoleObserverable DONE");
125
126         return true;
127     }
128
129     @Override
130     public boolean unregisterObserverable(IDeviceController controller) {
131         try {
132             logOutStream.close();
133         } catch (IOException e) {
134             e.printStackTrace();
135         }
136
137         controller.getLogManager().removeObserver(logObserverID, this);
138
139         return true;
140     }
141
142     private void writeFile(String log) {
143         if (makeFile) {
144             try {
145                 logOutStream.write(log.getBytes(), 0, log.getBytes().length);
146                 logOutStream.flush();
147             } catch (IOException e) {
148                 e.printStackTrace();
149             }
150         }
151     }
152
153     @Override
154     public void update(String log) {
155         if (isEnabled) {
156             //synchronized (this) {
157                 if ((pattern == null || pattern.isEmpty())) {
158                     logBuf.append(log);
159                     writeFile(log);
160                     return;
161                 }
162                 if (log.contains(pattern)) {
163                     logBuf.append(log);
164                     writeFile(log);
165                 }
166             //}
167         }
168     }
169 }